added socket.io server registry

Author: eadwinCode

ellar/socket_io/decorators/subscribe_message.py

@@ -1,5 +1,6 @@
 import typing as t
 
+from ellar.helper import get_name
 from ellar.reflect import reflect
 from ellar.socket_io.constants import MESSAGE_MAPPING_METADATA, MESSAGE_METADATA
 
@@ -10,4 +11,9 @@ def _decorator(func: t.Callable) -> t.Callable:
         reflect.define_metadata(MESSAGE_METADATA, message, func)
         return func
 
+    if callable(message):
+        func = message
+        message = get_name(func)
+        return _decorator(t.cast(t.Callable, func))
+
     return _decorator

ellar/socket_io/factory.py

@@ -18,6 +18,8 @@
 from ellar.socket_io.gateway import SocketMessageOperation, SocketOperationConnection
 from ellar.socket_io.model import GatewayBase, GatewayType
 
+_socket_servers: t.Dict[str, socketio.AsyncServer] = {}
+
 
 class GatewayRouterFactory(RouterBuilder, controller_type=type(GatewayBase)):
     @classmethod
@@ -35,7 +37,11 @@ def build(cls, controller_type: t.Union[t.Type[GatewayBase], t.Any]) -> "Mount":
             reflect.get_metadata(GATEWAY_MESSAGE_HANDLER_KEY, controller_type) or []
         )
 
-        socket_server = socketio.AsyncServer(**options)
+        socket_server = _socket_servers.get(path)
+
+        if not socket_server:
+            socket_server = socketio.AsyncServer(**options)
+            _socket_servers.update({path: socket_server})
 
         for handler in message_handlers:
             is_connection_handler = reflect.get_metadata(CONNECTION_EVENT, handler)