python里使用协程来创建echo服务器

最新推荐文章于 2022-07-11 07:36:00 发布

caimouse

最新推荐文章于 2022-07-11 07:36:00 发布

阅读量1.4k

点赞数

分类专栏： milang(小语）文章标签： python tensorflow asyncio

本文链接：https://blog.csdn.net/caimouse/article/details/77946753

版权

milang(小语）专栏收录该内容

389 篇文章 7 订阅

订阅专栏

asyncio库的主作用就是用来构造基于I/O方面的程序。在这里通过一个echo服务器来学会怎么使用协程来创建服务器的方式。
每次I/O操作完成之后，就会释放控制权，交回到事件循环里。

代码如下：

import asyncio
import logging
import sys

SERVER_ADDRESS = ('localhost', 10000)


class EchoServer(asyncio.Protocol):

    def connection_made(self, transport):
        self.transport = transport
        self.address = transport.get_extra_info('peername')
        self.log = logging.getLogger(
            'EchoServer_{}_{}'.format(*self.address)
        )
        self.log.debug('connection accepted')

    def data_received(self, data):
        self.log.debug('received {!r}'.format(data))
        self.transport.write(data)
        self.log.debug('sent {!r}'.format(data))

    def eof_received(self):
        self.log.debug('received EOF')
        if self.transport.can_write_eof():
            self.transport.write_eof()

    def connection_lost(self, error):
        if error:
            self.log.error('ERROR: {}'.format(error))
        else:
            self.log.debug('closing')
        super().connection_lost(error)


logging.basicConfig(
    level=logging.DEBUG,
    format='%(name)s: %(message)s',
    stream=sys.stderr,
)
log = logging.getLogger('main')

event_loop = asyncio.get_event_loop()

# Create the server and let the loop finish the coroutine before
# starting the real event loop.
factory = event_loop.create_server(EchoServer, *SERVER_ADDRESS)
server = event_loop.run_until_complete(factory)
log.debug('starting up on {} port {}'.format(*SERVER_ADDRESS))

# Enter the event loop permanently to handle all connections.
try:
    event_loop.run_forever()
finally:
    log.debug('closing server')
    server.close()
    event_loop.run_until_complete(server.wait_closed())
    log.debug('closing event loop')
    event_loop.close()

在代码开始地方导入asyncio和logging库，接着创建一个事件循环。
继承类asyncio.Protocol，创建子类EchoServer来处理与客户端的通讯。
当服务器接收到连接时，就会调用函数connection_made(self, transport)，参数transport是连接的socket封装。
当连接收到数据时，就会调用函数data_received(self, data)，参数data是接收到的数据。
当有一些情况下的连接收到EOF字符，就调用函数eof_received(self)。
当连接关闭时，就会调用connection_lost(self, error)，参数error是告诉出错原因。

logging.basicConfig是定义调试输出LOG类。
event_loop.create_server(EchoServer, *SERVER_ADDRESS)是创建服务器协程运行，调用
event_loop.run_until_complete(factory)来运行。
另外想服务器程序不要退出就运行event_loop.run_forever()。