在Python中,既可直接使用socket类,也可使用socketserverasyncore等经过封装的类来进行编码。asyncore这个库中主要包含了两个函数:

  • asyncore.loop([timeout[, use_poll[, map[, count]]]])

  • class asyncore.dispatcher

Eventdescription
handle_connect()Implied by the first read or write event
handle_close()Implied by a read event with no data available
handle_accepted()Implied by a read event on a listening socket

该模块是事件驱动的,loop()函数用于循环监听网络事件,dispatcher相当于一个socket对象,用于网络事件交互。loop()函数负责检测一个dict,dict中保存dispatcher的实例,这个字典被称为channel。每次创建一个dispatcher对象,都会把自己加入到一个默认的dict里面去(当然也可以自己指定channel)。当对象被加入到channel中的时候,socket的行为都已经被定义好,程序只需要调用loop(),一切功能就实现了。

我们定义一个类,它继承dispatcher类,然后我们重写基类的方法。每个继承dispatcher类的对象,都可以看做我们需要处理的一个socket,可以TCP也可以是UDP,甚至是一些不常用的。

这是标准文档中的一个例子。

import asyncore, socket
class http_client(asyncore.dispatcher):
    def __init__(self, host, path):
        asyncore.dispatcher.__init__(self)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.connect( (host, 80) )
        self.buffer = bytes('GET %s HTTP/1.0\r\nHost: %s\r\n\r\n' %
                            (path, host), 'ascii')
    def handle_connect(self):
        pass
    def handle_close(self):
        self.close()
    def handle_read(self):
        print( self.recv(8192))       
    def writable(self):
        return (len(self.buffer) > 0)
    
    def handle_write(self):
        sent = self.send(self.buffer)
        self.buffer = self.buffer[sent:]
c = http_client('www.python.org', '/')
asyncore.loop()

dispatcher类中的writable和readable在检测到一个socket可以写入或者数据到达的时候被调用,并返回一个bool值,决定是否调用handle_read或者handle_write。打开asyncore.py可以看到,dispatcher类中定义的方法writable和readable的定义相当的简单:


def readable(self):
    return True
def writable()
    return True

也就是说,一旦检测到可读或可写,就调用handle_read/handle_write,但在这个例子中重写了父类的writable方法,

def writable(self):
        return (len(self.buffer) > 0)

很明显,当我们有数据需要发送的时候,我们才给writable的调用者返回一个True,这样就不需要在handle_write中再做判断了,逻辑很明确,代码很清晰,美中不足的是理解需要一点时间,但是不算困难吧!

其余的代码看起来就很清晰了,当一个http服务器发送处理完成你的请求,close socket的时候,我们的handle_close()也相应完成自己的使命。close()将对象自身从channel中删除,并且负责销毁socket对象。

def close(self):
    self.del_channel()
    self.socket.close()

loop()函数检测到一个空的channel,将退出循环,程序完成任务,exit。