@kitlangton/motel 0.1.0

package/src/services/TraceQueryService.ts

@@ -0,0 +1,71 @@
+import { Effect, Layer, Context } from "effect"
+import type { SpanItem, TraceItem, TraceSummaryItem } from "../domain.js"
+import { TelemetryStore } from "./TelemetryStore.js"
+
+export class TraceQueryService extends Context.Service<
+	TraceQueryService,
+	{
+		readonly listServices: Effect.Effect<readonly string[], Error>
+		readonly listRecentTraces: (serviceName: string, options?: { readonly lookbackMinutes?: number; readonly limit?: number }) => Effect.Effect<readonly TraceItem[], Error>
+		readonly listTraceSummaries: (serviceName: string, options?: { readonly lookbackMinutes?: number; readonly limit?: number }) => Effect.Effect<readonly TraceSummaryItem[], Error>
+		readonly searchTraces: (input: { readonly serviceName?: string | null; readonly operation?: string | null; readonly status?: "ok" | "error" | null; readonly minDurationMs?: number | null; readonly lookbackMinutes?: number; readonly limit?: number; readonly attributeFilters?: Readonly<Record<string, string>> }) => Effect.Effect<readonly TraceItem[], Error>
+		readonly traceStats: (input: { readonly groupBy: string; readonly agg: "count" | "avg_duration" | "p95_duration" | "error_rate"; readonly serviceName?: string | null; readonly operation?: string | null; readonly status?: "ok" | "error" | null; readonly minDurationMs?: number | null; readonly lookbackMinutes?: number; readonly limit?: number; readonly attributeFilters?: Readonly<Record<string, string>> }) => Effect.Effect<readonly { readonly group: string; readonly value: number; readonly count: number }[], Error>
+		readonly getTrace: (traceId: string) => Effect.Effect<TraceItem | null, Error>
+		readonly getSpan: (spanId: string) => Effect.Effect<SpanItem | null, Error>
+		readonly listTraceSpans: (traceId: string) => Effect.Effect<readonly SpanItem[], Error>
+		readonly searchSpans: (input: { readonly serviceName?: string | null; readonly operation?: string | null; readonly parentOperation?: string | null; readonly status?: "ok" | "error" | null; readonly lookbackMinutes?: number; readonly limit?: number; readonly attributeFilters?: Readonly<Record<string, string>> }) => Effect.Effect<readonly SpanItem[], Error>
+	}
+>()("motel/TraceQueryService") {}
+
+export const TraceQueryServiceLive = Layer.effect(
+	TraceQueryService,
+	Effect.gen(function* () {
+		const store = yield* TelemetryStore
+
+		const listServices = Effect.fn("motel/TraceQueryService.listServices")(function* () {
+			const services = yield* store.listServices
+			yield* Effect.annotateCurrentSpan("trace.service_count", services.length)
+			return services
+		})()
+
+		const listRecentTraces = Effect.fn("motel/TraceQueryService.listRecentTraces")(function* (serviceName: string, options?: { readonly lookbackMinutes?: number; readonly limit?: number }) {
+			yield* Effect.annotateCurrentSpan({
+				"trace.service_name": serviceName,
+			})
+			const traces = yield* store.listRecentTraces(serviceName, options)
+			yield* Effect.annotateCurrentSpan("trace.result_count", traces.length)
+			return traces
+		})
+
+		const listTraceSummaries = Effect.fn("motel/TraceQueryService.listTraceSummaries")(function* (serviceName: string, options?: { readonly lookbackMinutes?: number; readonly limit?: number }) {
+			yield* Effect.annotateCurrentSpan({
+				"trace.service_name": serviceName,
+			})
+			const traces = yield* store.listTraceSummaries(serviceName, options)
+			yield* Effect.annotateCurrentSpan("trace.result_count", traces.length)
+			return traces
+		})
+
+		const getTrace = Effect.fn("motel/TraceQueryService.getTrace")(function* (traceId: string) {
+			yield* Effect.annotateCurrentSpan("trace.trace_id", traceId)
+			return yield* store.getTrace(traceId)
+		})
+
+		const getSpan = Effect.fn("motel/TraceQueryService.getSpan")(function* (spanId: string) {
+			yield* Effect.annotateCurrentSpan("trace.span_id", spanId)
+			return yield* store.getSpan(spanId)
+		})
+
+		return TraceQueryService.of({
+			listServices,
+			listRecentTraces,
+			listTraceSummaries,
+			searchTraces: store.searchTraces,
+			traceStats: store.traceStats,
+			getTrace,
+			getSpan,
+			listTraceSpans: store.listTraceSpans,
+			searchSpans: store.searchSpans,
+		})
+	}),
+)