Delphi-IOCP学习笔记<三>====工作线程和Listener

uses
  JwaWinsock2, Windows, SysUtils;

const
  DATA_BUFSIZE = 1024;

  IO_TYPE_Accept = 1;
  IO_TYPE_Recv = 2;



type
  //（1）：单IO数据结构
  LPVOID = Pointer;
  LPPER_IO_OPERATION_DATA = ^PER_IO_OPERATION_DATA ;
  PER_IO_OPERATION_DATA = packed record
    Overlapped: OVERLAPPED;
    IO_TYPE: Cardinal;
    DataBuf: TWSABUF;
    Buffer: array [0..1024] of CHAR;
  end;

 //下面循环进行循环获取客户端的请求。
  while (TRUE) do
  begin
     //当客户端有连接请求的时候，WSAAccept函数会新创建一个套接字cSocket。这个套接字就是和客户端通信的时候使用的套接字。
     cSocket:= WSAAccept(sSocket, nil, nil, nil, 0);

     //判断cSocket套接字创建是否成功，如果不成功则退出。
     if (cSocket= SOCKET_ERROR) then
     begin
        closesocket(sSocket);
        exit;
     end;

     //将套接字、完成端口绑定在一起。

     //     最开始的时候没有明白为什么还要调用一次createIoCompletionPort
     //
     //     后来经过google,和测试
     //
     //     是将新的套接字(socket)加入到iocp端口<绑定>
     //     这样工作线程才能处理这个套接字(socket)的数据包
     //如果把下面注释掉,WSARecv这个套接字时,GetQueuedCompletionStatus无法处理到收到的数据包
      

     //      2013年4月19日 09:56:00 

     //      注意第三个参数也需要进行绑定, 否则在工作线程中GetQueuedCompletionStatus时completionKey会取不到cSocket值

     lvPerIOPort := CreateIoCompletionPort(cSocket, lvIOPort, cSocket, 0);
     if (lvPerIOPort = 0) then
     begin
        Exit;
     end;

     //初始化数据包
     PerIoData := LPPER_IO_OPERATION_DATA(GlobalAlloc(GPTR, sizeof(PER_IO_OPERATION_DATA)));
     ZeroMemory(@PerIoData.Overlapped, sizeof(OVERLAPPED));

     //数据包中的IO类型:有连接请求
     PerIoData.IO_TYPE := IO_TYPE_Accept;

     //通知工作线程,有新的套接字连接<第三个参数>
     PostQueuedCompletionStatus(lvIOPort, 0, cSocket, POverlapped(PerIOData));
  end;

function ServerWorkerThread(pData:Pointer): Integer; stdcall;
var
  CompletionPort:THANDLE;
  BytesTransferred:Cardinal;
  PerIoData:LPPER_IO_OPERATION_DATA;
  cSocket:TSocket;
  Flags:Cardinal;
  RecvBytes:Cardinal;
  lvResultStatus:BOOL;
  lvRet:Integer;

begin
  CompletionPort:=THandle(pData);
  //得到创建线程是传递过来的IOCP
   while(TRUE) do
   begin
        //工作者线程会停止到GetQueuedCompletionStatus函数处，直到接受到数据为止
        lvResultStatus := GetQueuedCompletionStatus(CompletionPort,
          BytesTransferred,
          cSocket,
          POverlapped(PerIoData), INFINITE);
if (lvResultStatus = False) then
        begin
          //当客户端连接断开或者客户端调用closesocket函数的时候,函数GetQueuedCompletionStatus会返回错误。如果我们加入心跳后，在这里就可以来判断套接字是否依然在连接。
          if cSocket<>0 then
          begin
            closesocket(cSocket);
          end;
          if PerIoData<>nil then
          begin
            GlobalFree(DWORD(PerIoData));
          end;
          continue;
        end;

        if PerIoData = nil then
        begin
          closesocket(cSocket);
          Break;
        end else  if (PerIoData<>nil) then
        begin
          ////shutdown(PerHandleData.Socket, 1);
          if PerIoData.IO_TYPE = IO_TYPE_Accept then  //连接请求
          begin
            GlobalFree(DWORD(PerIoData));
          end else if PerIoData.IO_TYPE = IO_TYPE_Recv then
          begin

            ////可以在这里处理数据……

            GlobalFree(DWORD(PerIoData));
          end;

          /////分配内存<可以加入内存池>
          PerIoData := LPPER_IO_OPERATION_DATA(GlobalAlloc(GPTR, sizeof(PER_IO_OPERATION_DATA)));
          ZeroMemory(@PerIoData.Overlapped, sizeof(OVERLAPPED));
          Flags := 0;

          /////进入投递收取动作
          PerIoData.IO_TYPE := IO_TYPE_Recv;
          PerIoData.DataBuf.len:=DATA_BUFSIZE;
          ZeroMemory(@PerIoData.Buffer,sizeof(@PerIoData.Buffer));
          PerIoData.DataBuf.buf := @PerIoData.Buffer;

          /////异步收取数据
          WSARecv(cSocket,
             @PerIoData.DataBuf,
             1,
             RecvBytes,
             Flags,
             @PerIoData^, nil);
          if (WSAGetLastError() <> ERROR_IO_PENDING) then
          begin
            closesocket(cSocket);
            if PerIoData <> nil then
            begin
              GlobalFree(DWORD(PerIoData));
            end;
            Continue;
          end;
        end;
   end;
end;