Question

我正在尝试编写一个服务器端事件服务器，我可以使用telnet连接它，并将telnet内容推送到浏览器。使用Python和asyncio背后的想法是使用尽可能少的CPU，因为它将在Raspberry Pi上运行。

到目前为止，我有以下使用库中的库：https://pypi.python.org/pypi/asyncio-sse/0.1使用asyncio。

我还复制了一个使用asyncio的telnet服务器。

两者都分开工作，但我不知道如何将两者结合在一起。据我了解，我需要从send()内的SSEHandler课程中拨打Telnet.data_received，但我不知道如何访问它。这两种服务器都是＆＃39;需要在循环中运行以接受新连接或推送数据。

任何人都可以提供帮助，或指向另一个方向吗？

import asyncio
import sse

# Get an instance of the asyncio event loop
loop = asyncio.get_event_loop()

# Setup SSE address and port
sse_host, sse_port = '192.168.2.25', 8888

class Telnet(asyncio.Protocol):
    def connection_made(self, transport):
        print("Connection received!");
        self.transport = transport

    def data_received(self, data):
        print(data)
        self.transport.write(b'echo:')
        self.transport.write(data)

        # This is where I want to send data via SSE
        # SSEHandler.send(data)

        # Things I've tried :(
        #loop.call_soon_threadsafe(SSEHandler.handle_request());
        #loop.call_soon_threadsafe(sse_server.send("PAH!"));

    def connection_lost(self, esc):
        print("Connection lost!")
        telnet_server.close()

class SSEHandler(sse.Handler):
    @asyncio.coroutine
    def handle_request(self):
        self.send('Working')

# SSE server
sse_server = sse.serve(SSEHandler, sse_host, sse_port)

# Telnet server
telnet_server = loop.run_until_complete(loop.create_server(Telnet, '192.168.2.25', 7777))

#telnet_server.something = sse_server;

loop.run_until_complete(sse_server)
loop.run_until_complete(telnet_server.wait_closed())

Answer 1

服务器端事件是一种http协议;并且在任何给定时刻您可能有任意数量的并发http请求，如果没有人连接，您可能会为零，或者数十个。这个细微差别全部包含在两个sse.serve和sse.Handler结构中;前者代表一个侦听端口，它将每个单独的客户端请求分派给后者。

此外，每个客户端都会调用sse.Handler.handle_request()一次，并且一旦该协同例程终止，客户端就会断开连接。在您的代码中，该协程会立即终止，因此客户端会看到一个＆＃34; Working＆＃34;事件。所以，我们需要等待，或多或少永远。我们可以yield from asyncio.Future()来做到这一点。

第二个问题是，我们需要以某种方式获取SSEHandler()的所有单独实例，并以某种方式对每个实例使用send()方法。好吧，我们可以让每个人在他们的handle_request()方法中自我注册;通过将每一个添加到一个dict，将各个处理程序实例映射到他们正在等待的未来。

class SSEHandler(sse.Handler):
    _instances = {}

    @asyncio.coroutine
    def handle_request(self):
        self.send('Working')
        my_future = asyncio.Future()
        SSEHandler._instances[self] = my_future
        yield from my_future

现在，要向每次收听发送一个事件，我们只需访问我们创建的dict中注册的所有SSEHandler实例，并在每个实例上使用send()。

class SSEHandler(sse.Handler):

    #...

    @classmethod
    def broadcast(cls, message):
        for instance, future in cls._instances.items():
            instance.send(message)

class Telnet(asyncio.Protocol):

    #...

    def data_received(self, data):
        #...
        SSEHandler.broadcast(data.decode('ascii'))

最后，当telnet连接关闭时，您的代码将退出。那很好，但我们当时也应该清理。幸运的是，这只是在所有处理程序的所有期货上设置结果的问题

class SSEHandler(sse.Handler):

    #...

    @classmethod
    def abort(cls):
        for instance, future in cls._instances.items():
            future.set_result(None)
        cls._instances = {}

class Telnet(asyncio.Protocol):

    #...

    def connection_lost(self, esc):
        print("Connection lost!")
        SSEHandler.abort()
        telnet_server.close()

这是一个完整的工作转储，以防我的插图不明显。

import asyncio
import sse

loop = asyncio.get_event_loop()
sse_host, sse_port = '0.0.0.0', 8888

class Telnet(asyncio.Protocol):
    def connection_made(self, transport):
        print("Connection received!");
        self.transport = transport

    def data_received(self, data):
        SSEHandler.broadcast(data.decode('ascii'))

    def connection_lost(self, esc):
        print("Connection lost!")
        SSEHandler.abort()
        telnet_server.close()

class SSEHandler(sse.Handler):
    _instances = {}
    @classmethod
    def broadcast(cls, message):
        for instance, future in cls._instances.items():
            instance.send(message)

    @classmethod
    def abort(cls):
        for instance, future in cls._instances.items():
            future.set_result(None)
        cls._instances = {}

    @asyncio.coroutine
    def handle_request(self):
        self.send('Working')
        my_future = asyncio.Future()
        SSEHandler._instances[self] = my_future
        yield from my_future

sse_server = sse.serve(SSEHandler, sse_host, sse_port)
telnet_server = loop.run_until_complete(loop.create_server(Telnet, '0.0.0.0', 7777))
loop.run_until_complete(sse_server)
loop.run_until_complete(telnet_server.wait_closed())

在asyncio协议/服务器之间进行通信

1 个答案: