Added websockets server

Author: jscriptcoder

package.json

@@ -13,8 +13,11 @@
     "fastapi-websocket": "python src/servers/fastapi-websocket_server.py --port 5002",
     "socketio-flask-socketio": "PORT=5003 SERVER=flask-socketio babel-node src/clients/socketio_client.js",
     "flask-socketio": "python src/servers/flask-socketio_server.py --port 5003",
+    "ws-eventlet": "PORT=5004 SERVER=eventlet babel-node src/clients/ws_client.js",
     "websocket-eventlet": "PORT=5004 SERVER=eventlet babel-node src/clients/websocket_client.js",
-    "eventlet": "python src/servers/eventlet_server.py --port 5004"
+    "eventlet": "python src/servers/eventlet_server.py --port 5004",
+    "websocket-websockets": "PORT=5005 SERVER=websockets babel-node src/clients/websocket_client.js",
+    "websockets": "python src/servers/websockets_server.py --port 5005"
   },
   "repository": {
     "type": "git",

src/clients/websocket_client.js

@@ -6,7 +6,7 @@ const port = process.env.PORT || 5000
 const server = process.env.SERVER || 'unknown'
 const wsApi = `ws://${host}:${port}/hello`
 
-let iters = 1
+let iters = 10000
 
 async function runTest() {
     console.log(`websocket client <===> ${server} server on ws://${host}:${port}/hello`)

src/clients/ws_client.js

@@ -0,0 +1,51 @@
+const { PerformanceObserver, performance } = require('perf_hooks')
+const WebSocket = require('ws')
+
+const host = process.env.HOST || '0.0.0.0'
+const port = process.env.PORT || 5000
+const server = process.env.SERVER || 'unknown'
+const wsApi = `ws://${host}:${port}/hello`
+
+let iters = 10000
+
+async function runTest() {
+    console.log(`ws client <===> ${server} server on ws://${host}:${port}/hello`)
+    
+    const ws = new WebSocket(wsApi)
+
+    function requestHello() {
+        ws.send(JSON.stringify({ name: 'Fran' }))
+    }
+
+    ws.on('open', () => {
+        console.log(`Running test with ${iters} iterations...`)
+
+        ws.on('message', message => {
+            const data = JSON.parse(message)
+            const { hello } = data
+    
+            if (--iters > 0) {
+                requestHello()
+            } else {
+                performance.mark('END')
+                performance.measure('START to END', 'START', 'END')
+
+                ws.terminate()
+            }
+        })
+
+        performance.mark('START')
+
+        requestHello()
+    })
+}
+
+const obs = new PerformanceObserver(items => {
+    console.log('End test')
+    console.log(`Duration: ${items.getEntries()[0].duration}`)
+    performance.clearMarks()
+})
+
+obs.observe({ entryTypes: ['measure'] })
+
+runTest()

src/servers/websockets_server.py

@@ -0,0 +1,30 @@
+import asyncio
+import websockets
+import json
+import argparse
+import logging
+
+logger = logging.getLogger('websockets.server')
+logger.setLevel(logging.ERROR)
+
+async def hello(websocket, path):
+    if path == '/hello':
+        while True:
+            try:
+                message = await websocket.recv()
+                data = json.loads(message)
+                await websocket.send(json.dumps({ 'hello': data['name'] }))
+            except:
+                await websocket.close()
+                break
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-l', '--host', default='0.0.0.0')
+    parser.add_argument('-p', '--port', default=5005, type=int)
+
+    args = parser.parse_args()
+    print('Server starting at: ' + 'ws://{}:{}'.format(args.host, args.port))
+    start_server = websockets.serve(hello, host=args.host, port=args.port)
+    asyncio.get_event_loop().run_until_complete(start_server)
+    asyncio.get_event_loop().run_forever()