使用 Tornado 实现一个简单的 Websocket 服务器与客户端

robert

Tornado 是 Python 的一个 Web 服务器框架，它是非阻塞式的，非常适合用来做长连接服务器，http://www.tornadoweb.cn/。

具体文档请看链接，直接上代码了：

1. # Websocket 服务端
   
2. import tornado
   
3. import tornado.websocket
   
4. 
   
5. class EchoWebSocket(tornado.websocket.WebSocketHandler):
   
6. 
   
7.     def open(self):
   
8.         print("WebSocket opened")
   
9. 
   
10.     def on_message(self, message):
    
11.         print message
    
12.         self.write_message(u"You said: " + message)
    
13. 
    
14.     def on_close(self):
    
15.         print("WebSocket closed")
    
16. 
    
17. 
    
18. if __name__=='__main__':
    
19.     app = tornado.web.Application([
    
20.         (r'/', EchoWebSocket),
    
21.     ])
    
22.     app.listen(8000)
    
23.     tornado.ioloop.IOLoop.current().start()

复制代码

1. # Websocket 客户端
   
2. import tornado
   
3. from tornado.websocket import websocket_connect
   
4. 
   
5. 
   
6. class WSClient(object):
   
7. 
   
8.     def __init__(self, url):
   
9.         self.url = url
   
10.         self.ioloop = tornado.ioloop.IOLoop.current()
    
11. 
    
12.     def start(self):
    
13.         websocket_connect(
    
14.             self.url,
    
15.             self.ioloop,
    
16.             callback=self.on_connected,
    
17.             on_message_callback=self.on_message)
    
18.         self.ioloop.start()
    
19. 
    
20.     def on_connected(self, f):
    
21.         try:
    
22.             conn = f.result()
    
23.             conn.write_message("hello")
    
24.         except Exception, e:
    
25.             print str(e)
    
26.             self.ioloop.stop()
    
27. 
    
28.     def on_message(self, msg):
    
29.         print msg
    
30.         if msg.endswith("hello"):
    
31.             self.ioloop.stop()
    
32. 
    
33. 
    
34. if __name__=='__main__':
    
35.     wsc = WSClient('ws://localhost:8000')

复制代码

运行效果：

1. # server
   
2. $ python websocket.py
   
3. WebSocket opened
   
4. hello
   
5. WebSocket closed

复制代码

1. # client
   
2. $ python ws_client.py
   
3. You said: hello

复制代码

soso123

不明觉厉~