眾所周知，使用多進(jìn)程的服務(wù)端模型有利于程序的健壯性。傳統(tǒng)的做法是主進(jìn)程負(fù)責(zé)收發(fā)數(shù)據(jù)，然后傳給子進(jìn)程來處理。這種做法的缺陷是需要大量的父子進(jìn)程IPC，對(duì)效率來說是一種損失。
這里，我提出另外一種比較獨(dú)特的做法，就是多個(gè)進(jìn)程share socket，每次進(jìn)程都可以accept，然后來自己處理。
幾個(gè)關(guān)鍵點(diǎn):
1) CreateProcess使用InheritHandle標(biāo)記來share socket handle
2) 通過command line直接向子進(jìn)程來傳遞父socket的值
3）使用Global Mutext來實(shí)現(xiàn)子進(jìn)程互斥的accept
可以改進(jìn)的地方
1) 使用動(dòng)態(tài)進(jìn)程池來程序具有更大的伸縮性
2）監(jiān)控子進(jìn)程的狀態(tài)，處理僵死進(jìn)程
下面是一個(gè)echo server 的例子來展示這項(xiàng)技術(shù), FYI
父進(jìn)程(SSParent.cpp)
#include <stdio.h>
#include <winsock2.h>
#include <windows.h>
#include <process.h>
#define MUTEX_NAME "sschild"
int main(int argc, char* argv[])
{
{ //init
WORD wVersionRequested;
WSADATA wsaData;
wVersionRequested = MAKEWORD( 2, 2 );
WSAStartup( wVersionRequested, &wsaData );
}
SOCKET s = socket(AF_INET,SOCK_STREAM,0);
if(s==INVALID_SOCKET)
{
printf("create socket failed!\n");
return -1;
}
{ //bind&listen
sockaddr_in sa;
sa.sin_family = AF_INET;
sa.sin_port = htons( 1500 );
sa.sin_addr.s_addr = 0 ;
int rc = bind(s,(sockaddr *)&sa,sizeof(sa));
if(rc == SOCKET_ERROR)
{
printf("bind failed:%d\n",::WSAGetLastError());
return -1;
}
listen(s,SOMAXCONN);
}
HANDLE hSocketMutex;
{ //create mutex
hSocketMutex = ::CreateMutex(NULL,FALSE,MUTEX_NAME);
if(hSocketMutex==NULL)
{
printf("fail CreateMutex:%d\n",::GetLastError());
return -1;
}
}
const int CHILD_NUMBER = 5;
HANDLE hProcess[CHILD_NUMBER];
{ //create child process
STARTUPINFO si = { sizeof(si) };
PROCESS_INFORMATION piProcess[CHILD_NUMBER];
char pCmdLine[256];
sprintf(pCmdLine,"SSChild %d",s);
for(int i=0;i<CHILD_NUMBER;++i)
{
if(!CreateProcess(NULL,pCmdLine,NULL,NULL,TRUE,0, NULL, NULL, &si, &piProcess[i]))
{
printf("fail CreateProcess:%d\n",::GetLastError());
return -1;
}
hProcess[i] = piProcess[i].hProcess;
CloseHandle(piProcess[i].hThread);
}
}
::WaitForMultipleObjects(CHILD_NUMBER,hProcess,TRUE,INFINITE);
{//close all child handle
for(int i=0;i<CHILD_NUMBER;++i)
{
CloseHandle(hProcess[i]);
}
}
//clean
CloseHandle(hSocketMutex);
closesocket(s);
WSACleanup( );
return 0;
}
子進(jìn)程(SSChild.cpp)
#include <stdio.h>
#include <winsock2.h>
#include <windows.h>
#include <process.h>
#define MUTEX_NAME "sschild"
int main(int argc, char* argv[])
{
printf("sschild startup!\n");
{ //init
WORD wVersionRequested;
WSADATA wsaData;
wVersionRequested = MAKEWORD( 2, 2 );
WSAStartup( wVersionRequested, &wsaData );
}
DWORD pid = ::GetCurrentProcessId();
HANDLE hSocketMutex;
{ //open mutex
hSocketMutex = ::OpenMutex(MUTEX_ALL_ACCESS,FALSE,MUTEX_NAME);
if(hSocketMutex==NULL)
{
printf("fail OpenMutex:%d\n",::GetLastError());
return -1;
}
}
SOCKET s;
{ //get socket handle from cmdline
if(argc<=1)
{
printf("usage: sschild socket_handle\n");
return -1;
}
s = (SOCKET) atoi(argv[1]);
}
while(1)
{
WaitForSingleObject(hSocketMutex,INFINITE);
sockaddr_in sa;
int add_len = sizeof(sa);
SOCKET c = accept(s,(sockaddr*)&sa,&add_len);
ReleaseMutex(hSocketMutex);
if(c!=INVALID_SOCKET)
{
printf("[%d],client:%s port:%d connected!\n",pid,inet_ntoa(sa.sin_addr),sa.sin_port);
while(1)
{
char buffer[256]={0};
int rc= recv(c,buffer,255,0);
if(rc>0)
{
printf("[%d]recv msg:%s\n",pid,buffer);
send(c,buffer,strlen(buffer)+1,0);
}
else if(rc == SOCKET_ERROR)
{
printf("[%d]recv msg failed:%d\n",pid,::WSAGetLastError());
closesocket(c);
break;
}
else
{
printf("[%d]connection close\n",pid);
closesocket(c);
break;
}
}
}
else
{
printf("[%d]fail accept:%d\n",pid,::WSAGetLastError());
}
}
CloseHandle(hSocketMutex);
return 0;
}