Add ability to disable tracing to test if it’s the cause of the memory leak

Author: ericallam

apps/webapp/app/env.server.ts

@@ -139,6 +139,7 @@ const EnvironmentSchema = z.object({
   // this means 1/20 traces or 5% of traces will be sampled (sampled = recorded)
   INTERNAL_OTEL_TRACE_SAMPLING_RATE: z.string().default("20"),
   INTERNAL_OTEL_TRACE_INSTRUMENT_PRISMA_ENABLED: z.string().default("0"),
+  INTERNAL_OTEL_TRACE_DISABLED: z.string().default("0"),
 });
 
 export type Environment = z.infer<typeof EnvironmentSchema>;

apps/webapp/app/v3/marqs/sharedQueueConsumer.server.ts

@@ -77,7 +77,6 @@ export class SharedQueueConsumer {
   private _currentSpan: Span | undefined;
   private _endSpanInNextIteration = false;
   private _tasks = sharedQueueTasks;
-  private _inProgressAttempts: Map<string, string> = new Map(); // Keys are task attempt friendly IDs, values are TaskRun ids/queue message ids
 
   constructor(
     private _sender: ZodMessageSender<typeof serverWebsocketMessages>,
@@ -151,38 +150,6 @@ export class SharedQueueConsumer {
     }
   }
 
-  async #cancelInProgressAttempts(reason: string) {
-    const service = new CancelAttemptService();
-
-    const cancelledAt = new Date();
-
-    const inProgressAttempts = new Map(this._inProgressAttempts);
-
-    this._inProgressAttempts.clear();
-
-    for (const [attemptId, messageId] of inProgressAttempts) {
-      await this.#cancelInProgressAttempt(attemptId, messageId, service, cancelledAt, reason);
-    }
-  }
-
-  async #cancelInProgressAttempt(
-    attemptId: string,
-    messageId: string,
-    cancelAttemptService: CancelAttemptService,
-    cancelledAt: Date,
-    reason: string
-  ) {
-    try {
-      await cancelAttemptService.call(attemptId, messageId, cancelledAt, reason);
-    } catch (e) {
-      logger.error("Failed to cancel in progress attempt", {
-        attemptId,
-        messageId,
-        error: e,
-      });
-    }
-  }
-
   #enable() {
     if (this._enabled) {
       return;
@@ -495,8 +462,6 @@ export class SharedQueueConsumer {
               },
             });
           }
-
-          this._inProgressAttempts.set(taskRunAttempt.friendlyId, message.messageId);
         } catch (e) {
           if (e instanceof Error) {
             this._currentSpan?.recordException(e);

apps/webapp/app/v3/tracer.server.ts

@@ -77,6 +77,12 @@ class CustomWebappSampler implements Sampler {
 export const tracer = singleton("tracer", getTracer);
 
 function getTracer() {
+  if (env.INTERNAL_OTEL_TRACE_DISABLED === "1") {
+    console.log(`🔦 Tracer disabled, returning a noop tracer`);
+
+    return trace.getTracer("trigger.dev", "3.0.0.dp.1");
+  }
+
   diag.setLogger(new DiagConsoleLogger(), DiagLogLevel.ERROR);
 
   const samplingRate = 1.0 / Math.max(parseInt(env.INTERNAL_OTEL_TRACE_SAMPLING_RATE, 10), 1);