aboutsummaryrefslogtreecommitdiff
path: root/backend/microservice/api/ml_socket.py
diff options
context:
space:
mode:
authorDanijel Andjelkovic <adanijel99@gmail.com>2022-03-29 12:27:29 +0200
committerDanijel Andjelkovic <adanijel99@gmail.com>2022-03-29 12:27:29 +0200
commit2bae49ecb411d13d1d9d78e764340468f7f49f44 (patch)
tree3b7907e172438775291954fb5a8b54541059acc6 /backend/microservice/api/ml_socket.py
parent9f3d674413affc18377c8f2564bb660270ffe7f5 (diff)
parent98d580ca1ba501bc059ff417dd1794e0f6a68407 (diff)
Merge branch 'trening' of http://gitlab.pmf.kg.ac.rs/igrannonica/neuronstellar into dev
# Conflicts: # frontend/src/app/_services/models.service.ts
Diffstat (limited to 'backend/microservice/api/ml_socket.py')
-rw-r--r--backend/microservice/api/ml_socket.py28
1 files changed, 28 insertions, 0 deletions
diff --git a/backend/microservice/api/ml_socket.py b/backend/microservice/api/ml_socket.py
new file mode 100644
index 00000000..65dd7321
--- /dev/null
+++ b/backend/microservice/api/ml_socket.py
@@ -0,0 +1,28 @@
+import asyncio
+import websockets
+import json
+
+def get_or_create_eventloop():
+ try:
+ return asyncio.get_event_loop()
+ except RuntimeError as ex:
+ if "There is no current event loop in thread" in str(ex):
+ loop = asyncio.new_event_loop()
+ asyncio.set_event_loop(loop)
+ return asyncio.get_event_loop()
+
+# create handler for each connection
+async def handler(websocket, path):
+ #data = json.loads(await websocket.recv())
+ #print(data['test'])
+ msg = await websocket.recv()
+ print(msg)
+
+async def start():
+ start_server = websockets.serve(handler, "localhost", 5027)
+ print('Websocket starting...')
+ get_or_create_eventloop().run_until_complete(start_server)
+ get_or_create_eventloop().run_forever()
+
+async def send(msg):
+ await websocket.send(msg) \ No newline at end of file