107 lines
2.9 KiB
C++
107 lines
2.9 KiB
C++
|
|
/*
|
|
* @Author: your name
|
|
* @Date: 2021-06-30 16:23:10
|
|
* @LastEditTime: 2021-07-15 22:29:01
|
|
* @LastEditors: Please set LastEditors
|
|
* @Description: In User Settings Edit
|
|
* @FilePath: \server\tcp_server_libevent.h
|
|
*/
|
|
|
|
#ifndef GENERAL_TCPSERVER_H
|
|
#define GENERAL_TCPSERVER_H
|
|
|
|
#ifndef _WIN32_WINNT
|
|
#define _WIN32_WINNT 0x0500
|
|
#endif
|
|
#ifdef linux
|
|
#include<sys/types.h>
|
|
#include<sys/socket.h>
|
|
#include<arpa/inet.h>
|
|
#define EVENT__HAVE_PTHREADS
|
|
#endif
|
|
|
|
extern "C"{
|
|
#include "event2/bufferevent.h"
|
|
#include "event2/buffer.h"
|
|
#include "event2/listener.h"
|
|
#include "event2/util.h"
|
|
#include "event2/event.h"
|
|
#include "event2/thread.h"
|
|
#include <stdint.h>
|
|
};
|
|
|
|
#include <iostream>
|
|
#include <mutex>
|
|
#include <thread>
|
|
#include <map>
|
|
|
|
using namespace std;
|
|
|
|
class TcpServerLibevent;
|
|
// tcp 连接
|
|
class ConnectionLibevent{
|
|
public:
|
|
ConnectionLibevent(TcpServerLibevent *p,
|
|
struct bufferevent*v,
|
|
uint32_t fd,
|
|
struct sockaddr_in *p1);
|
|
ConnectionLibevent(struct bufferevent*v,
|
|
uint32_t fd,
|
|
struct sockaddr_in *p1);
|
|
virtual int OnRecv(char *p,uint32_t len); // 接收到
|
|
virtual int OnClose(); // 接收到
|
|
virtual int OnWrite();
|
|
int WriteData(const char *p,uint16_t len);
|
|
int SetServer(TcpServerLibevent *);
|
|
TcpServerLibevent *Server();
|
|
string IpAddress();
|
|
uint32_t SocketFd();
|
|
private:
|
|
int m_bytes_send;
|
|
int m_bytes_recv;
|
|
TcpServerLibevent *m_parent_server;
|
|
struct bufferevent *m_event;
|
|
struct sockaddr_in *m_addr;
|
|
uint32_t m_fd;
|
|
};
|
|
|
|
// 管理服务端
|
|
class TcpServerLibevent{
|
|
typedef enum{
|
|
RUNNING,
|
|
STOP,
|
|
FAIL
|
|
}SERVER_STATUS;
|
|
public:
|
|
typedef ConnectionLibevent* (*OnAccept)(struct bufferevent *ev,uint32_t fd,struct sockaddr_in *p1);
|
|
TcpServerLibevent(int port,string bindip);
|
|
SERVER_STATUS Status();
|
|
~TcpServerLibevent();
|
|
int StartServerSync(); // 同步启动服务器
|
|
int StartServerAsync(); // 异步启动服务
|
|
int RemoveConnection(uint32_t );
|
|
int SetNewConnectionHandle(OnAccept );
|
|
friend void cb_listener(struct evconnlistener *listener, evutil_socket_t fd, struct sockaddr *addr, int len, void *ptr);
|
|
friend void read_cb(struct bufferevent *bev, void *arg);
|
|
friend void event_cb(struct bufferevent *bev, short events, void *arg);
|
|
friend void write_cb(struct bufferevent *bev, void *arg);
|
|
friend void server_run(TcpServerLibevent *p);
|
|
|
|
private:
|
|
uint32_t m_port; // 监听端口号
|
|
string m_bind_ip; // 绑定端口号
|
|
int m_current_conection; // 当前连接数目
|
|
uint16_t m_backlog;
|
|
struct sockaddr_in m_server_addr; // 服务器地址
|
|
struct event_base * m_event_base;
|
|
struct evconnlistener* m_event_listener;
|
|
SERVER_STATUS m_status;
|
|
thread *m_thread;
|
|
map<uint32_t,ConnectionLibevent*> m_map_client;
|
|
OnAccept m_handle_accept;
|
|
int AddConnection(uint32_t fd,ConnectionLibevent *p);
|
|
};
|
|
|
|
|
|
#endif |