Как запускать функции за пределами цикла websocket в python (торнадо)

Я пытаюсь создать небольшой пример публичного потока Twitter через websockets. Это мой websocket.py, и он работает.

Мне интересно: как я могу взаимодействовать с веб-сайтом из "снаружи" класса WSHandler (т.е. не только отвечать при получении сообщения от websocket.js)? Скажем, я хочу запустить еще одну функцию внутри этого же script, которая будет публиковать "привет!". каждые пять секунд и отправлять их в websocket (браузер) без какого-либо взаимодействия с клиентской стороны. Как я могу это сделать?

Итак, это, как мне кажется, фундаментальный начинающий вопрос о том, как бороться с классами, как показано ниже. Любые указатели в любом направлении были бы очень признательны!

import os.path
import tornado.httpserver
import tornado.websocket
import tornado.ioloop
import tornado.web

# websocket
class FaviconHandler(tornado.web.RequestHandler):
    def get(self):
        self.redirect('/static/favicon.ico')

class WebHandler(tornado.web.RequestHandler):
    def get(self):
        self.render("websockets.html")

class WSHandler(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'new connection'
        self.write_message("Hi, client: connection is made ...")

    def on_message(self, message):
        print 'message received: \"%s\"' % message
        self.write_message("Echo: \"" + message + "\"")
        if (message == "green"):
            self.write_message("green!")

    def on_close(self):
        print 'connection closed'



handlers = [
    (r"/favicon.ico", FaviconHandler),
    (r'/static/(.*)', tornado.web.StaticFileHandler, {'path': 'static'}),
    (r'/', WebHandler),
    (r'/ws', WSHandler),
]

settings = dict(
    template_path=os.path.join(os.path.dirname(__file__), "static"),
)

application = tornado.web.Application(handlers, **settings)

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

Ответ 1

Вы можете вызвать

IOLoop.add_timeout(deadline, callback)

который вызывает обратный вызов в указанный тайм-аут таймаута (один снимок, но вы можете перенести его), или используйте

tornado.ioloop.PeriodicCallback, если у вас есть более периодическая задача.

Смотрите: http://www.tornadoweb.org/en/stable/ioloop.html#tornado.ioloop.IOLoop.add_timeout

Обновление: пример

import datetime

def test():
    print "scheduled event fired"
...

if __name__ == "__main__":
    http_server = tornado.httpserver.HTTPServer(application)
    http_server.listen(8888)
    main_loop = tornado.ioloop.IOLoop.instance()
    # Schedule event (5 seconds from now)
    main_loop.add_timeout(datetime.timedelta(seconds=5), test)
    # Start main loop
    main_loop.start()

он вызывает test() через 5 секунд.

Обновление 2:

import os.path
import tornado.httpserver
import tornado.websocket
import tornado.ioloop
import tornado.web

# websocket
class FaviconHandler(tornado.web.RequestHandler):
    def get(self):
        self.redirect('/static/favicon.ico')

class WebHandler(tornado.web.RequestHandler):
    def get(self):
        self.render("websockets.html")

class WSHandler(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'new connection'
        self.write_message("Hi, client: connection is made ...")
        tornado.ioloop.IOLoop.instance().add_timeout(datetime.timedelta(seconds=5), self.test)

    def on_message(self, message):
        print 'message received: \"%s\"' % message
        self.write_message("Echo: \"" + message + "\"")
        if (message == "green"):
            self.write_message("green!")

    def on_close(self):
        print 'connection closed'

    def test(self):
        self.write_message("scheduled!")

handlers = [
    (r"/favicon.ico", FaviconHandler),
    (r'/static/(.*)', tornado.web.StaticFileHandler, {'path': 'static'}),
    (r'/', WebHandler),
    (r'/ws', WSHandler),
]

settings = dict(
    template_path=os.path.join(os.path.dirname(__file__), "static"),
)

application = tornado.web.Application(handlers, **settings)

import datetime

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

Ответ 2

Я наткнулся на подобную проблему. Вот мое решение. Надеюсь, это будет полезно для кого-то там.

wss = []
class wsHandler(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'Online'
        if self not in wss:
            wss.append(self)

    def on_close(self):
        print 'Offline'
        if self in wss:
            wss.remove(self)

def wsSend(message):
    for ws in wss:
        ws.write_message(message)

Чтобы отправить сообщение в свои веб-сайты, просто используйте это:

wsSend(message)

wsSend update

Я получаю исключения из wsSend время от времени. Чтобы исправить это, я немного изменил код:

def wsSend(message):
    for ws in wss:
        if not ws.ws_connection.stream.socket:
            print "Web socket does not exist anymore!!!"
            wss.remove(ws)
        else:
            ws.write_message(message)

Ответ 3

Один из способов сделать это - использовать модуль pub-sub.

Предполагая, что ваши соединения подписываются на него, и вместо того, чтобы устанавливать тайм-ауты для каждого отдельного соединения, вы просто устанавливаете один таймаут для публикации после указанного периода времени.

Вероятно, один из самых реализованных - redis. Существуют также некоторые модули специально для торнадо: toredis или brükva например.

Конечно, это может быть не обязательно для простой страницы, но очень хорошо масштабируется, а также очень удобно поддерживать/продлевать, как только вы ее настроили.