Web服务的本质3
之前已经带过一点了，下面使用socket发一个请求并且接收返回的数据。去掉模块的封装，从比较底层的层面了解一下其中的过程。

HTTP/1.0

用socket自定义http请求：

import socketfrom bs4 import BeautifulSoupclient = socket.socket()# 连接client.connect((‘edu.51cto.com‘, 80))# 发送header = b‘GET / HTTP/1.0\r\nHost: edu.51cto.com\r\n\r\n‘client.sendall(header)# 接收data = client.recv(1024)content = b‘‘while data: ???content += data ???data = client.recv(1024)head, body = content.split(b‘\r\n\r\n‘)print(head)print(len(body), body)soup = BeautifulSoup(body.decode(), features=‘html.parser‘)title = soup.find(‘title‘)print(title)

这里发送的请求头里有HTTP的版本 "HTTP/1.0" ，所以返回的响应头里有这个 “Connection: close” ，这是一个短连接，接收数据就是上面的方式可以判断服务端是否传完了。接收数据到最后会收到一个空，就表示收完了。这个空应该是socket连接断开时发送的。

HTTP/1.1

如果发的HTTP请求版本是 “HTTP/1.1” ，返回的响应头里会有这些 “Transfer-Encoding: chunked\r\nConnection: keep-alive\r\n” 。然后在响应头和响应体之间会是这个 “\r\n\r\n2b0\r\n” ，前面的 “\r\n\r\n” 是响应头和响应体的分隔符，关键是中间的数字，这个是之后要发送的16进制字节数。也就是说这里的数据是分段发送的。每段数据都是前面是字节数，后面是数据，并且这里的分隔符也是 “\r\n” 。大概是这个样子的：

响应头\r\n\r\n2b0\r\n(0x2b0个字符)\r\n27e4\r\n(0x27e4个字符)\r\n1c7c\r\n(0x1c7c个字符)\r\n0\r\n\r\n

像上面这样，最后是会发一个空的，所以是以 "\r\n0\r\n\r\n" 结尾。下面是自己写的实现拼接head和body的方法：

import socketfrom bs4 import BeautifulSoupclient = socket.socket()# 连接client.connect((‘edu.51cto.com‘, 80))# 发送header = b‘GET / HTTP/1.1\r\nHost: edu.51cto.com\r\n\r\n‘client.sendall(header)# 接收data = client.recv(1024)body = b‘‘# 获取请求头while len(data.split(b‘\r\n\r\n‘, 1)) != 2: ???data += client.recv(1024)head, data = data.split(b‘\r\n\r\n‘, 1)print(‘HEAD‘, head)# 拼接bodywhile data != b‘0\r\n\r\n‘: ???while len(data.split(b‘\r\n‘, 1)) != 2: ???????data += client.recv(1024) ???l, b = data.split(b‘\r\n‘, 1) ???length = int(l, base=16) ???if len(b) <= length: ???????body += b ???????length -= len(b) ???????# 一下子把整段数据剩余的部分都读完 ???????# length 可能会很长，但是可能一次收不全，所以得用循环和计数直到收完 ???????while length: ???????????b = client.recv(length) ???????????body += b ???????????length -= len(b) ???????data = b‘‘ ???else: ???????body += b[:length] ???????data = b[length:] ???# 把下一段body开头的 b‘\r\n‘ 切掉 ???while len(data) < 2: ???????data += client.recv(1024) ???else: ???????# 这个断言可以验证之前的逻辑是否有问题 ???????assert data[0:2] == b‘\r\n‘, b‘data error: %d %b‘ % (len(data), data) ???????data = data[2:]print(len(body), body)soup = BeautifulSoup(body.decode(), features=‘html.parser‘)title = soup.find(‘title‘)print(title)

验证body接收是否正确，可以和上面的HTTP/1.0的结果对比一下，看一下body的长度。

自定义异步IO实现HTTP

先补充点 selector 模块的知识，再用异步的 socket 实现 HTTP 请求。

补充知识（fileno() 方法）

select 和 selectors 模块里，需要把创建的socket实例放到监听的列表里。这里，可以添加到监听列表里的可以不是原生的socket实例。这里可以是 fd 也可以是一个拥有 fileno() 方法的对象。
fd ： 文件描述符，是一个整数，它是文件对象的 fileno() 方法的返回值。
这里，我们不仅要把socket对象加到监听列表里，还需要给它绑定一些别的属性。这就需要对socket封装一下。写一个自己类，加一写自己的属性以及一个socket对象的实例属性。关键是在类里实现一个 fileno() 方法，该方法原样返回 socket 实例的 fileno() 方法就可以了：

class HttpResponse(object): ???"""接收一个实例化好的socket对象，在封装一些别的数据""" ???def __init__(self, sk, item): ???????self.sk = sk ???????self.item = item ???def fileno(self): ???????"""请求sockect对象的文件描述符，用于select监听""" ???????return self.sk.fileno()

selector 模块

官方文档：https://docs.python.org/3/library/selectors.html

模块定义了一个 BaseSelector 的抽象基类，以及它的子类，包括：SelectSelector，PollSelector，EpollSelector，DevpollSelector，KqueueSelector。
另外还有一个DefaultSelector类，它其实是以上其中一个子类的别名而已，它自动选择为当前环境中最有效的Selector，所以平时用 DefaultSelector类就可以了，其它用不着。

# 用之前先创建实例sel = selectors.DefaultSelector()

模块定义了两个常量，在注册事件的时候定义响应哪类事件：

• EVENT_READ ： 表示可读的； 它的值其实是1
• EVENT_WRITE ： 表示可写的； 它的值其实是2

上面两个常量是位掩码，是这样定义的：

# generic events, that must be mapped to implementation-specific onesEVENT_READ = (1 << 0)EVENT_WRITE = (1 << 1)

所以应该也可以同时监听两个事件， EVENT_READ+EVENT_WRITE ，也就是3。

抽象基类中的注册事件的方法
fileobj上一小节讲了，传入socket对象或者是其他实现了 fileno() 方法对象。events参数就是上面的两个常量。data参数在select方法里会返回：

register(fileobj, events, data=None) ?# 注册一个文件对象unregister(fileobj) ?# 注销一个已经注册过的文件对象modify(fileobj, events, data=None) ?# 用于修改一个注册过的文件对象，比如从监听可读变为监听可写。

一个文件对象只能注册一个事件。注册的事件可以调用上面的 unregister 方法注销。
另外如果要改变文件对象监听的 event ，则调用上面的 modify 方法。它其实就是 register + unregister，但是使用modify更高效。

抽象基类中的其他方法
select(timeout=None) ：用于选择满足我们监听的event的文件对象。
这个方法如果不设置参数就是阻塞的。返回1个元组 (key, mask) 。
key 就是一个SelectorKey类的实例，
key.fileobj 就是注册方法的第一个参数，也就是传入的文件对象，比如socket对象。
key.data 就是注册方式的第三个参数，一般可以把回调函数传进去。
mask 就是 EVENT 事件的常量，1、2或者也可能是3。

close() ： 关闭 selector。

get_key(fileobj) ： 返回注册文件对象的 key，返回的是 SelectorKey 类的实例，同select方法里的key。

实现异步 IO 的 HTTP

import selectorsimport socketfrom bs4 import BeautifulSoupurl_list = [ ???{‘host‘: ‘edu.51cto.com‘, ‘port‘: 80, }, ???{‘host‘: ‘www.baidu.com‘, ‘port‘: 80, }, ???{‘host‘: ‘www.python-requests.org‘, ‘port‘: 80, ‘url‘: ‘/en/master/‘}, ???{‘host‘: ‘open-falcon.org‘, ‘port‘: 80, ‘url‘: ‘/‘}, ???{‘host‘: ‘www.jetbrains.com‘, ‘port‘: 80},]class HttpSocket(object): ???"""接收一个实例化好的socket对象，在封装一些别的数据""" ???def __init__(self, sk, item): ???????self.sk = sk ???????self.item = item ???????self.host = self.item.get(‘host‘) ???????self.port = self.item.get(‘port‘, 80) ???????self.method = self.item.get(‘method‘, ‘GET‘) ???????self.url = self.item.get(‘url‘, ‘/‘) ???????self.body = self.item.get(‘body‘, ‘‘) ???????self.callback = self.item.get(‘callback‘) ???????self.buffer = [] ?# 请求的返回值记录在这里 ???def fileno(self): ???????"""请求sockect对象的文件描述符，用于select监听""" ???????return self.sk.fileno() ???def create_request_header(self): ???????"""创建请求信息""" ???????request = ‘%s %s HTTP/1.0\r\nHost: %s\r\n\r\n%s‘ % (self.method.upper(), self.url, self.host, self.body) ???????return request.encode(‘utf-8‘) ???def write(self, data): ???????"""把接收到的数据写入 self.buffer""" ???????self.buffer.append(data) ???def finish(self): ???????"""接收完毕后执行的函数""" ???????content = b‘‘.join(self.buffer) ???????head, body = content.split(b‘\r\n\r\n‘, 1) ???????print(head) ???????print(len(body), body) ???????soup = BeautifulSoup(body.decode(), features=‘html.parser‘) ???????title = soup.find(‘title‘) ???????print(title)class AsyncRequest(object): ???def __init__(self): ???????self.sel = selectors.DefaultSelector() ???def add_request(self, item): ???????"""创建连接请求""" ???????host = item.get(‘host‘) ???????port = item.get(‘port‘) ???????client = socket.socket() ???????client.setblocking(False) ???????try: ???????????client.connect((host, port)) ???????except BlockingIOError as e: ???????????pass ?# 至此，已经向服务器发出连接请求了 ???????hsk = HttpSocket(client, item) ???????self.sel.register(hsk, selectors.EVENT_WRITE, self.connect) ???????# 不同同时注册2个事件，下面的注册要等到连接建立之后执行 ???????# self.sel.register(sk, selectors.EVENT_READ, self.accept) ???def connect(self, hsk, mask): ???????"""建立连接后的回调函数 ???????发送请求，然后注册 EVENT_READ 事件 ???????""" ???????print("连接成功:", hsk.item) ???????content = hsk.create_request_header() ???????print("发送请求:", content) ???????hsk.sk.sendall(content) ???????self.sel.modify(hsk, selectors.EVENT_READ, self.accept) ???def accept(self, hsk, mask): ???????"""接收请求返回的内容""" ???????# print("返回信息:", hsk.item) ???????data = hsk.sk.recv(1024) ???????if data: ???????????hsk.write(data) ???????else: ???????????print("接收完毕", hsk.item) ???????????hsk.finish() ???????????self.sel.unregister(hsk) ???def run(self): ???????"""主函数""" ???????while self.sel._fd_to_key: ???????????events = self.sel.select() ???????????for key, mask in events: ???????????????callback = key.data ?# key.data就是sel.register里的第三个参数 ???????????????callback(key.fileobj, mask) ?# key.fileobj就是sel.register里第一次参数if __name__ == ‘__main__‘: ???obj = AsyncRequest() ???for url_dic in url_list: ???????obj.add_request(url_dic) ???obj.run()

接收完毕之后，最后执行的函数，这里是调用finish函数。这个函数最好可以自定义，那么就需要在搞一个callback参数。思路大概是这样的，最后就在finish函数里先可以做一些处理。然后判断一下，如果有callback，则调用callback。否则继续之后finish里之后的代码。
这个callback参数在哪里设置似乎在实现上都没问题：
可以在url_list里加，在HttpSocket的构造函数里提取出来。
或者是先给 AsyncRequest 类的构造函数，然后在add_request方法里实例化HttpSocket的时候再传过去。
再或者给add_request再加个参数，也是在add_request方法里实例化HttpSocket的时候再传过去。

自定义socket实现HTTP

原文地址：http://blog.51cto.com/steed/2310810