mirror of
https://github.com/cosmos/cosmjs.git
synced 2025-03-11 14:09:15 +00:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
#pylint:disable=missing-docstring,invalid-name
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import asyncio
|
||
|
|
import websockets
|
||
|
|
import time
|
||
|
|
|
||
|
|
HOST = "0.0.0.0"
|
||
|
|
|
||
|
|
def log(data):
|
||
|
|
print(data, flush=True)
|
||
|
|
|
||
|
|
@asyncio.coroutine
|
||
|
|
def connection_handler(connection, path):
|
||
|
|
connection_id = hex(id(connection))
|
||
|
|
log("{} opened connection via {}".format(connection_id, path))
|
||
|
|
try:
|
||
|
|
while True:
|
||
|
|
incoming_message = yield from connection.recv()
|
||
|
|
log("< {}".format(incoming_message))
|
||
|
|
outgoing_message = incoming_message
|
||
|
|
yield from connection.send(outgoing_message)
|
||
|
|
log("> {}".format(outgoing_message))
|
||
|
|
except websockets.exceptions.ConnectionClosed:
|
||
|
|
log("{} closed connection".format(connection_id))
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("--port",
|
||
|
|
help="Port to listen on",
|
||
|
|
type=int,
|
||
|
|
default=4000)
|
||
|
|
parser.add_argument("--delay",
|
||
|
|
help="Time in seconds that a connection will be delayed before establishing it",
|
||
|
|
type=int,
|
||
|
|
default=0)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
def delaying_process_request(path, request_headers):
|
||
|
|
time.sleep(args.delay)
|
||
|
|
return None
|
||
|
|
|
||
|
|
log("Starting server at {}:{}".format(HOST, args.port))
|
||
|
|
server = websockets.serve(connection_handler, HOST, args.port, process_request=delaying_process_request)
|
||
|
|
log("Running now.")
|
||
|
|
|
||
|
|
asyncio.get_event_loop().run_until_complete(server)
|
||
|
|
asyncio.get_event_loop().run_forever()
|