Python实现Socket服务操作技巧分享

Python实现Socket服务操作技巧分享

Python编程语言的出现给我们带来了哪些好处呢？对于大多数开发人员来说，应该是深有体会的。这一语言优秀的功能特点及简单易学的优势在很大程度上提高了编程人员程序开发的效率。在这里我们可以先从Python实现Socket服务的相关技巧来了解一下这一语言的编写方式。

• 深度剖析Python语言奥秘
• 深度讲述Python标准语言特性
• Python实现网页爬虫基本实现代码解读
• Python数组功能特点分析
• Python连接SqlServer基本应用方式浅析

首先，要明白不是所有的Socket服务都需要高性能。如果要求高性能，使用IOCP或EPoll模式用C/C++来完成，直接用API写，用ACE的proactor封装来完成是比较恰当的行为。但当性能不是主要问题时，用Python实现Socket服务，并享受高的开发效率将是一件快乐的事。下面，是用python完成的一个每thread/connect的一个echo服务。

经常的，在写一段Python代码时，我会先打开《Python Cookbook》O'Reilly）一书，看看有没有所需要的这也是保证效率的一种方式），下面的代码就是摘自此书。

import SocketServer  
class MyHandler(SocketServer.BaseRequestHandler):  
def handle(self):  
while 1:  
dataReceived = self.request.recv(1024)  
if not dataReceived: break  
self.request.send(dataReceived)  
myServer = SocketServer.ThreadingTCPServer(('',8881), MyHandler)  
myServer.serve_forever( ) 

只用数行代码就完成工作，是不是非常轻松愉快。注意，这还不是一个实用程序，只是一个简单的示例。但这个示例指示了方向，下面我会把Python实现Socket服务的一些小技巧一一列出。不过，在这之前，用几行代码完成一个测试用的客户端是一个不错的主意：

import Socket  
remote_host = '127.0.0.1' 
remote_port = 9919 
send_buf = open('binary.txt', 'rb').read()   
#send_bufsend_buf = send_buf.replace('\x0D\x0A', '')   
sock = Socket.Socket(Socket.AF_INET, Socket.SOCK_STREAM)  
sock.connect((remote_host, remote_port))  
sock.send(send_buf)  
response_data = sock.recv(1024)  
print response_data  
sock.close( ) 

看着上面写的这些代码，是不是感觉开发效率不一般的高 ^_^，下面进入正题

现在，我们来解决Python实现Socket服务中遇到第一个问题，MyHander是继承自SocketServer.BaseRequestHandler，但文档对这个模块介绍不怎么详细。不详细的原因？我想是因为这个类实在很简单。打开Lib目录下的SocketServer.py文件，我们直接看代码：

class BaseRequestHandler:  
def __init__(self, request, client_address, server):  
self.request = request  
self.client_address = client_address  
self.server = server  
try:  
self.setup()  
self.handle()  
self.finish()  
finally:  
sys.exc_traceback = None # Help garbage collection  
def setup(self):  
pass  
def handle(self):  
pass  
def finish(self):  
pass 

一眼可知，类实现的是一个简单的template模式，定义了setup, handle, finish让继承者重载，模式方法__init__则定义了三个方法的调用顺序同时保证三个方法的运行。 很显然，如果我们要在退出时关闭连接，重定义finish是一个很自然的行为。

def finish(self):  
self.request.close() 

第二个问题，如何记日志。Python有日志模块logging。

import logging  
logging.basicConfig(level=logging.DEBUG,  
format='%(asctime)s %(levelname)s %(message)s',  
filename='log.txt',  
filemode='a+') 

不过实际操作Python实现Socket服务中需要做一点点的补充。因为在多线程程序中，要记录日志需要线程相关的唯一ID来识别一些东西。我没有找到直接的线程ID哪位兄弟找到了请告知），但Python中有一个名为id的内建函数，用来返回一个对象的identity 注1）。将要记录的信息预定义一个模板，我们就能得到一个漂亮的输出了。

def LogTemplate(self, s):  
return '[id.' + str(id(self.request)) + ']: ' + str(s)  
def Log(self, s):  
ss = self.LogTemplate(s)  
print ss  
logging.info(ss)  
def LogErr(self, s):  
ss = self.LogTemplate(s)  
print ss  
logging.error(ss) 

下面我们可以这样写了

def setup(self):  
self.Log('进入处理线程')  
def finish(self):  
self.request.close()  
self.Log("退出处理线程") 

另外模块binascii对日志也很有用，我就会用到binascii.b2a_hex来帮助把一串二进制转成可见的ASCII，象接收到的数据就最好用b2a_hex转换后再记日志。

第三个问题，超时处理。不多说了，直接贴出代码。

def setup(self):  
self.request.settimeout(60)  
def handle(self):  
while 1:  
try:  
#接收和发送操作，略  
except Socket.timeout:  
print "caught Socket.timeout exception" 

每完成一小步，可以试试用测试程序发送你想发送的内容进行测试。你会非常高兴的看到，Python实现Socket服务是如此的简单。