Python学习笔记（五）多进程实现并发服务器

浏览数：29 / 时间：2015年06月08日

每创建一个TCP连接，就创建一个进程。

代码如下：

# coding: utf-8
import socket
import os
import sys
import signal
import errno
from time import ctime

def hanlde_sigchld(a, b):
    (pid, status) = os.wait()
    print ‘Child %d Finish, status = %d‘ % (pid, status)

def handle_connection(client_socket, client_address):
    while True:
        data = client_socket.recv(1024)
        if not data:
            print ‘disconnect‘, client_address
            client_socket.close()
            break;
        else:
            client_socket.send(‘[%s] %s‘ % (ctime(), data)) #回显消息


if __name__ == ‘__main__‘:
    signal.signal(signal.SIGCHLD, hanlde_sigchld) #安装SIGCHLD的处理函数

    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    listen_address = (‘localhost‘, 9981)
    server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    server_socket.bind(listen_address)
    server_socket.listen(10)

    while True:
        try:
            (client_socket, client_address) = server_socket.accept()
        except IOError, e:            
            if e.errno == errno.EINTR:
                continue #继续等待
            else:
                raise #将异常向外throw
        print ‘Got connection from ‘, client_address
        pid = os.fork()
        if pid == 0:
            server_socket.close()
            handle_connection(client_socket, client_address)
            sys.exit(0)    #防止子进程中忘记exit
        elif pid > 0:
            client_socket.close() #必须关闭