3

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

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

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

两者分开工作，但我不知道如何将两者结合在一起。据我了解，我需要从里面调用类send()，但我不知道如何访问它。这两个“服务器”都需要循环运行以接受新连接或推送数据。SSEHandlerTelnet.data_received

任何人都可以帮忙，或指出我另一个方向吗？

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())
4

1 回答 1

7

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

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

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

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

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

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())
于 2014-09-02T15:07:25.880 回答