capture cypher

PLATER/requirements.txt

@@ -15,3 +15,4 @@ pydantic>=1.8
 opentelemetry-sdk
 opentelemetry-exporter-jaeger
 opentelemetry-instrumentation-fastapi
+opentelemetry-instrumentation-httpx

PLATER/services/server.py

@@ -1,5 +1,5 @@
 """FastAPI app."""
-import os
+import logging, warnings, os, json
 
 from fastapi import  FastAPI
 from starlette.middleware.cors import CORSMiddleware
@@ -50,6 +50,13 @@
     from opentelemetry.sdk.resources import SERVICE_NAME as telemetery_service_name_key, Resource
     from opentelemetry.sdk.trace import TracerProvider
     from opentelemetry.sdk.trace.export import BatchSpanProcessor
+    from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
+
+    # httpx connections need to be open a little longer by the otel decorators
+    # but some libs display warnings of resource being unclosed.
+    # these supresses such warnings.
+    logging.captureWarnings(capture=True)
+    warnings.filterwarnings("ignore", category=ResourceWarning)
     service_name = os.environ.get('PLATER_TITLE', 'PLATER')
     assert service_name and isinstance(service_name, str)
     trace.set_tracer_provider(
@@ -67,6 +74,22 @@
     tracer = trace.get_tracer(__name__)
     FastAPIInstrumentor.instrument_app(APP, tracer_provider=trace, excluded_urls=
                                        "docs,openapi.json") #,*cypher,*1.3/sri_testing_data")
+    async def request_hook(span, request):
+        # logs cypher queries set to neo4j
+        # check url
+        if span.attributes.get('http.url').endswith('/db/data/transaction/commit'):
+            # if url matches try to json load the query
+            try:
+                neo4j_query = json.loads(
+                    request.stream._stream.decode('utf-8')
+                )['statements'][0]['statement']
+                span.set_attribute('cypher', neo4j_query)
+            except Exception as ex:
+                logger.error(f"error logging neo4j query when sending to OTEL: {ex}")
+                neo4j_query = ""
+    HTTPXClientInstrumentor().instrument(request_hook=request_hook)
+
+
 if __name__ == '__main__':
     import uvicorn
     uvicorn.run(APP, host='0.0.0.0', port=8080)