完成端口模型实现步骤:
1.创建完成端口
2.创建服务线程,通常服务线程数量为CPU数量的2倍
3.将套接字与完成端口关联在一起
4.调用输入输出函数,发起重叠I/O操作
5.在服务线程中,在完成端口上等待重叠I/O操作结果
一、创建完成端口对象
要在应用程序中利用完成端口模型,就必须首次创建完成端口对象。CreateIoCompletionPort()函数实现此功能,函数声明如下:
HANDLE CreateIoCompletionPort(HANDLE FileHandle, //文件句柄HANDLE ExistingCompletionPort, //存在的完成端口句柄DWORD CompletionKey, //完成键,通常应用程序利用该参数保存于套接字相关信息DWORD NumberOfConcurrentThreads //完成端口并发线程的数量。如果为0,则通知系统完成例程并发线程的数量等于CPU的数量);
如果该函数调用成功则返回完成端口的句柄,如果失败则返回NULL.
在套接字应用程序中,要创建完成端口对象,需要设置FileHandle参数为INVALID_HANDLE_VALUE,ExisstingCompletionPort参数为NULL,CompletionKey参数为0,NumberOfConcurrentThreads参数为0,代码如下:
1 HANDLE hIoCompletionPort;2 hIoCompletionPort=CreateIoCompletionPort(INVALID_HANDLE_VALUE,NULL,0,0);
二、创建服务线程
创建完成端口后,在创建服务线程。首先应用程序要获得计算机CPU的数量,着可以通过调用GetSystemInfo()函数来完成,然后调用CreateThread()或_beginthreadex()函数创建服务线程。通常应用程序要把完成例程作为线程参数传递给服务线程。
以下代码演示了创建服务线程的过程:
1 //服务线程函数 2 DWORD WINAPI ServerWorkerThread(LPVOID CompletionPortID); 3 4 SYSTEM_INFO SystemInfo; //计算机系统信息 5 GetSystemInfo(&SystemInfo); //获取计算机系统信息 6 for(int i=0;i
服务线程开始工作后有3种状态:
(1)等待状态。在没有I/O操作完成通知包被投递到完成端口时,服务线程在完成端口上等待。
(2)服务状态。当I/O操作完成通知到达时,服务线程按照LIFO方式被唤醒,开始为客户端提供服务,当完成服务后,服务线程回到完成端口,继续等待。
(3)阻塞状态。为客户端提供服务的线程,调用了Sleep()之类的函数,线程进入阻塞状态。
三、套接字与完成端口关联
创建完成端口后,需要将套接字与完成端口关联在一起,此功能还是通过调用CreateIoCompletionPort()函数来实现,如调用成功返回端口句柄,即ExistingCompletionPort,失败则返回NULL.
在套接字应用程序中,调用该函数即告知系统当I/O操作完成时,向完成端口发送一个I/O操作完成通知包。这些通知包按照FIFO方式,在完成端口上排队等待服务线程的读取。
以下程序中_completionKey结构体保存客户端套接字和地址,在调用CreateIoCompletionPort()函数时,将该结构体指针作为完成键传递进去。
1 //完成键 2 //结构体 3 typedef struct _completionKey 4 { 5 SOCKET s; 6 SOCKADDR_IN clientAddr; //客户端 7 }COMPLETIONKEY,*PCOMPLETIONKEY; 8 9 SOCKET sListen; //服务器监听套接字10 SOCKET sAccept; //客户端套接字11 HANDLE hIoCompletionPort; //完成端口12 PCOMPLETIONKEY pCompleKey; //完成键13 SOCKADDR_IN addr; //服务器地址14 SOCKADDR_IN clientLen=sizeof(addr); 15 sAccept=accept(sListen,(SOCKADDR*)&addr,&addr); //接受客户端连接请求16 //定义完成键17 pCompleKey->s=sAccept; //端套接字18 pCompleKey->clientAddr=addr; //地址19 //将套接字与完成端口关联起来20 HANDLE h=CreateIoCompletionPort((HANDLE)sAccept,hIoCompletionPort,(DWORD)pCompleKey,0)
四、发起重叠I/O操作
将套接字与完成端口关联起来后,应用程序调用下面的函数,发起发起重叠I/O操作
(1)WSASend()和WSASenTo()函数:发送数据
(2)WSARecv()和WSARecvFrom()函数:接收数据
在应用程序中,通常声明一个和I/O操作相关的结构体,以保存I/O操作的相关信息
以下代码演示了调用WSARecv()函数发起异步接收数据的过程
如果函数返回ERROR_IO_PENDING,则说明成功的发起了一个异步接收数据操作。
1 #define DATA_BUFSIZE 512 2 SOCKET sAccept //客户端套接字 3 //I/O操作数据结构 4 typedef struct _io_operation_data 5 { 6 WSAOVERLAPPEO overlapped; //重叠结构 7 WSABUF dataBuf; 8 CHAR buffer[DATA_BUFSIZE]; //接收数据缓存区 9 }IO_OPERATION_DATA,*PIO_OPERATION_DATA;10 11 12 //定义I/O操作数据13 PIO_OPERATION_DATA pIoDate;14 pIoDate->dataBuf.buf=pIoDate->buffer;15 pIoDate->data.len=DATA_BUFSIZE;16 ZeroMemory(&pIoDate->overlapped,sizeof(pIoDate->overlapped))17 DWORD flags=0;18 DWORD recvBytes;19 20 //接收数据21 if(WSARecv(sAccept,&(pIoDate->dataBuf),1,&recvBytes,&flags,&(pIoDate->overlapped),NULL)==SOCKET_ERROR)22 {23 if(WSAGetLastError()!=ERROR_IO_PENDING) //成功发起异步接收数据操作24 {25 printf("调用WSARecv()函数失败%d\n",WSAGetLastError());26 return;27 }28 }
五、等待重叠I/O操作结果
服务线程被启动后,调用GetQueuedCompletionStatus()(该函数的用法请参看MSDN)函数等待重叠I/O操作的完成结果。当重叠I/O操作完成时,I/O操作完成通知包被发送到完成端口上,此时该函数返回。完成通知包包含的信息有传输的字节数,完成键和重叠结构。
以下代码演示了GetQueuedCompletionStatus()函数的用法:
1 #define THREAD_SLEEP_TIME //超时时间 2 //完成键 3 typedef struct_completionKey 4 { 5 SOCKET s; 6 SOCKADDR_IN clientAddr; //客户端地址 7 }COMPLETIONKEY,*PCOMPLETIONKEY; 8 HANDLE hIoComPort; //完成端口 9 DWORD dwNumberOfByte; //传输字节10 PCOMPLETIONKEY pCompletionKey; //完成键11 LPOVERLAPPED lpOverlapped; //重叠结构指针12 BOOL bRet=GetQueuedCompletionStatus(hIoComPort,&dwNumberOfByte,(LPDWORD )pCompletionKey,&lpOverlapped,THREAD_SLEEP_TIME);13 if(TRUE==bRet)14 {15 //成功的I/O异步操作,处理数据16 }17 else18 {19 int nErrCode=WSAGetLastError(); 20 if(NULL!=lpOverlapped)21 {22 //检查错误代码23 }24 else if(WAIT_TIMEOUT==nErrCode)25 {26 //函数调用超时27 continue;28 }29 else30 {31 //检查错误代码32 }33 }
六、取消异步操作
1 BOOL CancelIo2 {3 HANDLE hFile;4 }
当关闭套接字应用程序时,如果此时系统中还有未完成的异步操作,那么应用程序可以调用CanceIo()函数取消等待执行的异步操作。该函数调用成功则返回值为TRUE,所有在曦套接字上等待的异步操作都被成功地取消。
七、投递完成通知包
当服务线程退出,应用程序可以调用PostQueuedCompletionStatus()(Posts an I/O completion packet to an I/O completion port.)函数想服务线程发送一个特殊的完成通知包。在服务线程中,接收到这通知包后,线程退出。
PostQueuedCompletionStatus()函数功能为向完成端口上发送一个I/O操作完成通知包。在服务线程中,GetQueuedCompletionStatus()函数返回调用PostQueuedCompletionStatus()函数时传递的第2个、第3个和第4个参数值。
以下代码表明利用PostQueuedCompletionStatus()函数向服务线程发送退出通知包的过程。设置PostQueuedCompletionStatus()函数的第3和第4个参数为NULL。在服务线程中,根据GetQueuedCompletionStatus()函数的第3和第4个参数值为NULL,服务线程退出。
1 HANDLE hIoComPort; //完成端口 2 PCOMPLETIONKEY pCompletionKey; //完成键 3 LPOVERLAPPED pOverlapped; //重叠结构指针 4 DWORD dwNumberOfByte; //传输字节数 5 //发送服务线程退出通知包 6 PostQueuedCompletionStatus(hIoComPort,0,(DWORD)NULL,NULL); 7 //等待重叠I/O操作结果 8 GetQueuedCompletionStatus(hIoComPort,&dwNumberOfByte,(LPDWORD)pCompletionKey,&pOverlapped,THREAD_SLEEP_TIME); 9 //其他处理10 if(NULL==pOverlapped&&NULL==pCompletionKey)11 {12 //服务线程退出13 }
这个笔记终于记完了。so pain