mirror of https://github.com/Chocobozzz/PeerTube
116 lines
3.4 KiB
TypeScript
116 lines
3.4 KiB
TypeScript
import { Application, Request, Response } from 'express'
|
|
import { Meter, metrics } from '@opentelemetry/api'
|
|
import { PrometheusExporter } from '@opentelemetry/exporter-prometheus'
|
|
import { MeterProvider } from '@opentelemetry/sdk-metrics'
|
|
import { logger } from '@server/helpers/logger.js'
|
|
import { CONFIG } from '@server/initializers/config.js'
|
|
import { MVideoImmutable } from '@server/types/models/index.js'
|
|
import { PlaybackMetricCreate } from '@peertube/peertube-models'
|
|
import {
|
|
BittorrentTrackerObserversBuilder,
|
|
JobQueueObserversBuilder,
|
|
LivesObserversBuilder,
|
|
NodeJSObserversBuilder,
|
|
PlaybackMetrics,
|
|
StatsObserversBuilder,
|
|
ViewersObserversBuilder
|
|
} from './metric-helpers/index.js'
|
|
import { WorkerThreadsObserversBuilder } from './metric-helpers/worker-threads-observers.js'
|
|
|
|
class OpenTelemetryMetrics {
|
|
|
|
private static instance: OpenTelemetryMetrics
|
|
|
|
private meter: Meter
|
|
|
|
private onRequestDuration: (req: Request, res: Response) => void
|
|
|
|
private playbackMetrics: PlaybackMetrics
|
|
|
|
private constructor () {}
|
|
|
|
init (app: Application) {
|
|
if (CONFIG.OPEN_TELEMETRY.METRICS.ENABLED !== true) return
|
|
|
|
app.use((req, res, next) => {
|
|
res.once('finish', () => {
|
|
if (!this.onRequestDuration) return
|
|
|
|
this.onRequestDuration(req as Request, res as Response)
|
|
})
|
|
|
|
next()
|
|
})
|
|
}
|
|
|
|
registerMetrics (options: { trackerServer: any }) {
|
|
if (CONFIG.OPEN_TELEMETRY.METRICS.ENABLED !== true) return
|
|
|
|
logger.info('Registering Open Telemetry metrics')
|
|
|
|
const provider = new MeterProvider({
|
|
views: [
|
|
...NodeJSObserversBuilder.getViews()
|
|
],
|
|
readers: [
|
|
new PrometheusExporter({
|
|
host: CONFIG.OPEN_TELEMETRY.METRICS.PROMETHEUS_EXPORTER.HOSTNAME,
|
|
port: CONFIG.OPEN_TELEMETRY.METRICS.PROMETHEUS_EXPORTER.PORT
|
|
})
|
|
]
|
|
})
|
|
|
|
metrics.setGlobalMeterProvider(provider)
|
|
|
|
this.meter = metrics.getMeter('default')
|
|
|
|
if (CONFIG.OPEN_TELEMETRY.METRICS.HTTP_REQUEST_DURATION.ENABLED === true) {
|
|
this.buildRequestObserver()
|
|
}
|
|
|
|
this.playbackMetrics = new PlaybackMetrics(this.meter)
|
|
this.playbackMetrics.buildCounters()
|
|
|
|
new NodeJSObserversBuilder(this.meter).buildObservers()
|
|
new JobQueueObserversBuilder(this.meter).buildObservers()
|
|
new StatsObserversBuilder(this.meter).buildObservers()
|
|
new LivesObserversBuilder(this.meter).buildObservers()
|
|
new ViewersObserversBuilder(this.meter).buildObservers()
|
|
new WorkerThreadsObserversBuilder(this.meter).buildObservers()
|
|
new BittorrentTrackerObserversBuilder(this.meter, options.trackerServer).buildObservers()
|
|
}
|
|
|
|
observePlaybackMetric (video: MVideoImmutable, metrics: PlaybackMetricCreate) {
|
|
this.playbackMetrics.observe(video, metrics)
|
|
}
|
|
|
|
private buildRequestObserver () {
|
|
const requestDuration = this.meter.createHistogram('http_request_duration_ms', {
|
|
unit: 'milliseconds',
|
|
description: 'Duration of HTTP requests in ms'
|
|
})
|
|
|
|
this.onRequestDuration = (req: Request, res: Response) => {
|
|
const duration = Date.now() - res.locals.requestStart
|
|
|
|
requestDuration.record(duration, {
|
|
path: this.buildRequestPath(req.originalUrl),
|
|
method: req.method,
|
|
statusCode: res.statusCode + ''
|
|
})
|
|
}
|
|
}
|
|
|
|
private buildRequestPath (path: string) {
|
|
return path.split('?')[0]
|
|
}
|
|
|
|
static get Instance () {
|
|
return this.instance || (this.instance = new this())
|
|
}
|
|
}
|
|
|
|
export {
|
|
OpenTelemetryMetrics
|
|
}
|