83 lines
2 KiB
TypeScript
83 lines
2 KiB
TypeScript
import retry from 'next/dist/compiled/async-retry'
|
|
import { randomBytes } from 'crypto'
|
|
import fetch from 'node-fetch'
|
|
import * as Log from '../../../build/output/log'
|
|
// Jaeger uses Zipkin's reporting
|
|
import { batcher } from './to-zipkin'
|
|
|
|
let traceId = process.env.TRACE_ID
|
|
let batch: ReturnType<typeof batcher> | undefined
|
|
|
|
const localEndpoint = {
|
|
serviceName: 'nextjs',
|
|
ipv4: '127.0.0.1',
|
|
port: 9411,
|
|
}
|
|
// Jaeger supports Zipkin's reporting API
|
|
const zipkinUrl = `http://${localEndpoint.ipv4}:${localEndpoint.port}`
|
|
const jaegerWebUiUrl = `http://${localEndpoint.ipv4}:16686`
|
|
const zipkinAPI = `${zipkinUrl}/api/v2/spans`
|
|
|
|
function logWebUrl() {
|
|
Log.info(
|
|
`Jaeger trace will be available on ${jaegerWebUiUrl}/trace/${traceId}`
|
|
)
|
|
}
|
|
|
|
const reportToLocalHost = (
|
|
name: string,
|
|
duration: number,
|
|
timestamp: number,
|
|
id: string,
|
|
parentId?: string,
|
|
attrs?: Object
|
|
) => {
|
|
if (!traceId) {
|
|
traceId = process.env.TRACE_ID = randomBytes(8).toString('hex')
|
|
logWebUrl()
|
|
}
|
|
|
|
if (!batch) {
|
|
batch = batcher((events) => {
|
|
const eventsJson = JSON.stringify(events)
|
|
// Ensure ECONNRESET error is retried 3 times before erroring out
|
|
return retry(
|
|
() =>
|
|
// Send events to zipkin
|
|
fetch(zipkinAPI, {
|
|
method: 'POST',
|
|
headers: { 'Content-Type': 'application/json' },
|
|
body: eventsJson,
|
|
}),
|
|
{ minTimeout: 500, retries: 3, factor: 1 }
|
|
)
|
|
.then(async (res: any) => {
|
|
if (res.status !== 202) {
|
|
console.log({
|
|
status: res.status,
|
|
body: await res.text(),
|
|
events: eventsJson,
|
|
})
|
|
}
|
|
})
|
|
.catch(console.log)
|
|
})
|
|
}
|
|
|
|
batch.report({
|
|
traceId,
|
|
parentId,
|
|
name,
|
|
id,
|
|
timestamp,
|
|
duration,
|
|
localEndpoint,
|
|
tags: attrs,
|
|
})
|
|
}
|
|
|
|
export default {
|
|
flushAll: () =>
|
|
batch ? batch.flushAll().then(() => logWebUrl()) : undefined,
|
|
report: reportToLocalHost,
|
|
}
|