168 lines
4.6 KiB
C++
168 lines
4.6 KiB
C++
#include<hgl/network/TCPClient.h>
|
||
#include<hgl/log/LogInfo.h>
|
||
#include<hgl/network/SocketInputStream.h>
|
||
#include<hgl/network/SocketOutputStream.h>
|
||
|
||
/*
|
||
发送数据
|
||
|
||
模式一：
|
||
理论：主线程每循环结束时 主动进行 接收缓冲区锁定、发送缓冲区锁定，如锁定成功，下一循环开始时将其提交到发送线程。
|
||
缺点：虽是多线程，但利用率不高，缓冲区仍是独占，而且长期独占
|
||
|
||
模式二：
|
||
理论：每次发包时锁定缓冲区，发完解锁。
|
||
缺点：过于频繁加锁解锁，影响整体程序效能。
|
||
|
||
理想模式：
|
||
独立的双缓冲区：主线程每次写入的为后备缓冲区，当每循环刷新或是缓冲区满时，锁定缓冲区，与发送线程交换或是复制到发送线程缓冲区。
|
||
*/
|
||
|
||
/*
|
||
接收
|
||
|
||
接收线程模式（现有）：
|
||
理论：接收线程独立一个暂存缓冲区，先从socket接收到暂存缓冲区，然后锁定主缓冲区，复制完成后解锁。
|
||
|
||
主线程模式（旧方案）：
|
||
理论：每次读包时，锁定缓冲区，读完解锁。
|
||
缺点：过于频繁加锁解锁，影响整体程序效能。
|
||
|
||
主线程模式（备选方案）：
|
||
理论：每次访问开始锁定缓冲区，完全读走所有包后，再解锁。
|
||
缺点：长期锁定主缓冲区，会造成接收线程收到数据后无法写入，长时间会造成socket接收缓冲区暴满。
|
||
*/
|
||
|
||
namespace hgl
|
||
{
|
||
namespace network
|
||
{
|
||
void TCPClient::InitPrivate(int sock)
|
||
{
|
||
TimeOut=HGL_NETWORK_TIME_OUT;
|
||
|
||
sis=new SocketInputStream(sock);
|
||
sos=new SocketOutputStream(sock);
|
||
|
||
ipstr = nullptr;
|
||
}
|
||
|
||
/**
|
||
* TCP客户端连接类构造函数
|
||
*/
|
||
TCPClient::TCPClient():TCPSocket()
|
||
{
|
||
InitPrivate(-1);
|
||
}
|
||
|
||
TCPClient::TCPClient(int sock,const IPAddress *addr):TCPSocket(sock,addr)
|
||
{
|
||
InitPrivate(sock);
|
||
}
|
||
|
||
TCPClient::~TCPClient()
|
||
{
|
||
Disconnect();
|
||
SAFE_CLEAR(sis);
|
||
SAFE_CLEAR(sos);
|
||
SAFE_CLEAR_ARRAY(ipstr);
|
||
}
|
||
|
||
/**
|
||
* 连接到服务器
|
||
* @return 是否连接成功
|
||
*/
|
||
bool TCPClient::Connect()
|
||
{
|
||
if(ThisSocket<0||!ThisAddress)RETURN_FALSE;
|
||
|
||
if(connect(ThisSocket,ThisAddress->GetSockAddr(),ThisAddress->GetSockAddrInSize()))
|
||
{
|
||
SAFE_CLEAR(ipstr);
|
||
ipstr=ThisAddress->CreateString();
|
||
|
||
LOG_HINT(U8_TEXT("Don't Connect to TCPServer ")+UTF8String(ipstr));
|
||
this->CloseSocket();
|
||
return(false);
|
||
}
|
||
|
||
// LOG_INFO(u"connect "+u8_to_u16(host)+u':'+UTF16String(port)+u" ok,socket:"+UTF16String(ThisSocket));
|
||
|
||
SetBlock(true,TimeOut); //阻塞模式
|
||
|
||
ResetConnect();
|
||
|
||
return(true);
|
||
}
|
||
|
||
/**
|
||
* 创建一个连接
|
||
* @param addr 服务器地址
|
||
*/
|
||
bool TCPClient::CreateConnect(const IPAddress *addr)
|
||
{
|
||
if(!addr)
|
||
RETURN_FALSE;
|
||
|
||
if(!addr->IsTCP())
|
||
RETURN_FALSE;
|
||
|
||
if(!InitSocket(addr))
|
||
RETURN_FALSE;
|
||
|
||
return this->Connect();
|
||
}
|
||
|
||
/**
|
||
* 与服务器断开
|
||
*/
|
||
void TCPClient::Disconnect()
|
||
{
|
||
SAFE_CLEAR_ARRAY(ipstr);
|
||
|
||
if(ThisSocket==-1)
|
||
return;
|
||
|
||
this->CloseSocket(); //两个线程会自动因为socket关闭而退出
|
||
//注：线程退出会调用ThreadExit，再次调用Disconnect
|
||
}
|
||
|
||
/**
|
||
* 使用指定socket
|
||
* @param sock 指定socket编号
|
||
* @param addr socket地址
|
||
*/
|
||
bool TCPClient::UseSocket(int sock,const IPAddress *addr)
|
||
{
|
||
if(!TCPSocket::UseSocket(sock,addr))
|
||
RETURN_FALSE;
|
||
|
||
((SocketInputStream *)sis)->SetSocket(sock);
|
||
((SocketOutputStream *)sos)->SetSocket(sock);
|
||
|
||
return(true);
|
||
}
|
||
|
||
TCPClient *CreateTCPClient(IPAddress *addr)
|
||
{
|
||
if(!addr)RETURN_ERROR_NULL;
|
||
|
||
if(!addr->IsTCP())
|
||
RETURN_ERROR_NULL;
|
||
|
||
int sock=CreateSocket(addr);
|
||
|
||
if(sock<0)
|
||
RETURN_ERROR_NULL;
|
||
|
||
if(!Connect(sock,addr))
|
||
{
|
||
CloseSocket(sock);
|
||
RETURN_ERROR_NULL;
|
||
}
|
||
|
||
return(new TCPClient(sock,addr));
|
||
}
|
||
}//namespace network
|
||
}//namespace hgl
|