76 lines
2.4 KiB
TypeScript
76 lines
2.4 KiB
TypeScript
|
import { SpanKind, SpanStatusCode, type TracerProvider, trace } from '@opentelemetry/api'
|
||
|
|
import {
|
||
|
|
ATTR_HTTP_REQUEST_HEADER,
|
||
|
|
ATTR_HTTP_REQUEST_METHOD,
|
||
|
|
ATTR_HTTP_RESPONSE_HEADER,
|
||
|
|
ATTR_HTTP_RESPONSE_STATUS_CODE,
|
||
|
|
ATTR_URL_FULL,
|
||
|
|
ATTR_HTTP_ROUTE,
|
||
|
|
} from '@opentelemetry/semantic-conventions'
|
||
|
|
import type { Env, Input } from 'hono'
|
||
|
|
import { createMiddleware } from 'hono/factory'
|
||
|
|
import metadata from '../package.json' with { type: 'json'}
|
||
|
|
|
||
|
|
const PACKAGE_NAME = metadata.name
|
||
|
|
const PACKAGE_VERSION = metadata.version
|
||
|
|
|
||
|
|
export type OtelOptions = {
|
||
|
|
augmentSpan?: false;
|
||
|
|
tracerProvider?: TracerProvider
|
||
|
|
} | {
|
||
|
|
augmentSpan: true;
|
||
|
|
}
|
||
|
|
|
||
|
|
export const otel = <E extends Env = any, P extends string = any, I extends Input = {}>(
|
||
|
|
options: OtelOptions = {}
|
||
|
|
) => {
|
||
|
|
if (options.augmentSpan) {
|
||
|
|
return createMiddleware<E, P, I>(async (c, next) => {
|
||
|
|
const result = await next()
|
||
|
|
const span = trace.getActiveSpan()
|
||
|
|
if (span != null) {
|
||
|
|
const route = c.req.matchedRoutes[c.req.matchedRoutes.length - 1]
|
||
|
|
span.setAttribute(ATTR_HTTP_ROUTE, route.path)
|
||
|
|
span.updateName(`${c.req.method} ${route.path}`)
|
||
|
|
}
|
||
|
|
return result
|
||
|
|
})
|
||
|
|
}
|
||
|
|
const tracerProvider = options.tracerProvider ?? trace.getTracerProvider()
|
||
|
|
const tracer = tracerProvider.getTracer(PACKAGE_NAME, PACKAGE_VERSION)
|
||
|
|
return createMiddleware<E, P, I>(async (c, next) => {
|
||
|
|
const route = c.req.matchedRoutes[c.req.matchedRoutes.length - 1]
|
||
|
|
await tracer.startActiveSpan(
|
||
|
|
`${c.req.method} ${route.path}`,
|
||
|
|
{
|
||
|
|
kind: SpanKind.SERVER,
|
||
|
|
attributes: {
|
||
|
|
[ATTR_HTTP_REQUEST_METHOD]: c.req.method,
|
||
|
|
[ATTR_URL_FULL]: c.req.url,
|
||
|
|
[ATTR_HTTP_ROUTE]: route.path,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
async (span) => {
|
||
|
|
for (const [name, value] of Object.entries(c.req.header())) {
|
||
|
|
span.setAttribute(ATTR_HTTP_REQUEST_HEADER(name), value)
|
||
|
|
}
|
||
|
|
try {
|
||
|
|
await next()
|
||
|
|
span.setAttribute(ATTR_HTTP_RESPONSE_STATUS_CODE, c.res.status)
|
||
|
|
for (const [name, value] of c.res.headers.entries()) {
|
||
|
|
span.setAttribute(ATTR_HTTP_RESPONSE_HEADER(name), value)
|
||
|
|
}
|
||
|
|
if (c.error) {
|
||
|
|
span.setStatus({ code: SpanStatusCode.ERROR, message: String(c.error) })
|
||
|
|
}
|
||
|
|
} catch (e) {
|
||
|
|
span.setStatus({ code: SpanStatusCode.ERROR, message: String(e) })
|
||
|
|
throw e
|
||
|
|
} finally {
|
||
|
|
span.end()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
)
|
||
|
|
})
|
||
|
|
}
|