Added KV and Object store instrumentations. (#52)

Author: s3rius

python/natsrpy/instrumentation/__init__.py

@@ -40,7 +40,9 @@ async def main() -> None:
 
 from .js_consumer import JSConsumerInstrumentation
 from .js_publish import JSPublishInstrumentation
+from .kv import KVInstrumentation
 from .nats_core import NatsCoreInstrumentator
+from .object_store import ObjectStoreInstrumentation
 
 try:
     import opentelemetry  # noqa: F401
@@ -105,8 +107,18 @@ def _instrument(self, **kwargs: Any) -> None:
             capture_body=capture_body,
             capture_headers=capture_headers,
         ).instrument()
+        KVInstrumentation(
+            tracer,
+            capture_body=capture_body,
+        ).instrument()
+        ObjectStoreInstrumentation(
+            tracer,
+            capture_body=capture_body,
+        ).instrument()
 
     def _uninstrument(self, **kwargs: Any) -> None:
         NatsCoreInstrumentator.uninstrument()
         JSConsumerInstrumentation.uninstrument()
         JSPublishInstrumentation.uninstrument()
+        KVInstrumentation.uninstrument()
+        ObjectStoreInstrumentation.uninstrument()