软件测试线程池的编码实现[2] 软件测试工具
关键字:数据库设计 @param void *——未使用
@return void——无
*/
void ProcessRequest(void *)
{
SOCKET *pSock;
while (pSock = MsgQueue_Recv(g_pQue))
{
int len;
char buf[1024];
len = recv(*pSock, buf, sizeof(buf)-1, 0);
if ( len > 0 )
{
buf[len] = '\0';
if ( strcmp(buf, "Hello!")== 0 )
{
strcpy(buf, "Ok");
send(*pSock, buf, strlen(buf), 0);
}
}
}
}
可以看出,使用了消息队列后,整个线程池调度非常简单,由于本书中实现的消息队列在进行遍历时不会影响收发,因此可以实时查看有多少客户在线,使用消息队列的以下调用函数便可以实现对整个消息队列的遍历。
INT MsgQueue_EnumNext(MSGQUEUE *pQueue, VISITFUNC VisitFunc);
这里只需要写一个访问函数VisitFunc即可,比如要想将当前所有在线的客户端连接SOCKET ID打印到图形界面中,可以写一个简单的访问函数来实现,编码如下。
INT PrintAllIDToScreen(void *pData)
{
SOCKET *pSock = pData;
WriteToScreen(*pSock); /* WriteToScreen是将数据写到图形界面的函数 */
}
再调用“MsgQueue_EnumNext(g_pQue, PrintAllIPToScreen)”语句就可以实现遍历。
需要注意的是,上面的编码只是在消息队列中放入了一个SOCKET ID,在程序中可以定义一个结构体,结构体中可以包含SOCKET ID、客户端IP地址、连接建立时间等各种信息,然后将结构体发送到消息队列即可。这样遍历的时候就可以看到客户端IP地址、连接建立的时间等更多的信息了。