pulled out operation logic

Author: daniel-sanche

google/cloud/bigtable/data/_async/metrics_interceptor.py

@@ -34,32 +34,6 @@ class AsyncBigtableMetricsInterceptor(
     An async gRPC interceptor to add client metadata and print server metadata.
     """
 
-    def __init__(self):
-        super().__init__()
-        self.operation_map = {}
-
-    def register_operation(self, operation):
-        """
-        Register an operation object to be tracked my the interceptor
-
-        When registered, the operation will receive metadata updates:
-        - start_attempt if attempt not started when rpc is being sent
-        - add_response_metadata after call is complete
-        - end_attempt_with_status if attempt receives an error
-
-        The interceptor will register itself as a handeler for the operation,
-        so it can unregister the operation when it is complete
-        """
-        self.operation_map[operation.uuid] = operation
-        operation.handlers.append(self)
-
-    def on_operation_complete(self, op):
-        if op.uuid in self.operation_map:
-            del self.operation_map[op.uuid]
-
-    def on_operation_cancelled(self, op):
-        self.on_operation_complete(op)
-
     @CrossSync.convert
     async def intercept_unary_unary(self, continuation, client_call_details, request):
         try:

google/cloud/bigtable/data/_sync_autogen/metrics_interceptor.py

@@ -26,30 +26,6 @@ class BigtableMetricsInterceptor(
     An async gRPC interceptor to add client metadata and print server metadata.
     """
 
-    def __init__(self):
-        super().__init__()
-        self.operation_map = {}
-
-    def register_operation(self, operation):
-        """Register an operation object to be tracked my the interceptor
-
-        When registered, the operation will receive metadata updates:
-        - start_attempt if attempt not started when rpc is being sent
-        - add_response_metadata after call is complete
-        - end_attempt_with_status if attempt receives an error
-
-        The interceptor will register itself as a handeler for the operation,
-        so it can unregister the operation when it is complete"""
-        self.operation_map[operation.uuid] = operation
-        operation.handlers.append(self)
-
-    def on_operation_complete(self, op):
-        if op.uuid in self.operation_map:
-            del self.operation_map[op.uuid]
-
-    def on_operation_cancelled(self, op):
-        self.on_operation_complete(op)
-
     def intercept_unary_unary(self, continuation, client_call_details, request):
         try:
             call = continuation(client_call_details, request)