mirror of
https://github.com/caozhiyi/CppNet.git
synced 2026-01-29 16:51:12 +08:00
84 lines
3.0 KiB
C++
84 lines
3.0 KiB
C++
#ifndef HEADER_NET_CPPNETIMPL
|
|
#define HEADER_NET_CPPNETIMPL
|
|
|
|
#include <unordered_map>
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <string>
|
|
|
|
#include "PoolSharedPtr.h"
|
|
#include "EventHandler.h"
|
|
#include "MemaryPool.h"
|
|
#include "Timer.h"
|
|
#include "Single.h"
|
|
#include "CppDefine.h"
|
|
|
|
namespace cppnet {
|
|
|
|
class CEventActions;
|
|
class CSocket;
|
|
class CAcceptSocket;
|
|
class CCppNetImpl : public base::CSingle<CCppNetImpl>
|
|
{
|
|
public:
|
|
CCppNetImpl();
|
|
~CCppNetImpl();
|
|
// common
|
|
void Init(uint32_t thread_num, bool per_handl_thread);
|
|
void Dealloc();
|
|
void Join();
|
|
|
|
// set call back
|
|
void SetReadCallback(const read_call_back& func);
|
|
void SetWriteCallback(const write_call_back& func);
|
|
void SetDisconnectionCallback(const connection_call_back& func);
|
|
|
|
// about timer
|
|
uint64_t SetTimer(uint32_t interval, const std::function<void(void*)>& func, void* param = nullptr, bool always = false);
|
|
void RemoveTimer(uint64_t timer_id);
|
|
|
|
//server
|
|
void SetAcceptCallback(const connection_call_back& func);
|
|
bool ListenAndAccept(uint16_t port, std::string ip, uint32_t listen_num);
|
|
|
|
//client
|
|
void SetConnectionCallback(const connection_call_back& func);
|
|
#ifndef __linux__
|
|
Handle Connection(uint16_t port, std::string ip, const char* buf, uint32_t buf_len);
|
|
#endif
|
|
Handle Connection(uint16_t port, std::string ip);
|
|
|
|
// get socket
|
|
base::CMemSharePtr<CSocketImpl> GetSocket(const Handle& handle);
|
|
bool RemoveSocket(const Handle& handle);
|
|
// get thread number
|
|
uint32_t GetThreadNum();
|
|
|
|
private:
|
|
void _AcceptFunction(base::CMemSharePtr<CAcceptEventHandler>& event, uint32_t err);
|
|
void _ReadFunction(base::CMemSharePtr<CEventHandler>& event, uint32_t err);
|
|
void _WriteFunction(base::CMemSharePtr<CEventHandler>& event, uint32_t err);
|
|
std::shared_ptr<CEventActions>& _RandomGetActions();
|
|
|
|
private:
|
|
friend class CSocketImpl;
|
|
friend class CAcceptSocket;
|
|
read_call_back _read_call_back = nullptr;
|
|
write_call_back _write_call_back = nullptr;
|
|
connection_call_back _connection_call_back = nullptr;
|
|
connection_call_back _disconnection_call_back = nullptr;
|
|
connection_call_back _accept_call_back = nullptr;
|
|
|
|
base::CMemoryPool _pool;
|
|
bool _per_epoll_handle;
|
|
|
|
std::mutex _mutex;
|
|
std::vector<std::shared_ptr<std::thread>> _thread_vec;
|
|
std::unordered_map<uint64_t, base::CMemSharePtr<CAcceptSocket>> _accept_socket;
|
|
std::unordered_map<uint64_t, base::CMemSharePtr<CSocketImpl>> _socket_map;
|
|
std::unordered_map<std::thread::id, std::shared_ptr<CEventActions>> _actions_map;
|
|
std::unordered_map<uint64_t, std::weak_ptr<CEventActions>> _timer_actions_map;
|
|
};
|
|
|
|
}
|
|
#endif |