python网络编程socket之多进程

Posted 2020-07-12 similarface

#coding:utf-8
__author__ = ‘similarface‘

import os,socket,threading,SocketServer
SERVER_HOST=‘localhost‘
SERVER_PORT=0
BUF_SIZE=1024
ECHO_MSG=‘HELLO‘

class ForkedClient():
    def __init__(self,ip,port):
        #建立sock
        self.sock=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
        #连接server
        self.sock.connect((ip,port))

    def run(self):
        #获取当前的进程号
        current_process_id=os.getpid()
        print ‘PID %s ：" "%s"‘%(current_process_id,ECHO_MSG)
        sent_data_length = self.sock.send(ECHO_MSG)
        print(‘发送:%d 长数据‘ % sent_data_length)
        #sever response
        response=self.sock.recv(BUF_SIZE)
        print "PID %s received: %s" % (current_process_id,response)

    def shutdown(self):
        ‘‘‘
        清理sock的资源
        :return:
        ‘‘‘
        self.sock.close()
        
class ForkingServerRequestHandler(SocketServer.BaseRequestHandler):
    def handle(self):
        #接受客户端的数据
        data=self.request.recv(BUF_SIZE)
        #当前的进程号码
        current_process_id=os.getpid()
        #返回的数据包
        response=‘%s: %s‘ % (current_process_id, data)
        print "服务器返回 [current_process_id: data] =[%s]" %response
        #发送返回数据
        self.request.send(response)
        return

class ForkingServer(SocketServer.ForkingMixIn,SocketServer.TCPServer):
    pass

def main():
    #建立server
    server=ForkingServer((SERVER_HOST,SERVER_PORT),ForkingServerRequestHandler)
    #获取IP 和 端口
    ip, port = server.server_address # Retrieve the port number
    #服务进程
    server_thread = threading.Thread(target=server.serve_forever)
    #是否放入后台
    server_thread.setDaemon(True) # don‘t hang on exit
    #启动
    server_thread.start()
    print ‘Server loop running PID: %s‘ %os.getpid()
    # Launch the client(s)
    client1 =  ForkedClient(ip, port)
    client1.run()
    client2 =  ForkedClient(ip, port)
    client2.run()
    # Clean them up
    server.shutdown()
    client1.shutdown()
    client2.shutdown()
    server.socket.close()
if __name__ == ‘__main__‘:
    main()