Python實(shí)現(xiàn)Socket服務(wù)操作技巧分享
Python編程語言的出現(xiàn)給我們帶來了哪些好處呢?對于大多數(shù)開發(fā)人員來說,應(yīng)該是深有體會(huì)的。這一語言優(yōu)秀的功能特點(diǎn)及簡單易學(xué)的優(yōu)勢在很大程度上提高了編程人員程序開發(fā)的效率。在這里我們可以先從Python實(shí)現(xiàn)Socket服務(wù)的相關(guān)技巧來了解一下這一語言的編寫方式。#t#
首先,要明白不是所有的Socket服務(wù)都需要高性能。如果要求高性能,使用IOCP或EPoll模式用C/C++來完成,直接用API寫,用ACE的proactor封裝來完成是比較恰當(dāng)?shù)男袨?。但?dāng)性能不是主要問題時(shí),用Python實(shí)現(xiàn)Socket服務(wù),并享受高的開發(fā)效率將是一件快樂的事。下面,是用python完成的一個(gè)每thread/connect的一個(gè)echo服務(wù)。
經(jīng)常的,在寫一段Python代碼時(shí),我會(huì)先打開《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( )
只用數(shù)行代碼就完成工作,是不是非常輕松愉快。注意,這還不是一個(gè)實(shí)用程序,只是一個(gè)簡單的示例。但這個(gè)示例指示了方向,下面我會(huì)把Python實(shí)現(xiàn)Socket服務(wù)的一些小技巧一一列出。不過,在這之前,用幾行代碼完成一個(gè)測試用的客戶端是一個(gè)不錯(cuò)的主意:
- 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( )
看著上面寫的這些代碼,是不是感覺開發(fā)效率不一般的高 ^_^,下面進(jìn)入正題
現(xiàn)在,我們來解決Python實(shí)現(xiàn)Socket服務(wù)中遇到***個(gè)問題,MyHander是繼承自SocketServer.BaseRequestHandler,但文檔對這個(gè)模塊介紹不怎么詳細(xì)。不詳細(xì)的原因?我想是因?yàn)檫@個(gè)類實(shí)在很簡單。打開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
一眼可知,類實(shí)現(xiàn)的是一個(gè)簡單的template模式,定義了setup, handle, finish讓繼承者重載,模式方法__init__則定義了三個(gè)方法的調(diào)用順序同時(shí)保證三個(gè)方法的運(yùn)行。 很顯然,如果我們要在退出時(shí)關(guān)閉連接,重定義finish是一個(gè)很自然的行為。
- def finish(self):
- self.request.close()
第二個(gè)問題,如何記日志。Python有日志模塊logging。
- import logging
- logging.basicConfig(level=logging.DEBUG,
- format='%(asctime)s %(levelname)s %(message)s',
- filename='log.txt',
- filemode='a+')
不過實(shí)際操作Python實(shí)現(xiàn)Socket服務(wù)中需要做一點(diǎn)點(diǎn)的補(bǔ)充。因?yàn)樵诙嗑€程程序中,要記錄日志需要線程相關(guān)的唯一ID來識別一些東西。我沒有找到直接的線程ID(哪位兄弟找到了請告知),但Python中有一個(gè)名為id的內(nèi)建函數(shù),用來返回一個(gè)對象的identity (注1)。將要記錄的信息預(yù)定義一個(gè)模板,我們就能得到一個(gè)漂亮的輸出了。
- 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('進(jìn)入處理線程')
- def finish(self):
- self.request.close()
- self.Log("退出處理線程")
另外模塊binascii對日志也很有用,我就會(huì)用到binascii.b2a_hex來幫助把一串二進(jìn)制轉(zhuǎn)成可見的ASCII,象接收到的數(shù)據(jù)就***用b2a_hex轉(zhuǎn)換后再記日志。
第三個(gè)問題,超時(shí)處理。不多說了,直接貼出代碼。
- def setup(self):
- self.request.settimeout(60)
- def handle(self):
- while 1:
- try:
- #接收和發(fā)送操作,略
- except Socket.timeout:
- print "caught Socket.timeout exception"
每完成一小步,可以試試用測試程序發(fā)送你想發(fā)送的內(nèi)容進(jìn)行測試。你會(huì)非常高興的看到,Python實(shí)現(xiàn)Socket服務(wù)是如此的簡單。