chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,102 @@
|
||||
import type { Span, SpanOptions, Tracer } from "@opentelemetry/api";
|
||||
import { SpanKind, SpanStatusCode } from "@opentelemetry/api";
|
||||
import type { Logger } from "@opentelemetry/api-logs";
|
||||
import { SeverityNumber } from "@opentelemetry/api-logs";
|
||||
import { flattenAttributes } from "@trigger.dev/core/v3/utils/flattenAttributes";
|
||||
import type { AuthenticatedEnvironment } from "~/services/apiAuth.server";
|
||||
import { attributesFromAuthenticatedEnv } from "./tracer.server";
|
||||
|
||||
export async function startSpan<T>(
|
||||
tracer: Tracer,
|
||||
name: string,
|
||||
fn: (span: Span) => Promise<T>,
|
||||
options?: SpanOptions
|
||||
): Promise<T> {
|
||||
return tracer.startActiveSpan(name, options ?? {}, async (span) => {
|
||||
try {
|
||||
return await fn(span);
|
||||
} catch (error) {
|
||||
if (error instanceof Error) {
|
||||
span.recordException(error);
|
||||
} else if (typeof error === "string") {
|
||||
span.recordException(new Error(error));
|
||||
} else {
|
||||
span.recordException(new Error(String(error)));
|
||||
}
|
||||
|
||||
span.setStatus({
|
||||
code: SpanStatusCode.ERROR,
|
||||
message: error instanceof Error ? error.message : String(error),
|
||||
});
|
||||
|
||||
throw error;
|
||||
} finally {
|
||||
span.end();
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
export async function startSpanWithEnv<T>(
|
||||
tracer: Tracer,
|
||||
name: string,
|
||||
env: AuthenticatedEnvironment,
|
||||
fn: (span: Span) => Promise<T>,
|
||||
options?: SpanOptions
|
||||
): Promise<T> {
|
||||
return startSpan(tracer, name, fn, {
|
||||
...options,
|
||||
attributes: {
|
||||
...attributesFromAuthenticatedEnv(env),
|
||||
...options?.attributes,
|
||||
},
|
||||
kind: SpanKind.SERVER,
|
||||
});
|
||||
}
|
||||
|
||||
export async function emitDebugLog(
|
||||
logger: Logger,
|
||||
message: string,
|
||||
params: Record<string, unknown> = {}
|
||||
) {
|
||||
logger.emit({
|
||||
severityNumber: SeverityNumber.DEBUG,
|
||||
body: message,
|
||||
attributes: { ...flattenAttributes(params, "params") },
|
||||
});
|
||||
}
|
||||
|
||||
export async function emitInfoLog(
|
||||
logger: Logger,
|
||||
message: string,
|
||||
params: Record<string, unknown> = {}
|
||||
) {
|
||||
logger.emit({
|
||||
severityNumber: SeverityNumber.INFO,
|
||||
body: message,
|
||||
attributes: { ...flattenAttributes(params, "params") },
|
||||
});
|
||||
}
|
||||
|
||||
export async function emitErrorLog(
|
||||
logger: Logger,
|
||||
message: string,
|
||||
params: Record<string, unknown> = {}
|
||||
) {
|
||||
logger.emit({
|
||||
severityNumber: SeverityNumber.ERROR,
|
||||
body: message,
|
||||
attributes: { ...flattenAttributes(params, "params") },
|
||||
});
|
||||
}
|
||||
|
||||
export async function emitWarnLog(
|
||||
logger: Logger,
|
||||
message: string,
|
||||
params: Record<string, unknown> = {}
|
||||
) {
|
||||
logger.emit({
|
||||
severityNumber: SeverityNumber.WARN,
|
||||
body: message,
|
||||
attributes: { ...flattenAttributes(params, "params") },
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user