Как запускать функции за пределами цикла 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 например.
Конечно, это может быть не обязательно для простой страницы, но очень хорошо масштабируется, а также очень удобно поддерживать/продлевать, как только вы ее настроили.