Tornado：不接受连接

Question

这是一个基本的网页套接字测试：

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)

你可以通过 查看 tornado 的源代码，来了解所有可以检查的属性。

如果你想根据密码来接受或拒绝连接，那么你需要把这个逻辑加到 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')

请注意，这并不是我实际会使用的密码协议。这只是一个示例，展示你可能如何创建一个。