refactor: rename remoteQuery to query and db instrumentation flag (#9159)

This commit is contained in:
Harminder Virk
2024-09-17 14:09:02 +05:30
committed by GitHub
parent 987d007ba8
commit a872c76b85

View File

@@ -262,34 +262,37 @@ export function registerOtel(options: {
exporter: SpanExporter
instrument?: Partial<{
http: boolean
remoteQuery: boolean
query: boolean
workflows: boolean
db: boolean
}>
instrumentations?: Instrumentation[]
}) {
const sdk = new NodeSDK({
serviceName: options.serviceName,
resource: new Resource({
"service.name": options.serviceName,
}),
spanProcessor: new SimpleSpanProcessor(options.exporter),
instrumentations: [
new PgInstrumentation(),
...(options.instrumentations || []),
],
})
const instrument = options.instrument || {}
const instrumentations = options.instrumentations || []
if (instrument.db) {
instrumentations.push(new PgInstrumentation())
}
if (instrument.http) {
instrumentHttpLayer()
}
if (instrument.remoteQuery) {
if (instrument.query) {
instrumentRemoteQuery()
}
if (instrument.workflows) {
instrumentWorkflows()
}
const sdk = new NodeSDK({
serviceName: options.serviceName,
resource: new Resource({
"service.name": options.serviceName,
}),
spanProcessor: new SimpleSpanProcessor(options.exporter),
instrumentations: instrumentations,
})
sdk.start()
return sdk
}