Merge pull request #21 from pyropy/add-async-transport-mode

Add async transport mode
pyropy · Jun 17, 2021 · 0a511cc · 0a511cc
2 parents c97f4cb + fa4280e
commit 0a511cc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/fastapi_socketio/socket_manager.py b/fastapi_socketio/socket_manager.py
@@ -2,7 +2,6 @@
 from typing import Optional, Union
 from fastapi import FastAPI
 
-
 class SocketManager:
     """
     Integrates SocketIO with FastAPI app. 
@@ -23,9 +22,10 @@ def __init__(
         mount_location: str = "/ws",
         socketio_path: str = "socket.io",
         cors_allowed_origins: Union[str, list] = '*',
+        async_mode: str = "asgi"
     ) -> None:
         # TODO: Change Cors policy based on fastapi cors Middleware
-        self._sio = socketio.AsyncServer(async_mode="asgi", cors_allowed_origins=cors_allowed_origins)
+        self._sio = socketio.AsyncServer(async_mode=async_mode, cors_allowed_origins=cors_allowed_origins)
         self._app = socketio.ASGIApp(
             socketio_server=self._sio, socketio_path=socketio_path
         )

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 import os
 
-VERSION = "0.0.7"
+VERSION = "0.0.8"
 
 
 def get_long_description():