133 lines
3.8 KiB
Python
133 lines
3.8 KiB
Python
"""This is a websocket chat example with many servers. A client can connect to
|
|
any of the servers and their messages will be received by all clients connected
|
|
to any of the servers.
|
|
|
|
Run the examples like this:
|
|
|
|
$ python examples/chat_bridge.py tcp://127.0.0.1:12345 tcp://127.0.0.1:12346
|
|
|
|
and the servers like this (changing the port for each one obviously):
|
|
|
|
$ python examples/distributed_websocket_chat.py -p tcp://127.0.0.1:12345 -s tcp://127.0.0.1:12346 7000
|
|
|
|
So all messages are published to port 12345 and the device forwards all the
|
|
messages to 12346 where they are subscribed to
|
|
"""
|
|
import os
|
|
import sys
|
|
import eventlet
|
|
from collections import defaultdict
|
|
from eventlet import spawn_n, sleep
|
|
from eventlet import wsgi
|
|
from eventlet import websocket
|
|
from eventlet.green import zmq
|
|
from eventlet.hubs import get_hub, use_hub
|
|
from uuid import uuid1
|
|
|
|
use_hub('zeromq')
|
|
ctx = zmq.Context()
|
|
|
|
|
|
class IDName(object):
|
|
|
|
def __init__(self):
|
|
self.id = uuid1()
|
|
self.name = None
|
|
|
|
def __str__(self):
|
|
if self.name:
|
|
return self.name
|
|
else:
|
|
return str(self.id)
|
|
|
|
def pack_message(self, msg):
|
|
return self, msg
|
|
|
|
def unpack_message(self, msg):
|
|
sender, message = msg
|
|
sender_name = 'you said' if sender.id == self.id \
|
|
else '%s says' % sender
|
|
return "%s: %s" % (sender_name, message)
|
|
|
|
|
|
participants = defaultdict(IDName)
|
|
|
|
|
|
def subscribe_and_distribute(sub_socket):
|
|
global participants
|
|
while True:
|
|
msg = sub_socket.recv_pyobj()
|
|
for ws, name_id in participants.items():
|
|
to_send = name_id.unpack_message(msg)
|
|
if to_send:
|
|
try:
|
|
ws.send(to_send)
|
|
except:
|
|
del participants[ws]
|
|
|
|
|
|
@websocket.WebSocketWSGI
|
|
def handle(ws):
|
|
global pub_socket
|
|
name_id = participants[ws]
|
|
ws.send("Connected as %s, change name with 'name: new_name'" % name_id)
|
|
try:
|
|
while True:
|
|
m = ws.wait()
|
|
if m is None:
|
|
break
|
|
if m.startswith('name:'):
|
|
old_name = str(name_id)
|
|
new_name = m.split(':', 1)[1].strip()
|
|
name_id.name = new_name
|
|
m = 'Changed name from %s' % old_name
|
|
pub_socket.send_pyobj(name_id.pack_message(m))
|
|
sleep()
|
|
finally:
|
|
del participants[ws]
|
|
|
|
|
|
def dispatch(environ, start_response):
|
|
"""Resolves to the web page or the websocket depending on the path."""
|
|
global port
|
|
if environ['PATH_INFO'] == '/chat':
|
|
return handle(environ, start_response)
|
|
else:
|
|
start_response('200 OK', [('content-type', 'text/html')])
|
|
return [open(os.path.join(
|
|
os.path.dirname(__file__),
|
|
'websocket_chat.html')).read() % dict(port=port)]
|
|
|
|
port = None
|
|
|
|
if __name__ == "__main__":
|
|
usage = 'usage: websocket_chat -p pub address -s sub address port number'
|
|
if len(sys.argv) != 6:
|
|
print(usage)
|
|
sys.exit(1)
|
|
|
|
pub_addr = sys.argv[2]
|
|
sub_addr = sys.argv[4]
|
|
try:
|
|
port = int(sys.argv[5])
|
|
except ValueError:
|
|
print("Error port supplied couldn't be converted to int\n", usage)
|
|
sys.exit(1)
|
|
|
|
try:
|
|
pub_socket = ctx.socket(zmq.PUB)
|
|
pub_socket.connect(pub_addr)
|
|
print("Publishing to %s" % pub_addr)
|
|
sub_socket = ctx.socket(zmq.SUB)
|
|
sub_socket.connect(sub_addr)
|
|
sub_socket.setsockopt(zmq.SUBSCRIBE, "")
|
|
print("Subscribing to %s" % sub_addr)
|
|
except:
|
|
print("Couldn't create sockets\n", usage)
|
|
sys.exit(1)
|
|
|
|
spawn_n(subscribe_and_distribute, sub_socket)
|
|
listener = eventlet.listen(('127.0.0.1', port))
|
|
print("\nVisit http://localhost:%s/ in your websocket-capable browser.\n" % port)
|
|
wsgi.server(listener, dispatch)
|