Class TraceAdapter

worker/src/adapters/trace.adapter.ts:11–139

 * can persist events to PostgreSQL via Drizzle when a database instance is provided.
 */
export class TraceAdapter implements ITraceService {
  private readonly bufferEnabled: boolean;
  private readonly eventsByRun: Map<string, TraceEvent[]> | undefined;
  private readonly sequenceByRun = new Map<string, number>();
  private readonly metadataByRun = new Map<
    string,
    { workflowId?: string; organizationId?: string | null }
  >();
  private readonly logger: Pick<Console, 'log' | 'error'>;

  constructor(
    private readonly db?: NodePgDatabase<typeof schema>,
    options: {
      buffer?: boolean;
      logger?: Pick<Console, 'log' | 'error'>;
    } = {},
  ) {
    this.bufferEnabled = options.buffer ?? false;
    this.eventsByRun = this.bufferEnabled ? new Map<string, TraceEvent[]>() : undefined;
    this.logger = options.logger ?? console;
  }

  record(event: TraceEvent): void {
    if (this.bufferEnabled && this.eventsByRun) {
      const list = this.eventsByRun.get(event.runId) ?? [];
      list.push(event);
      this.eventsByRun.set(event.runId, list);
    }

    const context =
      event.message !== undefined
        ? `${event.type} - ${event.nodeRef}: ${event.message}`
        : `${event.type} - ${event.nodeRef}`;
    this.logger.log(`[TRACE][${event.level}] ${context}`);

    if (!this.db) {
      return;
    }

    const sequence = this.nextSequence(event.runId);
    void this.persist(event, sequence).catch((error) => {
      this.logger.error('[TRACE] Failed to persist trace event', error);
    });
  }

  getEvents(runId: string): TraceEvent[] {
    if (!this.bufferEnabled || !this.eventsByRun) {
      return [];
    }
    return this.eventsByRun.get(runId) ?? [];
  }

  clear(): void {
    if (this.eventsByRun) {
      this.eventsByRun.clear();
    }
    this.sequenceByRun.clear();
    this.metadataByRun.clear();