Python使用TCPServer编写（多线程）Socket服务

SocketServer包对socket包进行了包装（封装），使得创建socket服务非常简单。

TCPServer+BaseRequestHandler

使用TCPServer和BaseRequestHandler编写socket服务的样例。

[python] view plaincopy

1. #-*- coding:utf-8 -*-  
2.   
3. from SocketServer import TCPServer, BaseRequestHandler  
4. import traceback  
5.   
6. class MyBaseRequestHandlerr(BaseRequestHandler):  
7.     """ 
8.     #从BaseRequestHandler继承，并重写handle方法 
9.     """  
10.     def handle(self):  
11.         #循环监听（读取）来自客户端的数据  
12.         while True:  
13.             #当客户端主动断开连接时，self.recv(1024)会抛出异常  
14.             try:  
15.                 #一次读取1024字节,并去除两端的空白字符(包括空格,TAB,\r,\n)  
16.                 data = self.request.recv(1024).strip()  
17.                   
18.                 #self.client_address是客户端的连接(host, port)的元组  
19.                 print "receive from (%r):%r" % (self.client_address, data)  
20.                   
21.                 #转换成大写后写回(发生到)客户端  
22.                 self.request.sendall(data.upper())  
23.             except:  
24.                 traceback.print_exc()  
25.                 break  
26.   
27. if __name__ == "__main__":  
28.     #telnet 127.0.0.1 9999  
29.     host = ""       #主机名，可以是ip,像localhost的主机名,或""  
30.     port = 9999     #端口  
31.     addr = (host, port)  
32.       
33.     #购置TCPServer对象，  
34.     server = TCPServer(addr, MyBaseRequestHandlerr)  
35.       
36.     #启动服务监听  
37.     server.serve_forever()  

TCPServer+StreamRequestHandler

使用TCPServer和StreamRequestHandler编写socket服务的样例。StreamRequestHandler从BaseRequestHandler，并做了封装，使得读写数据更容易。

[python] view plaincopy

1. #-*- coding:utf-8 -*-  
2.   
3. from SocketServer import TCPServer, StreamRequestHandler  
4. import traceback  
5.   
6. class MyStreamRequestHandlerr(StreamRequestHandler):  
7.     """ 
8.     #继承StreamRequestHandler，并重写handle方法 
9.     #（StreamRequestHandler继承自BaseRequestHandler） 
10.     """  
11.     def handle(self):  
12.         while True:  
13.             #客户端主动断开连接时，self.rfile.readline()会抛出异常  
14.             try:  
15.                 #self.rfile类型是socket._fileobject,读写模式是"rb",方法有  
16.                 #read,readline,readlines,write(data),writelines(list),close,flush  
17.                 data = self.rfile.readline().strip()  
18.                 print "receive from (%r):%r" % (self.client_address, data)  
19.                   
20.                 #self.wfile类型是socket._fileobject,读写模式是"wb"  
21.                 self.wfile.write(data.upper())  
22.             except:  
23.                 traceback.print_exc()  
24.                 break  
25.   
26. if __name__ == "__main__":  
27.     host = ""       #主机名，可以是ip,像localhost的主机名,或""  
28.     port = 9999     #端口  
29.     addr = (host, port)  
30.       
31.     server = TCPServer(addr, MyStreamRequestHandlerr)  
32.     server.serve_forever()  

ThreadingTCPServer+StreamRequestHandler=多线程socket

使用ThreadingTCPServer和StreamRequestHandler编写socket服务的样例。ThreadingTCPServer从ThreadingMixIn和TCPServer继承，实现多线程。

[python] view plaincopy

1. #-*- coding:utf-8 -*-  
2.   
3. from SocketServer import ThreadingTCPServer, StreamRequestHandler  
4. import traceback  
5.   
6. class MyStreamRequestHandlerr(StreamRequestHandler):  
7.     def handle(self):  
8.         while True:  
9.             try:  
10.                 data = self.rfile.readline().strip()  
11.                 print "receive from (%r):%r" % (self.client_address, data)  
12.                 self.wfile.write(data.upper())  
13.             except:  
14.                 traceback.print_exc()  
15.                 break  
16.   
17. if __name__ == "__main__":  
18.     host = ""       #主机名，可以是ip,像localhost的主机名,或""  
19.     port = 9999     #端口  
20.     addr = (host, port)  
21.       
22.     #ThreadingTCPServer从ThreadingMixIn和TCPServer继承  
23.     #class ThreadingTCPServer(ThreadingMixIn, TCPServer): pass  
24.     server = ThreadingTCPServer(addr, MyStreamRequestHandlerr)  
25.     server.serve_forever()  

多线程socket效果：

样例代码下载