]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blame - server/lib/opentelemetry/tracing.ts
Refactor playlist creation for lives
[github/Chocobozzz/PeerTube.git] / server / lib / opentelemetry / tracing.ts
CommitLineData
ce6b3765 1import { SequelizeInstrumentation } from 'opentelemetry-instrumentation-sequelize'
2a95b884 2import { context, diag, DiagLogLevel, trace } from '@opentelemetry/api'
630d0a1b
C
3import { JaegerExporter } from '@opentelemetry/exporter-jaeger'
4import { registerInstrumentations } from '@opentelemetry/instrumentation'
5import { DnsInstrumentation } from '@opentelemetry/instrumentation-dns'
6import { ExpressInstrumentation } from '@opentelemetry/instrumentation-express'
7import FsInstrumentation from '@opentelemetry/instrumentation-fs'
8import { HttpInstrumentation } from '@opentelemetry/instrumentation-http'
077600e0 9import { IORedisInstrumentation } from '@opentelemetry/instrumentation-ioredis'
630d0a1b 10import { PgInstrumentation } from '@opentelemetry/instrumentation-pg'
630d0a1b
C
11import { Resource } from '@opentelemetry/resources'
12import { BatchSpanProcessor } from '@opentelemetry/sdk-trace-base'
13import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node'
14import { SemanticResourceAttributes } from '@opentelemetry/semantic-conventions'
15import { logger } from '@server/helpers/logger'
16import { CONFIG } from '@server/initializers/config'
17
2a95b884
C
18const tracer = trace.getTracer('peertube')
19
630d0a1b
C
20function registerOpentelemetryTracing () {
21 if (CONFIG.OPEN_TELEMETRY.TRACING.ENABLED !== true) return
22
23 logger.info('Registering Open Telemetry tracing')
24
25 const customLogger = (level: string) => {
26 return (message: string, ...args: unknown[]) => {
27 let fullMessage = message
28
29 for (const arg of args) {
30 if (typeof arg === 'string') fullMessage += arg
31 else break
32 }
33
34 logger[level](fullMessage)
35 }
36 }
37
38 diag.setLogger({
39 error: customLogger('error'),
40 warn: customLogger('warn'),
41 info: customLogger('info'),
42 debug: customLogger('debug'),
43 verbose: customLogger('verbose')
44 }, DiagLogLevel.INFO)
45
46 const tracerProvider = new NodeTracerProvider({
47 resource: new Resource({
48 [SemanticResourceAttributes.SERVICE_NAME]: 'peertube'
49 })
50 })
51
52 registerInstrumentations({
ba2684ce 53 tracerProvider,
630d0a1b
C
54 instrumentations: [
55 new PgInstrumentation({
56 enhancedDatabaseReporting: true
57 }),
58 new DnsInstrumentation(),
59 new HttpInstrumentation(),
60 new ExpressInstrumentation(),
077600e0 61 new IORedisInstrumentation({
630d0a1b
C
62 dbStatementSerializer: function (cmdName, cmdArgs) {
63 return [ cmdName, ...cmdArgs ].join(' ')
64 }
65 }),
ce6b3765
C
66 new FsInstrumentation(),
67 new SequelizeInstrumentation()
630d0a1b
C
68 ]
69 })
70
71 tracerProvider.addSpanProcessor(
72 new BatchSpanProcessor(
73 new JaegerExporter({ endpoint: CONFIG.OPEN_TELEMETRY.TRACING.JAEGER_EXPORTER.ENDPOINT })
74 )
75 )
76
77 tracerProvider.register()
78}
79
2a95b884
C
80async function wrapWithSpanAndContext <T> (spanName: string, cb: () => Promise<T>) {
81 const span = tracer.startSpan(spanName)
82 const activeContext = trace.setSpan(context.active(), span)
83
84 const result = await context.with(activeContext, () => cb())
85 span.end()
86
87 return result
88}
630d0a1b
C
89
90export {
91 registerOpentelemetryTracing,
2a95b884
C
92 tracer,
93 wrapWithSpanAndContext
630d0a1b 94}