龙卷风：不接受连接

Question

这是基本的web套接字测试:

import tornado.httpserver
import tornado.websocket
import tornado.ioloop
import tornado.web
class WSHandler(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'new connection'
        self.write_message("Hello World")
    def on_message(self, message):
        print 'message received %s' % message
    def on_close(self):
      print 'connection closed'

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

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

我想处理许多连接，对于一些连接，我想接受而另一些拒绝。

有什么简洁的方法吗?

Answer 1

WebSocketHandler实例具有request属性。

因此，我会检查open方法中的self.request，以决定是否立即关闭连接，发送错误消息，或将连接添加到稍后处理的列表中。

类似:

class WSHandler(tornado.websocket.WebSocketHandler):
    connections = []
    def open(self):
        if self.request.remote_ip == '127.0.0.1':
            self.write_message('go away home')
            self.close()
        else:
            self.write_message('welcome')
            self.connections.append(self)

您可以通过阅读龙卷风源Luke查看所有可用于检查的属性。

如果您想要的是基于密码接受或拒绝连接，那么您需要将该逻辑添加到on_message方法中。您还需要在连接上维护一些状态。就像这样。

class WSHandler(tornado.websocket.WebSocketHandler):
    users = {'dave': 'password1234'}
    def open(self):
        self.authed = False
    def on_message(self, message):
        if self.authed:
            pass # handle already authed messages here
        else:
            # authenticate a connection here
            username, password = message.split(':')
            if self.users.get(username) == password:
                self.authed = True
                self.write_message('Welcome back %s.' % (username,))
            else:
                self.write_message('Auth failed')

请记住，这不是一个密码协议，我实际上会使用。这只是一个示例，用于展示如何创建一个。