+++ /dev/null
-"""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)