89 lines
2.5 KiB
C
89 lines
2.5 KiB
C
|
#ifndef HGL_NETWORK_MULTI_THREAD_ACCEPT_INCLUDE
|
|||
|
|
#define HGL_NETWORK_MULTI_THREAD_ACCEPT_INCLUDE
|
|||
|
|
|
|||
|
|
#include<hgl/thread/Thread.h>
|
|||
|
|
#include<hgl/type/Stack.h>
|
|||
|
|
#include<hgl/thread/Semaphore.h>
|
|||
|
|
namespace hgl
|
|||
|
|
{
|
|||
|
|
namespace network
|
|||
|
|
{
|
|||
|
|
class IPAddress;
|
|||
|
|
class AcceptServer;
|
|||
|
|
|
|||
|
|
using IPAddressStack=Stack<IPAddress *>;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* Socket接入线程
|
|||
|
|
*/
|
|||
|
|
class AcceptThread:public Thread
|
|||
|
|
{
|
|||
|
|
AcceptServer *server;
|
|||
|
|
Semaphore *active_semaphore;
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
|
|||
|
|
IPAddressStack ip_stack; ///<IP地址空间堆栈，会提前分配好空间供未来使用
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
AcceptThread(AcceptServer *,Semaphore *);
|
|||
|
|
virtual ~AcceptThread()=default;
|
|||
|
|
|
|||
|
|
bool Execute() override;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 接受一个接入
|
|||
|
|
* @param sock Socket值
|
|||
|
|
* @param addr ip地址(注:需自行delete)
|
|||
|
|
*/
|
|||
|
|
virtual bool OnAccept(int sock,IPAddress *addr)=0; ///<接入一个新Socket
|
|||
|
|
};//class AcceptThread:public Thread
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 多Socket接入线程管理模板
|
|||
|
|
*/
|
|||
|
|
template<typename ACCEPT_THREAD> class MultiThreadAccept
|
|||
|
|
{
|
|||
|
|
MultiThreadManage<ACCEPT_THREAD> accept_thread_manage;
|
|||
|
|
|
|||
|
|
Semaphore active_semaphore;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
|
|||
|
|
virtual ~MultiThreadAccept()=default;
|
|||
|
|
|
|||
|
|
bool Init(AcceptServer *server,const uint thread_count)
|
|||
|
|
{
|
|||
|
|
if(!server)return(false);
|
|||
|
|
if(thread_count<=0)return(false);
|
|||
|
|
|
|||
|
|
for(int i=0;i<thread_count;i++)
|
|||
|
|
accept_thread_manage.Add(new ACCEPT_THREAD(server,&active_semaphore));
|
|||
|
|
|
|||
|
|
return(true);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ACCEPT_THREAD *GetAcceptThread(const int index)
|
|||
|
|
{
|
|||
|
|
return accept_thread_manage.GetThread(index);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bool Start (){return accept_thread_manage.Start()>0;}
|
|||
|
|
int IsLive (){return accept_thread_manage.IsLive();}
|
|||
|
|
bool Close (){return accept_thread_manage.Close();}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 等待接收信号
|
|||
|
|
*/
|
|||
|
|
bool Wait(const double &time_over=5)
|
|||
|
|
{
|
|||
|
|
return active_semaphore.Acquire(time_over);
|
|||
|
|
}
|
|||
|
|
};//template<typename ACCEPT_THREAD> class MultiThreadAccept
|
|||
|
|
}//namespace network
|
|||
|
|
}//namespace hgl
|
|||
|
|
#endif//HGL_NETWORK_MULTI_THREAD_ACCEPT_INCLUDE
|