Update util/oteltest/src/oteltest/sink/__init__.py

Co-authored-by: Riccardo Magliocchetti <[email protected]>

Author: pmcollins

util/oteltest/src/oteltest/sink/__init__.py

@@ -71,8 +71,8 @@ class GrpcSink:
     logs. It requires a RequestHandler implementation passed in.
     """
 
-    def __init__(self, request_handler: RequestHandler):
-        self.svr = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
+    def __init__(self, request_handler: RequestHandler, max_workers: int = 10, address: str = "0.0.0.0:4317"):
+        self.svr = grpc.server(futures.ThreadPoolExecutor(max_workers=max_workers))
         trace_service_pb2_grpc.add_TraceServiceServicer_to_server(
             _TraceServiceServicer(request_handler.handle_trace), self.svr
         )
@@ -82,7 +82,7 @@ def __init__(self, request_handler: RequestHandler):
         logs_service_pb2_grpc.add_LogsServiceServicer_to_server(
             _LogsServiceServicer(request_handler.handle_logs), self.svr
         )
-        self.svr.add_insecure_port("0.0.0.0:4317")
+        self.svr.add_insecure_port(address)
 
     def start(self):
         """Starts the server. Does not block."""