Added KV instrumentation.

Author: s3rius

python/natsrpy/instrumentation/__init__.py

@@ -40,6 +40,7 @@ async def main() -> None:
 
 from .js_consumer import JSConsumerInstrumentation
 from .js_publish import JSPublishInstrumentation
+from .kv import KVInstrumentation
 from .nats_core import NatsCoreInstrumentator
 
 try:
@@ -105,8 +106,13 @@ def _instrument(self, **kwargs: Any) -> None:
             capture_body=capture_body,
             capture_headers=capture_headers,
         ).instrument()
+        KVInstrumentation(
+            tracer,
+            capture_body=capture_body,
+        ).instrument()
 
     def _uninstrument(self, **kwargs: Any) -> None:
         NatsCoreInstrumentator.uninstrument()
         JSConsumerInstrumentation.uninstrument()
         JSPublishInstrumentation.uninstrument()
+        KVInstrumentation.uninstrument()