confluent-kafka: fix incorrect number of arguments

Author: papigers

instrumentation/opentelemetry-instrumentation-confluent-kafka/src/opentelemetry/instrumentation/confluent_kafka/__init__.py

@@ -132,8 +132,8 @@ def produce(self, topic, value=None, *args, **kwargs):  # pylint: disable=keywor
 
 
 class AutoInstrumentedConsumer(Consumer):
-    def __init__(self, config):
-        super().__init__(config)
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
         self._current_consume_span = None
 
     # This method is deliberately implemented in order to allow wrapt to wrap this function
@@ -183,9 +183,9 @@ def __init__(self, consumer: Consumer, tracer: Tracer):
         self._current_consume_span = None
         self._current_context_token = None
 
-    def close(self):
+    def close(self, *args, **kwargs):
         return ConfluentKafkaInstrumentor.wrap_close(
-            self._consumer.close, self
+            self._consumer.close, self, args, kwargs
         )
 
     def committed(self, partitions, timeout=-1):
@@ -306,8 +306,10 @@ def _inner_wrap_consume(func, instance, args, kwargs):
                 func, instance, self._tracer, args, kwargs
             )
 
-        def _inner_wrap_close(func, instance):
-            return ConfluentKafkaInstrumentor.wrap_close(func, instance)
+        def _inner_wrap_close(func, instance, args, kwargs):
+            return ConfluentKafkaInstrumentor.wrap_close(
+                func, instance, args, kwargs
+            )
 
         wrapt.wrap_function_wrapper(
             AutoInstrumentedProducer,
@@ -419,7 +421,7 @@ def wrap_consume(func, instance, tracer, args, kwargs):
         return records
 
     @staticmethod
-    def wrap_close(func, instance):
+    def wrap_close(func, instance, args, kwargs):
         if instance._current_consume_span:
             _end_current_consume_span(instance)
-        func()
+        func(*args, **kwargs)