feat: enable grpc reflection

shanbay · Aug 21, 2023 · 0c08e7d · 0c08e7d
1 parent 6c60a91
commit 0c08e7d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,6 @@
 grpcio>=1.27.0,<1.49.0
 grpcio-tools>=1.27.0,<1.49.0
+grpcio-reflection
 protobuf<4.0.0
 pendulum
 blinker

diff --git a/sea/server/multiprocessing.py b/sea/server/multiprocessing.py
@@ -45,6 +45,9 @@ def _run_server(self, bind_address):
             ],
         )
         self.server = server  # set server in slave process
+        if self.app.config.get("GRPC_REFLECTION", True):
+            from grpc_reflection.v1alpha import reflection
+            reflection.enable_server_reflection((reflection.SERVICE_NAME,), server)
 
         for _, (add_func, servicer) in self.app.servicers.items():
             add_func(servicer(), server)

diff --git a/sea/server/threading.py b/sea/server/threading.py
@@ -28,6 +28,9 @@ def run(self):
             from prometheus_client import start_http_server
 
             start_http_server(self.app.config["PROMETHEUS_PORT"])
+        if self.app.config.get("GRPC_REFLECTION", True):
+            from grpc_reflection.v1alpha import reflection
+            reflection.enable_server_reflection((reflection.SERVICE_NAME,), self.server)
         # run grpc server
         for name, (add_func, servicer) in self.app.servicers.items():
             add_func(servicer(), self.server)