opentelemetry-instrumentation-confluent-kafka 0.46b0__py3-none-any.whl → 0.47b0__py3-none-any.whl

opentelemetry/instrumentation/confluent_kafka/__init__.py

@@ -144,6 +144,10 @@ class AutoInstrumentedConsumer(Consumer):
     ):  # pylint: disable=useless-super-delegation
         return super().consume(*args, **kwargs)
 
+    # This method is deliberately implemented in order to allow wrapt to wrap this function
+    def close(self):  # pylint: disable=useless-super-delegation
+        return super().close()
+
 
 class ProxiedProducer(Producer):
     def __init__(self, producer: Producer, tracer: Tracer):
@@ -156,6 +160,9 @@ class ProxiedProducer(Producer):
     def poll(self, timeout=-1):
         return self._producer.poll(timeout)
 
+    def purge(self, in_queue=True, in_flight=True, blocking=True):
+        self._producer.purge(in_queue, in_flight, blocking)
+
     def produce(
         self, topic, value=None, *args, **kwargs
     ):  # pylint: disable=keyword-arg-before-vararg
@@ -178,6 +185,11 @@ class ProxiedConsumer(Consumer):
         self._current_consume_span = None
         self._current_context_token = None
 
+    def close(self):
+        return ConfluentKafkaInstrumentor.wrap_close(
+            self._consumer.close, self
+        )
+
     def committed(self, partitions, timeout=-1):
         return self._consumer.committed(partitions, timeout)
 
@@ -300,6 +312,9 @@ class ConfluentKafkaInstrumentor(BaseInstrumentor):
                 func, instance, self._tracer, args, kwargs
             )
 
+        def _inner_wrap_close(func, instance):
+            return ConfluentKafkaInstrumentor.wrap_close(func, instance)
+
         wrapt.wrap_function_wrapper(
             AutoInstrumentedProducer,
             "produce",
@@ -318,6 +333,12 @@ class ConfluentKafkaInstrumentor(BaseInstrumentor):
             _inner_wrap_consume,
         )
 
+        wrapt.wrap_function_wrapper(
+            AutoInstrumentedConsumer,
+            "close",
+            _inner_wrap_close,
+        )
+
     def _uninstrument(self, **kwargs):
         confluent_kafka.Producer = self._original_kafka_producer
         confluent_kafka.Consumer = self._original_kafka_consumer
@@ -400,3 +421,9 @@ class ConfluentKafkaInstrumentor(BaseInstrumentor):
         )
 
         return records
+
+    @staticmethod
+    def wrap_close(func, instance):
+        if instance._current_consume_span:
+            _end_current_consume_span(instance)
+        func()

opentelemetry/instrumentation/confluent_kafka/package.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 
 
-_instruments = ("confluent-kafka >= 1.8.2, <= 2.3.0",)
+_instruments = ("confluent-kafka >= 1.8.2, <= 2.4.0",)

opentelemetry/instrumentation/confluent_kafka/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.46b0"
+__version__ = "0.47b0"

{opentelemetry_instrumentation_confluent_kafka-0.46b0.dist-info → opentelemetry_instrumentation_confluent_kafka-0.47b0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: opentelemetry-instrumentation-confluent-kafka
-Version: 0.46b0
+Version: 0.47b0
 Summary: OpenTelemetry Confluent Kafka instrumentation
 Project-URL: Homepage, https://github.com/open-telemetry/opentelemetry-python-contrib/tree/main/instrumentation/opentelemetry-instrumentation-confluent-kafka
 Author-email: OpenTelemetry Authors <cncf-opentelemetry-contributors@lists.cncf.io>
@@ -12,12 +12,16 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.8
 Requires-Dist: opentelemetry-api~=1.12
-Requires-Dist: opentelemetry-instrumentation==0.46b0
+Requires-Dist: opentelemetry-instrumentation==0.47b0
 Requires-Dist: wrapt<2.0.0,>=1.0.0
 Provides-Extra: instruments
-Requires-Dist: confluent-kafka<=2.3.0,>=1.8.2; extra == 'instruments'
+Requires-Dist: confluent-kafka<=2.4.0,>=1.8.2; extra == 'instruments'
 Description-Content-Type: text/x-rst
 
 OpenTelemetry confluent-kafka Instrumentation

opentelemetry_instrumentation_confluent_kafka-0.47b0.dist-info/RECORD

@@ -0,0 +1,9 @@
+opentelemetry/instrumentation/confluent_kafka/__init__.py,sha256=bMfMwSQoSINvstmUEPe5-OCzPq_gZXdtmD_7_aVTf94,14266
+opentelemetry/instrumentation/confluent_kafka/package.py,sha256=qHUkPcyPmxXSU2ztkYUBTixbJ0dzXJA_idhtkw73_RI,641
+opentelemetry/instrumentation/confluent_kafka/utils.py,sha256=2fGEUpZNKtw2oels6lEYczVuOq1_tJmRjIcaH-3L1Sw,4507
+opentelemetry/instrumentation/confluent_kafka/version.py,sha256=6FmWdr2KRW0MGpqEO5NBzLrm6URCtz_6ixXi_ALh6JA,608
+opentelemetry_instrumentation_confluent_kafka-0.47b0.dist-info/METADATA,sha256=PyjtzWC8Uzy1gHhUGw-e6o8BRSb2nqR7iY8t_mH9Apo,1908
+opentelemetry_instrumentation_confluent_kafka-0.47b0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+opentelemetry_instrumentation_confluent_kafka-0.47b0.dist-info/entry_points.txt,sha256=hAIEwx8JI8eFsUgk3ewaMdYw215At11XX-1a4WjAHBY,120
+opentelemetry_instrumentation_confluent_kafka-0.47b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+opentelemetry_instrumentation_confluent_kafka-0.47b0.dist-info/RECORD,,

{opentelemetry_instrumentation_confluent_kafka-0.46b0.dist-info → opentelemetry_instrumentation_confluent_kafka-0.47b0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.24.2
+Generator: hatchling 1.25.0
 Root-Is-Purelib: true
 Tag: py3-none-any

opentelemetry_instrumentation_confluent_kafka-0.46b0.dist-info/RECORD

@@ -1,9 +0,0 @@
-opentelemetry/instrumentation/confluent_kafka/__init__.py,sha256=pYiVUXVjGoqane994zN2A3f13_l4pUHlkAhubZY_8KQ,13404
-opentelemetry/instrumentation/confluent_kafka/package.py,sha256=uNjGCB1rkU0qOKP-2e3cck1MU2MopuPB3fJt70-jH6I,641
-opentelemetry/instrumentation/confluent_kafka/utils.py,sha256=2fGEUpZNKtw2oels6lEYczVuOq1_tJmRjIcaH-3L1Sw,4507
-opentelemetry/instrumentation/confluent_kafka/version.py,sha256=asa7q2ly73330-K1Q7vujpvZk7hfhasEZUzSFDPu1N8,608
-opentelemetry_instrumentation_confluent_kafka-0.46b0.dist-info/METADATA,sha256=Kb0f4eQpewdgKkEYV0uYNd_rmixWot-L5IqI1l4VJhk,1705
-opentelemetry_instrumentation_confluent_kafka-0.46b0.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-opentelemetry_instrumentation_confluent_kafka-0.46b0.dist-info/entry_points.txt,sha256=hAIEwx8JI8eFsUgk3ewaMdYw215At11XX-1a4WjAHBY,120
-opentelemetry_instrumentation_confluent_kafka-0.46b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-opentelemetry_instrumentation_confluent_kafka-0.46b0.dist-info/RECORD,,