Class OTelCommandMetrics

packages/client/lib/opentelemetry/metrics.ts:54–189

}

class OTelCommandMetrics implements IOTelCommandMetrics {
  readonly #instruments: MetricInstruments;
  readonly #options: MetricOptions;
  readonly #metricsState = new WeakMap<object, CommandMetricsState>();
  readonly #unsubscribers: Array<() => void> = [];

  constructor(options: MetricOptions, instruments: MetricInstruments) {
    this.#options = options;
    this.#instruments = instruments;

    this.#subscribeToTracingChannel();
  }

  #subscribeToTracingChannel() {
    const commandTC = getTracingChannel(CHANNELS.TRACE_COMMAND);
    const batchTC = getTracingChannel(CHANNELS.TRACE_BATCH);
    if (!commandTC || !batchTC) return;

    const onStart = (ctx: any) => {
      const commandName = ctx.command?.toString() || "UNKNOWN";

      if (this.#isCommandExcluded(commandName)) return;

      this.#metricsState.set(ctx, {
        startTime: performance.now(),
        clientAttributes: resolveClientAttributes(ctx.clientId),
        commandName,
      });
    };

    const onAsyncEnd = (ctx: any) => {
      const state = this.#metricsState.get(ctx);
      if (!state) return;
      this.#metricsState.delete(ctx);

      this.#instruments.dbClientOperationDuration.record(
        (performance.now() - state.startTime) / 1000,
        {
          ...this.#options.attributes,
          [OTEL_ATTRIBUTES.dbNamespace]: state.clientAttributes?.db?.toString(),
          [OTEL_ATTRIBUTES.serverAddress]: state.clientAttributes?.host,
          [OTEL_ATTRIBUTES.serverPort]: state.clientAttributes?.port?.toString(),
          [OTEL_ATTRIBUTES.dbOperationName]: state.commandName,
        },
      );
    };

    const onError = (ctx: any) => {
      const state = this.#metricsState.get(ctx);
      if (!state) return;
      this.#metricsState.delete(ctx);

      const errorInfo = getErrorInfo(ctx.error);
      this.#instruments.dbClientOperationDuration.record(
        (performance.now() - state.startTime) / 1000,
        {
          ...this.#options.attributes,
          [OTEL_ATTRIBUTES.dbNamespace]: state.clientAttributes?.db?.toString(),