SocketServer包对socket包进行了包装(封装),使得创建socket服务非常简单。

TCPServer+BaseRequestHandler

使用TCPServerBaseRequestHandler编写socket服务的样例。

[python] view plaincopy
  1. #-*- coding:utf-8 -*-  
  2.   
  3. from SocketServer import TCPServer, BaseRequestHandler  
  4. import traceback  
  5.   
  6. class MyBaseRequestHandler(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, MyBaseRequestHandler)  
  35.       
  36.     #启动服务监听  
  37.     server.serve_forever()  


TCPServer+StreamRequestHandler

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


[python] view plaincopy
  1. #-*- coding:utf-8 -*-  
  2.   
  3. from SocketServer import TCPServer, StreamRequestHandler  
  4. import traceback  
  5.   
  6. class MyStreamRequestHandler(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, MyStreamRequestHandler)  
  32.     server.serve_forever()  


ThreadingTCPServer+StreamRequestHandler=多线程socket

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


[python] view plaincopy
  1. #-*- coding:utf-8 -*-  
  2.   
  3. from SocketServer import ThreadingTCPServer, StreamRequestHandler  
  4. import traceback  
  5.   
  6. class MyStreamRequestHandler(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, MyStreamRequestHandler)  
  25.     server.serve_forever()  

多线程socket效果: