从Tornado WebSocket服务器主动发送消息

Question

我有一个Python WebSocket服务器。当它接收到一条消息时，它可以返回响应。

import tornado.web
import tornado.websocket
import tornado.ioloop

class WebSocketHandler(tornado.websocket.WebSocketHandler):

    def open(self):
        print("New client connected")
        self.write_message("You are connected")

    def on_message(self, message):
        self.write_message(u"You said: " + message)

    def on_close(self):
        print("Client disconnected")

    def check_origin(self, origin):
        return True

application = tornado.web.Application([
        (r"/", WebSocketHandler),
])


if __name__ == "__main__":
    application.listen(8888)
    tornado.ioloop.IOLoop.instance().start()

然而，在 WebSocket 接收到信息前无法主动发送信息。那么，如何主动发送信息呢？例如，计时器若 10 秒内未接收到信息，则主动发送 "Are you sleeping?"。

我想使用 WebSocket 制作聊天机器人。我选用 tornado 和 websocket 是因为这是我了解的方法，如果您知道更好的方法，请告诉我。

- user8483278

Answer 1

您可以使用tornado.ioloop.PeriodicCallback添加一个计划，并根据时间简单地检查是否有活动或无活动，如下所示（我改编自您的代码和此答案）：

import tornado.web
import tornado.websocket
import tornado.ioloop
import time

class WebSocketHandler(tornado.websocket.WebSocketHandler):
    def simple_init(self):
        self.last = time.time()
        self.stop = False

    def open(self):
        self.simple_init()
        print("New client connected")
        self.write_message("You are connected")
        self.loop = tornado.ioloop.PeriodicCallback(self.check_ten_seconds, 1000, io_loop=tornado.ioloop.IOLoop.instance())
        self.loop.start()

    def on_message(self, message):
        self.write_message(u"You said: " + message)
        self.last = time.time()

    def on_close(self):
        print("Client disconnected")
        self.loop.stop()

    def check_origin(self, origin):
        return True

    def check_ten_seconds(self):
        print("Just checking")
        if (time.time() - self.last > 10):
            self.write_message("You sleeping mate?")
            self.last = time.time()

现在有一个客户端交替处于闲置和写入状态（改编自这里）：

class Client(object):
    def __init__(self, url, timeout):
        self.url = url
        self.timeout = timeout
        self.ioloop = IOLoop.instance()
        self.ws = None
        self.connect()
        PeriodicCallback(self.keep_alive, 20000, io_loop=self.ioloop).start()
        self.ioloop.start()

    @gen.coroutine
    def connect(self):
        print("trying to connect")
        try:
            self.ws = yield websocket_connect(self.url)
        except Exception as e:
            print("connection error")
        else:
            print("connected")
            self.run()

    @gen.coroutine
    def run(self):
        once = False
        while True:
            msg = yield self.ws.read_message()
            print(msg)
            if once:
                time.sleep(11)
                once = False
            else:
                time.sleep(1)
                once = True
            self.ws.write_message("Hello matey")
            if msg is None:
                print("connection closed")
                self.ws = None
                break

    def keep_alive(self):
        if self.ws is None:
            self.connect()
        else:
            self.ws.write_message("keep alive")

我们得到了预期的输出：

trying to connect
connected
You are connected
You said: Hello matey
You sleeping mate?
You said: Hello matey
You said: Hello matey
You sleeping mate?

- Adonis