feat(server): add GRPC_WORKER_MODE config to change worker mode

sea/app.py

@@ -30,6 +30,7 @@ class BaseApp:
             "TIMEZONE": "UTC",
             "GRPC_WORKERS": 4,
             "GRPC_THREADS": 1,  # Only appliable in multiprocessing server
+            "GRPC_WORKER_MODE": "threading",  # Worker mode. threading|multiprocessing
             "GRPC_HOST": "0.0.0.0",
             "GRPC_PORT": 6000,
             "GRPC_LOG_LEVEL": "WARNING",

sea/cmds.py

@@ -2,12 +2,26 @@
 
 from sea import current_app
 from sea.cli import JobException, jobm
-from sea.server import Server
 
 
 @jobm.job("server", aliases=["s"], help="Run Server")
-def server():
-    s = Server(current_app)
+@jobm.option(
+    "-M",
+    "--worker_mode",
+    required=False,
+    action="store",
+    help="Worker mode. threading|multiprocessing",
+)
+def server(worker_mode):
+    worker_mode = worker_mode or current_app.config["GRPC_WORKER_MODE"]
+    if worker_mode == "threading":
+        from sea.server.threading import Server
+
+        s = Server(current_app)
+    else:
+        from sea.server.multiprocessing import Server
+
+        s = Server(current_app)
     s.run()
     return 0
 

tests/test_cmds.py

@@ -1,15 +1,21 @@
-import sys
-import pytest
 import os
 import shutil
+import sys
 from unittest import mock
 
+import pytest
+
 from sea import cli
 
 
 def test_cmd_server(app):
     sys.argv = "sea s".split()
-    with mock.patch("sea.cmds.Server", autospec=True) as mocked:
+    with mock.patch("sea.server.threading.Server", autospec=True) as mocked:
+        assert cli.main() == 0
+        mocked.return_value.run.assert_called_with()
+
+    sys.argv = "sea s -M multiprocessing".split()
+    with mock.patch("sea.server.multiprocessing.Server", autospec=True) as mocked:
         assert cli.main() == 0
         mocked.return_value.run.assert_called_with()