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属性。

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

类似的东西：

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)

您可以reading the tornado source 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')

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

龙卷风：不接受连接

1 个答案: