python Tornado websockets如何每X秒发送一次消息?

时间:2013-07-04 09:35:31

标签: python websocket tornado

我正在尝试拼凑一个允许websockets客户端连接到Tornado服务器的测试,我希望Tornado服务器每X秒向所有客户端发送一条消息。

我这样做的原因是因为wbesockets连接被静默地丢弃到某处,我想知道websockets服务器发送的定期“ping”将保持连接。

我担心这是一个相当愚蠢的问题,下面的代码相当混乱。我只是没有把头包裹在龙卷风和足够的范围以使其工作。

import tornado.httpserver
import tornado.websocket
import tornado.ioloop
import tornado.web
import tornado.gen
import time
from tornado import gen

class WSHandler(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'http://mailapp.crowdwave.com/girlthumb.jpg'
        self.write_message("http://www.example.com/girlthumb.jpg")

    def on_message(self, message):
        print 'Incoming message:', message
        self.write_message("http://www.example.com/girlthumb.jpg")


    def on_close(self):
        print 'Connection was closed...'

@gen.engine
def f():
    yield gen.Task(tornado.ioloop.IOLoop.instance().add_timeout, time.time() + 8)
    self.write_message("http://www.example.com/x.png")
    print 'x'

@gen.engine
def g():
     yield gen.Task(tornado.ioloop.IOLoop.instance().add_timeout, time.time() + 4)
     self.write_message("http://www.example.com/y.jpg")
     print 'y'

application = tornado.web.Application([
    (r'/ws', WSHandler),
    ])

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

2 个答案:

答案 0 :(得分:9)

为什么不尝试在其中编写调度程序? :)

def schedule_func():
    #DO SOMETHING#

#milliseconds
interval_ms = 15
main_loop = tornado.ioloop.IOLoop.instance()
sched = tornado.ioloop.PeriodicCallback(schedule_func,interval_ms, io_loop = main_loop)
#start your period timer
sched.start()
#start your loop
main_loop.start()

答案 1 :(得分:3)

发现接受的答案几乎就是我想要的:

How to run functions outside websocket loop in python (tornado)

稍作修改后,上述链接中接受的答案会不断发送ping消息。这是mod:

变化:

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

为:

def test(self):
    self.write_message("scheduled!")
    tornado.ioloop.IOLoop.instance().add_timeout(datetime.timedelta(seconds=5), self.test)