server_impl.h

/****************************************************************************
 Copyright (c) 2013-2014 King Lee

 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
 in the Software without restriction, including without limitation the rights
 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 copies of the Software, and to permit persons to whom the Software is
 furnished to do so, subject to the following conditions:

 The above copyright notice and this permission notice shall be included in
 all copies or substantial portions of the Software.

 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
 ****************************************************************************/
#ifndef server_impl_h__
#define server_impl_h__
/************************************************************************/
/*
 *  a ring buffer to work with network buffer cache
 *  bugs:
 *  #20004	2014-12-08
 *  memory overflow
 *  #20005	2015-2-4
 *  data lose
 *
 */
/************************************************************************/
#include <map>
#include <vector>
#include <list>
#include "event_handle_srv.h"
//	the follows files can get from git@github.com:yuyunliuhen/easy.git,make the easy project at the same directory.
#include "easy_ring_buffer.h"
#include "easy_allocator.h"
#include "easy_lock.h"
#include "easy_locked_queue.h"

//	class forward declaration
class Reactor;
struct Buffer;

struct Buffer {
    typedef easy::EasyRingbuffer<easy_uint8,easy::alloc,easy::mutex_lock>	ring_buffer;

    static const size_t MAX_POOL_SIZE = 50000;

    typedef  easy_int32 _Key;
    //	for input buffer
    ring_buffer*		input_;
    //	for output buffer
    ring_buffer*		output_;
    //	incoming socket
    easy_int32			fd_;
    //	the status of incoming socket
    easy_int32			invalid_fd_;

    Buffer(easy_int32 __fd,easy_uint32 __max_buffer_size) {
        input_ = new easy::EasyRingbuffer<easy_uint8,easy::alloc,easy::mutex_lock>(__max_buffer_size);
        output_ = new easy::EasyRingbuffer<easy_uint8,easy::alloc,easy::mutex_lock>(__max_buffer_size);
        fd_ = __fd;
        invalid_fd_ = 1;
    }
    void init(easy_int32 __fd,easy_uint32 __max_buffer_size) {
        input_->reset();
        output_->reset();
        fd_ = __fd;
        invalid_fd_ = 1;
    }

    void clear() {
        if(input_) {
            delete input_;
            input_ = NULL;
        }
        if(output_) {
            delete output_;
            output_ = NULL;
        }
    }
};

class Server_Impl : public Event_Handle_Srv {
  public:
    //	constructor function
    Server_Impl(Reactor* __reactor,const easy_char* __host ,easy_uint32 __port);

    //	destructor function, add virtual qualifier to avoid memory leak
    virtual ~Server_Impl();

    //	a new connection coming
    void on_connected(easy_int32 __fd);

    //	a connection disconnect
    void on_disconnect(easy_int32 __fd);

    //	a read event trigger
    void on_read(easy_int32 __fd);

    //	time to handler packet
    //	for byte stream, it is the  default way
    easy_int32 on_packet(easy_int32 __fd,const easy_char* __packet,easy_int32 __length);

    //	for protobuf
    easy_int32 on_packet(easy_int32 __fd,const std::string& __string_packet);

  public:
    //	callback function, you should define this interface as follows
    //	called at a packet to be handle
    //	for byte stream, it is the  default way
    virtual easy_int32 handle_packet(easy_int32 __fd,const easy_char* __packet,easy_int32 __length) = 0;

    //	for protobuf
    virtual easy_int32 handle_packet(easy_int32 __fd,const std::string& __string_packet,void* __user_data) = 0;

    //	called at a connection coming
    virtual	void connected(easy_int32 __fd) = 0;

    //	called at a connection leaving
    virtual	void dis_connected(easy_int32 __fd) = 0;

    //	is proxy server
    virtual easy_bool is_proxy() {
        return false;
    }

    //	is login server,just for test
    virtual easy_bool is_login() {
        return false;
    }

//protected:
    //	send packet to special connection
    void send_packet(easy_int32 __fd,const easy_char* __packet,easy_int32 __length);

  private:
    //	read completely from system cache,it design for EPOLL model
    void _read_completely(easy_int32 __fd);

    //	do not process the date and read directly,use mmap replace recv/read,usually it use as proxy server
    void _read_directly(easy_int32 __fd);

    //	a thread work with read data
    void _read_thread();

    //	a thread work with write data
    void _write_thread();

    //	disconnect a connection
    void _disconnect(Buffer* __buffer);

  private:
    //	all connection
    typedef	std::map<easy_int32,Buffer*>	map_buffer;
    std::map<easy_int32,Buffer*>			connects_;

    //	a copy of all connections, whiych stores a point, it's quickly to travel all.
    typedef	std::vector<Buffer*>	vector_buffer;
    std::vector<Buffer*>			connects_copy;

    //	the max size of read/write buffer
    static const easy_uint32		max_buffer_size_;

    //	the max time of read/write sleep time
    static const easy_uint32		max_sleep_time_;

    //	a lock for all connections
    easy::mutex_lock				lock_;

    //	buffer memory manager queue
    easy::lock_queue<Buffer,easy::mutex_lock,std::list<Buffer*> >	buffer_queue_;
};

#endif // server_impl_h__