some cleanup and working flask example

Author: wyattwalter

example/app.py

@@ -0,0 +1,22 @@
+from arango import ArangoClient
+from random import randint
+from flask import Flask, request
+import logging
+
+app = Flask(__name__)
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+client = ArangoClient()
+db = client.db("test", username="root", password="password")
+
+@app.route("/random_doc")
+def random_doc():
+    item_id = str(randint(1, 1000))
+    doc = db.aql.execute("FOR doc IN test FILTER doc._key == @item_id RETURN doc", bind_vars={"item_id": item_id})
+
+    try:
+        result = doc.pop()
+        return {"doc": result}, 200
+    except Exception as e:
+        return {"error": str(item_id) + " not found"}, 404

example/application.py

@@ -0,0 +1,52 @@
+asgiref==3.8.1
+blinker==1.8.2
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+Deprecated==1.2.14
+Flask==3.0.3
+googleapis-common-protos==1.63.1rc0
+grpcio==1.64.0
+idna==3.7
+importlib_metadata==7.1.0
+itsdangerous==2.2.0
+Jinja2==3.1.4
+MarkupSafe==2.1.5
+opentelemetry-api==1.25.0
+opentelemetry-distro==0.46b0
+opentelemetry-exporter-otlp==1.25.0
+opentelemetry-exporter-otlp-proto-common==1.25.0
+opentelemetry-exporter-otlp-proto-grpc==1.25.0
+opentelemetry-exporter-otlp-proto-http==1.25.0
+opentelemetry-instrumentation==0.46b0
+opentelemetry-instrumentation-arangodb==0.1.0
+opentelemetry-instrumentation-asgi==0.46b0
+opentelemetry-instrumentation-asyncio==0.46b0
+opentelemetry-instrumentation-aws-lambda==0.46b0
+opentelemetry-instrumentation-dbapi==0.46b0
+opentelemetry-instrumentation-flask==0.46b0
+opentelemetry-instrumentation-jinja2==0.46b0
+opentelemetry-instrumentation-logging==0.46b0
+opentelemetry-instrumentation-requests==0.46b0
+opentelemetry-instrumentation-sqlite3==0.46b0
+opentelemetry-instrumentation-threading==0.46b0
+opentelemetry-instrumentation-urllib==0.46b0
+opentelemetry-instrumentation-urllib3==0.46b0
+opentelemetry-instrumentation-wsgi==0.46b0
+opentelemetry-propagator-aws-xray==1.0.1
+opentelemetry-proto==1.25.0
+opentelemetry-sdk==1.25.0
+opentelemetry-semantic-conventions==0.46b0
+opentelemetry-test-utils==0.46b0
+opentelemetry-util-http==0.46b0
+packaging==24.0
+protobuf==4.25.3
+PyJWT==2.8.0
+python-arango==8.0.0
+requests==2.32.3
+requests-toolbelt==1.0.0
+typing_extensions==4.12.0
+urllib3==2.2.1
+Werkzeug==3.0.3
+wrapt==1.16.0
+zipp==3.19.1

example/requirements.txt

@@ -34,11 +34,12 @@ def _get_instrumented_execute(self):
         @functools.wraps(self._original_method)
         def instrumented_execute(*args, **kwargs):
             query = args[1] if len(args) > 1 else kwargs.get('query', 'Unknown query')
+            dbname = args[0].db_name if len(args) > 0 else kwargs.get('db_name', 'Unknown db')
             with tracer.start_as_current_span("ArangoDB Execute", kind=SpanKind.CLIENT) as span:
-                span.set_attribute("db.system", "arangodb")
+                span.set_attribute("db.name", dbname)
                 span.set_attribute("db.query", query)
                 if kwargs.get('bind_vars'):
-                    span.set_attribute("db.bind_vars", kwargs.get('bind_vars'))
+                    span.set_attribute("db.bind_vars", str(kwargs.get('bind_vars')))
                 try:
                     result = self._original_method(*args, **kwargs)
                     span.set_status(StatusCode.OK)
@@ -54,7 +55,7 @@ def instrumented_execute(*args, **kwargs):
                         for key, value in result.statistics().items():
                             span.set_attribute("db." + key, value)
                     if result.warnings():
-                        span.set_attribute("db.warnings", result.warnings())
+                        span.set_attribute("db.warnings", str(result.warnings()))
                 except Exception as e:
                     logger.debug("Error setting trace attributes: %s", e)
                 return result