Pass kwargs to grpc server/client for customization

Author: elbakramer

koapy/backend/kiwoom_open_api_plus/grpc/KiwoomOpenApiPlusServiceClient.py

@@ -25,14 +25,15 @@ def __init__(self, host=None, port=None, credentials=None, **kwargs):
         self._host = host
         self._port = port
         self._credentials = credentials
+        self._kwargs = kwargs
 
         self._target = self._host + ":" + str(self._port)
 
         if self._credentials is None:
-            self._channel = grpc.insecure_channel(self._target, **kwargs)
+            self._channel = grpc.insecure_channel(self._target, **self._kwargs)
         else:
             self._channel = grpc.secure_channel(
-                self._target, self._credentials, **kwargs
+                self._target, self._credentials, **self._kwargs
             )
 
         self._stub = KiwoomOpenApiPlusService_pb2_grpc.KiwoomOpenApiPlusServiceStub(

koapy/backend/kiwoom_open_api_plus/grpc/KiwoomOpenApiPlusServiceServer.py

@@ -15,7 +15,13 @@
 
 class KiwoomOpenApiPlusServiceServer(Logging):
     def __init__(
-        self, control, host=None, port=None, max_workers=None, credentials=None
+        self,
+        control,
+        host=None,
+        port=None,
+        max_workers=None,
+        credentials=None,
+        **kwargs,
     ):
         if host is None:
             host = config.get_string(
@@ -46,10 +52,15 @@ def __init__(
         self._port = port
         self._max_workers = max_workers
         self._credentials = credentials
+        self._kwargs = kwargs
 
-        self._address = self._host + ":" + str(self._port)
         self._servicer = KiwoomOpenApiPlusServiceServicer(self._control)
-        self._executor = futures.ThreadPoolExecutor(max_workers=self._max_workers)
+        self._address = self._host + ":" + str(self._port)
+
+        if "thread_pool" in self._kwargs:
+            self._executor = self._kwargs.pop("thread_pool")
+        else:
+            self._executor = futures.ThreadPoolExecutor(max_workers=self._max_workers)
 
         self._server = None
         self._server_started = False
@@ -67,7 +78,7 @@ def reinitialize_server(self):
             self.stop()
             self.wait_for_termination()
 
-        self._server = grpc.server(self._executor)
+        self._server = grpc.server(self._executor, **self._kwargs)
         self._server_started = False
         self._server_stopped = False