软件测试线程池的编码实现[2]

发布: 2009-9-15 10:16 | 作者: 不详 | 来源: 领测软件测试网 | 查看: 34次 | 进入软件测试论坛讨论

软件测试线程池的编码实现[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地址、连接建立的时间等更多的信息了。