服务器 消息18456,TeamTalk--消息服务器的轮训

消息服务器支持TCP长链接和HTTP长轮询两种接入方式，本节主要讲SOCKET的轮训，先看以下几个函数。服务器

void CMsgConn::OnConnect(net_handle_thandle)socket

{函数

m_handle = handle;oop

m_login_time = get_tick_count();ui

g_msg_conn_map.insert(make_pair(handle,this));this

netlib_option(handle,NETLIB_OPT_SET_CALLBACK, (void*)imconn_callback);spa

netlib_option(handle,NETLIB_OPT_SET_CALLBACK_DATA, (void*)&g_msg_conn_map);code

netlib_option(handle,NETLIB_OPT_GET_REMOTE_IP, (void*)&m_peer_ip);ip

netlib_option(handle,NETLIB_OPT_GET_REMOTE_PORT, (void*)&m_peer_port);ci

//根据handle，为socket类添加如上4个参数。

}

// for client connect in

void msg_serv_callback(void* callback_data, uint8_t msg, uint32_t handle, void* pParam)

{

if (msg ==NETLIB_MSG_CONNECT)

{

CMsgConn* pConn = new CMsgConn();

pConn->OnConnect(handle);

//有客户端connect后，将accept后的SOCKET跟新建的CMsgConn相关联，并将SOCKET加入轮训。

}

else

{

log("!!!error msg: %d\n", msg);

}

}

int main(intargc,char*argv[])

{

…  //读配置，初始化SOCKET等操做。

CStrExplode listen_ip_list(listen_ip, ';');

for (uint32_ti = 0;i

ret = netlib_listen(listen_ip_list.GetItem(i),listen_port,msg_serv_callback,NULL);

if (ret ==NETLIB_ERROR)

return ret;

}

……

netlib_eventloop();  //开始轮训，处理定时回调，SOCKET。

}

netlib_eventloop执行后，进开始进行循环，处理定时回调，SOCKET轮训，如下以读为例。

void CBaseSocket::OnRead()

{

if (m_state ==SOCKET_STATE_LISTENING)

{

_AcceptNewSocket();

}

else

{

u_long avail = 0;

if ( (ioctlsocket(m_socket,FIONREAD, &avail) ==SOCKET_ERROR) || (avail == 0) )

{

m_callback(m_callback_data,NETLIB_MSG_CLOSE, (net_handle_t)m_socket,NULL);

}

else

{

m_callback(m_callback_data,NETLIB_MSG_READ, (net_handle_t)m_socket,NULL);

}

}

}

SOCKET可读后，执行如上CBaseSocket::OnRead，紧接着会执行m_callback(m_callback_data,NETLIB_MSG_READ, (net_handle_t)m_socket,NULL);，此时会转到以下回调函数中：

void imconn_callback(void* callback_data, uint8_t msg, uint32_t handle, void* pParam)

{

NOTUSED_ARG(handle);

NOTUSED_ARG(pParam);

if (!callback_data)

return;

ConnMap_t* conn_map = (ConnMap_t*)callback_data;

CImConn* pConn = FindImConn(conn_map,handle);

if (!pConn)

return;

//log("msg=%d, handle=%d\n", msg, handle);

switch (msg)

{

case NETLIB_MSG_CONFIRM:

pConn->OnConfirm();

break;

case NETLIB_MSG_READ:

pConn->OnRead();

break;

case NETLIB_MSG_WRITE:

pConn->OnWrite();

break;

case NETLIB_MSG_CLOSE:

pConn->OnClose();

break;

default:

log("!!!imconn_callback error msg: %d\n", msg);

break;

}

pConn->ReleaseRef();

}

能够看到，会执行以下函数，先从SOCKET中读出数据，在后面会执行CImPdu::ReadPdu。

void CImConn::OnRead()

{

for (;;)

{

uint32_t free_buf_len = m_in_buf.GetAllocSize() -m_in_buf.GetWriteOffset();

if (free_buf_len

m_in_buf.Extend(READ_BUF_SIZE);

int ret = netlib_recv(m_handle,m_in_buf.GetBuffer() +m_in_buf.GetWriteOffset(),READ_BUF_SIZE);

if (ret <= 0)

break;

m_recv_bytes += ret;

m_in_buf.IncWriteOffset(ret);

m_last_recv_tick = get_tick_count();

}

if (m_policy_conn) {

return;

}

// no received data is read by ReadPdu(), check if this is a flash security policy request

if (m_recv_bytes ==m_in_buf.GetWriteOffset()) {

if ( (m_in_buf.GetBuffer()[0] =='

log("policy request, handle=%d\n", m_handle);

m_policy_conn = true;

Send(g_policy_content,g_policy_len);

return;

}

}

try {

CImPdu* pPdu = NULL;

while ( ( pPdu = CImPdu::ReadPdu(m_in_buf.GetBuffer(),m_in_buf.GetWriteOffset()) ) )

{

uint32_t pdu_len = pPdu->GetLength();

HandlePdu(pPdu);

m_in_buf.Read(NULL,pdu_len);

delete pPdu;

++g_recv_pkt_cnt;

}

} catch (CPduException&ex) {

log("!!!catch exception, sid=%u, cid=%u, err_code=%u, err_msg=%s, close the connection\n",

ex.GetModuleId(),ex.GetCommandId(),ex.GetErrorCode(),ex.GetErrorMsg());

OnClose();

}

}

CImPdu* CImPdu::ReadPdu(uchar_t *buf,uint32_tlen)

{

uint32_t pdu_len = 0;

if (!_IsPduAvailable(buf,len,pdu_len))

return NULL;

uint16_t service_id = CByteStream::ReadUint16(buf + 4);

uint16_t command_id = CByteStream::ReadUint16(buf + 6);

CImPdu* pPdu = NULL;

switch (service_id)

{

case SID_LOGIN:

pPdu = ReadPduLogin(command_id, buf, pdu_len);

break;

case SID_BUDDY_LIST:

pPdu = ReadPduBuddyList(command_id, buf, pdu_len);

break;

case SID_GROUP:

pPdu = ReadPduGroup(command_id, buf, pdu_len);

break;

case SID_MSG:

pPdu = ReadPduMsg(command_id, buf, pdu_len);

break;

case SID_OTHER:

pPdu = ReadPduOther(command_id, buf, pdu_len);

break;

case SID_SWITCH_SERVICE:

pPdu = ReadPduSwitchService(command_id, buf, pdu_len);

break;

case SID_FILE:

pPdu = ReadPduFile(command_id, buf, pdu_len);

break;

default:

throw CPduException(service_id, command_id, ERROR_CODE_WRONG_SERVICE_ID,"wrong service id");

}

pPdu->_SetIncomingLen(pdu_len);

pPdu->_SetIncomingBuf(buf);

return pPdu;

}

如今以登录为例子，进入CImPdu*CImPdu::ReadPduLogin。

CImPdu* CImPdu::ReadPduLogin(uint16_tcommand_id,uchar_t*pdu_buf,uint32_tpdu_len)

{

CImPdu* pPdu = NULL;

switch (command_id) {

case CID_LOGIN_REQ_MSGSERVER:

pPdu = new CImPduMsgServRequest(pdu_buf,pdu_len);

break;

case CID_LOGIN_RES_MSGSERVER:

pPdu = new CImPduMsgServResponse(pdu_buf,pdu_len);

break;

case CID_LOGIN_REQ_USERLOGIN:

pPdu = new CImPduLoginRequest(pdu_buf,pdu_len);

break;

case CID_LOGIN_RES_USERLOGIN:

pPdu = new CImPduLoginResponse(pdu_buf,pdu_len);

break;

case CID_LOGIN_KICK_USER:

pPdu = new CImPduKickUser(pdu_buf,pdu_len);

break;

default:

throw CPduException(SID_LOGIN, command_id, ERROR_CODE_WRONG_COMMAND_ID, "wrong command id");

}

return pPdu;

}

能够看到是在打包回复的内容，返回pPdu，而后回到CImConn::OnRead，执行HandlePdu(pPdu)->_HandleLoginRequest-> SendPdu(&pdu)->CBaseSocket::Send。