manual instrumentation

biothings · Dec 23, 2023 · 635226e · 635226e
1 parent a6369fe
commit 635226e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/controllers/threading/taskHandler.js b/src/controllers/threading/taskHandler.js
@@ -14,6 +14,7 @@ const { getQueryQueue } = require("../async/asyncquery_queue");
 const Sentry = require("@sentry/node");
 const { ProfilingIntegration } = require("@sentry/profiling-node");
 const opentelemetry = require('@opentelemetry/api');
+const { Telemetry } = require('@biothings-explorer/utils');
 
 // use SENTRY_DSN environment variable
 try {
@@ -82,6 +83,7 @@ const runTask = async ({ req, route, port, job: { jobId, queueName } = {} }) =>
 
     span = opentelemetry.trace.getTracer('biothings-explorer-thread').startSpan(routeNames[route])
     span.setAttribute("request", req.data.queryGraph);
+    Telemetry.setOtelContext(opentelemetry.trace.setSpan(opentelemetry.context.active(), span));
   } catch (error) {
     debug("Sentry/OpenTelemetry transaction start error. This does not affect execution.");
     debug(error);
@@ -93,6 +95,7 @@ const runTask = async ({ req, route, port, job: { jobId, queueName } = {} }) =>
   try {
     transaction.finish();
     span.end();
+    Telemetry.removeOtelContext();
   } catch (error) {
     debug("Sentry/OpenTelemetry transaction finish error. This does not affect execution.");
     debug(error);