|
| 1 | +import { Config } from '../config/app' |
| 2 | +import { OTLPMetricExporter } from '@opentelemetry/exporter-metrics-otlp-grpc' |
| 3 | +import { Resource } from '@opentelemetry/resources' |
| 4 | +import { |
| 5 | + MeterProvider, |
| 6 | + PeriodicExportingMetricReader |
| 7 | +} from '@opentelemetry/sdk-metrics' |
| 8 | + |
| 9 | +import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-grpc' |
| 10 | +import { api } from '@opentelemetry/sdk-node' |
| 11 | +import { PgInstrumentation } from '@opentelemetry/instrumentation-pg' |
| 12 | +import { GraphQLInstrumentation } from '@opentelemetry/instrumentation-graphql' |
| 13 | + |
| 14 | +import { HttpInstrumentation } from '@opentelemetry/instrumentation-http' |
| 15 | +import { |
| 16 | + BatchSpanProcessor, |
| 17 | + NodeTracerProvider |
| 18 | +} from '@opentelemetry/sdk-trace-node' |
| 19 | +import { registerInstrumentations } from '@opentelemetry/instrumentation' |
| 20 | +import { UndiciInstrumentation } from '@opentelemetry/instrumentation-undici' |
| 21 | + |
| 22 | +// debug logger: |
| 23 | +// diag.setLogger(new DiagConsoleLogger(), DiagLogLevel.DEBUG) |
| 24 | + |
| 25 | +const SERVICE_NAME = 'RAFIKI_NETWORK' |
| 26 | +const rafikiResource = new Resource({ |
| 27 | + 'service.name': SERVICE_NAME, |
| 28 | + instance: Config.instanceName |
| 29 | +}) |
| 30 | + |
| 31 | +if (Config.enableTelemetry) { |
| 32 | + const meterReaders = [] |
| 33 | + |
| 34 | + for (const url of Config.openTelemetryCollectors) { |
| 35 | + const metricExporter = new PeriodicExportingMetricReader({ |
| 36 | + exporter: new OTLPMetricExporter({ |
| 37 | + url |
| 38 | + }), |
| 39 | + exportIntervalMillis: Config.openTelemetryExportInterval ?? 15000 |
| 40 | + }) |
| 41 | + |
| 42 | + meterReaders.push(metricExporter) |
| 43 | + } |
| 44 | + |
| 45 | + const meterProvider = new MeterProvider({ |
| 46 | + resource: rafikiResource, |
| 47 | + readers: meterReaders |
| 48 | + }) |
| 49 | + |
| 50 | + api.metrics.setGlobalMeterProvider(meterProvider) |
| 51 | +} |
| 52 | + |
| 53 | +if (Config.enableTelemetryTraces) { |
| 54 | + const tracerProvider = new NodeTracerProvider({ |
| 55 | + resource: rafikiResource |
| 56 | + }) |
| 57 | + |
| 58 | + for (const url of Config.openTelemetryTraceCollectorUrls) { |
| 59 | + const traceExporter = new OTLPTraceExporter({ |
| 60 | + url |
| 61 | + }) |
| 62 | + |
| 63 | + tracerProvider.addSpanProcessor(new BatchSpanProcessor(traceExporter)) |
| 64 | + } |
| 65 | + |
| 66 | + tracerProvider.register() |
| 67 | + |
| 68 | + registerInstrumentations({ |
| 69 | + instrumentations: [ |
| 70 | + new UndiciInstrumentation(), |
| 71 | + new HttpInstrumentation(), |
| 72 | + new PgInstrumentation(), |
| 73 | + new GraphQLInstrumentation({ |
| 74 | + mergeItems: true, |
| 75 | + ignoreTrivialResolveSpans: true, |
| 76 | + ignoreResolveSpans: true |
| 77 | + }) |
| 78 | + ] |
| 79 | + }) |
| 80 | +} |
0 commit comments