ClassworksKV/utils/instrumentation.js

import "dotenv/config";
import {NodeSDK} from "@opentelemetry/sdk-node";
import {getNodeAutoInstrumentations} from "@opentelemetry/auto-instrumentations-node";
import {OTLPTraceExporter} from "@opentelemetry/exporter-trace-otlp-proto";
import {BatchSpanProcessor} from "@opentelemetry/sdk-trace-base";
import {resourceFromAttributes} from "@opentelemetry/resources";
import {SemanticResourceAttributes} from "@opentelemetry/semantic-conventions";

if (process.env.AXIOM_TOKEN && process.env.AXIOM_DATASET) {
    // Initialize OTLP trace exporter with the endpoint URL and headers
    // Initialize OTLP trace exporter with the endpoint URL and headers
    const traceExporter = new OTLPTraceExporter({
        url: "https://api.axiom.co/v1/traces",
        headers: {
            Authorization: `Bearer ${process.env.AXIOM_TOKEN}`,
            "X-Axiom-Dataset": process.env.AXIOM_DATASET,
        },
    });

    const resourceAttributes = {
        [SemanticResourceAttributes.SERVICE_NAME]: "node traces",
    };

    const resource = resourceFromAttributes(resourceAttributes);

    // Configuring the OpenTelemetry Node SDK
    const sdk = new NodeSDK({
        // Adding a BatchSpanProcessor to batch and send traces
        spanProcessor: new BatchSpanProcessor(traceExporter),

        // Registering the resource to the SDK
        resource: resource,

        // Adding auto-instrumentations to automatically collect trace data
        instrumentations: [getNodeAutoInstrumentations()],
    });

    console.log("✅成功加载 Axiom 遥测");
    // Starting the OpenTelemetry SDK to begin collecting telemetry data
    sdk.start();
} else {
    console.log("❌未设置 Axiom 遥测");
}