@m1212e/rumble 0.23.2 → 0.23.4

package/out/index.mjs

@@ -1,10 +1,9 @@
 import { t as lazy } from "./lazy-C7IyM4M4.mjs";
 import { t as generateFromSchema } from "./generate-0QFVc3Sg.mjs";
-import { GraphQLError, GraphQLSchema } from "graphql";
+import { GraphQLError, GraphQLSchema, execute as defaultExecute, subscribe as defaultSubscribe } from "graphql";
 import { EnvelopArmorPlugin } from "@escape.tech/graphql-armor";
 import { useDisableIntrospection } from "@graphql-yoga/plugin-disable-introspection";
 import { SpanStatusCode, trace } from "@opentelemetry/api";
-import { AttributeNames, SpanNames, createOpenTelemetryWrapper } from "@pothos/tracing-opentelemetry";
 import { capitalize, cloneDeep, debounce, merge, toMerged } from "es-toolkit";
 import { createPubSub, createYoga } from "graphql-yoga";
 import { One, count, getTableColumns, getTableName, is, isTable, relationsFilterToSQL, sql } from "drizzle-orm";
@@ -12,6 +11,7 @@ import { toCamelCase } from "drizzle-orm/casing";
 import { PgColumn, PgEnumColumn, PgEnumObjectColumn, PgTable, isPgEnum } from "drizzle-orm/pg-core";
 import { MySqlEnumColumn, MySqlEnumObjectColumn, MySqlTable } from "drizzle-orm/mysql-core";
 import pluralize from "pluralize";
+import { AttributeNames, SpanNames, createOpenTelemetryWrapper } from "@pothos/tracing-opentelemetry";
 import { SQLiteTable } from "drizzle-orm/sqlite-core";
 import SchemaBuilder, { BasePlugin } from "@pothos/core";
 import DrizzlePlugin from "@pothos/plugin-drizzle";
@@ -180,7 +180,7 @@ function mapNullFieldsToUndefined(obj) {
 }
 //#endregion
 //#region package.json
-var version = "0.23.2";
+var version = "0.23.4";
 //#endregion
 //#region lib/helpers/mergeFilters.ts
 function mergeFilters(filterA, filterB, mode = "AND") {
@@ -1272,6 +1272,165 @@ const sofaOpenAPIWebhookDocs = {
 	} }
 };
 //#endregion
+//#region lib/helpers/tracing.ts
+function isAsyncIterable(value) {
+	return typeof value === "object" && value !== null && Symbol.asyncIterator in value;
+}
+async function* wrapSubscriptionIterator(iterator, log, operationName) {
+	let eventCount = 0;
+	try {
+		for await (const event of iterator) {
+			eventCount++;
+			if (event.errors?.length) log.error({
+				"graphql.operation.name": operationName,
+				eventCount,
+				errors: event.errors.map((e) => e.message)
+			}, "graphql subscription event error");
+			yield event;
+		}
+		log.info({
+			"graphql.operation.name": operationName,
+			eventCount
+		}, "graphql subscription completed");
+	} catch (error) {
+		log.error({
+			"graphql.operation.name": operationName,
+			eventCount,
+			err: error
+		}, "graphql subscription threw");
+		throw error;
+	}
+}
+function buildTracedExecute(executeFn, rumbleInput) {
+	return async (options) => {
+		let log = rumbleInput.logger?.enabled ? rumbleInput.logger.logger : void 0;
+		const operationName = options.operationName ?? "anonymous";
+		const start = Date.now();
+		log?.info({ "graphql.operation.name": operationName }, "graphql execute start");
+		const run = async () => {
+			const result = await executeFn(options);
+			if (result && "errors" in result && result.errors?.length) log?.error({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start,
+				errors: result.errors.map((e) => e.message)
+			}, "graphql execute completed with errors");
+			else log?.info({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start
+			}, "graphql execute completed");
+			return result;
+		};
+		if (rumbleInput.otel?.enabled) return rumbleInput.otel.tracer.startActiveSpan(SpanNames.EXECUTE, { attributes: {
+			[AttributeNames.OPERATION_NAME]: operationName,
+			[AttributeNames.SOURCE]: options.document
+		} }, async (span) => {
+			if (log && rumbleInput.logger?.injectTraceId !== false) {
+				const { traceId, spanId } = span.spanContext();
+				log = log.child({
+					traceId,
+					spanId
+				});
+			}
+			try {
+				const result = await run();
+				if (result && "errors" in result && result.errors?.length) {
+					for (const error of result.errors) span.recordException(error);
+					span.setStatus({ code: SpanStatusCode.ERROR });
+				}
+				return result;
+			} catch (error) {
+				if (error instanceof Error) span.recordException(error);
+				log?.error({
+					"graphql.operation.name": operationName,
+					durationMs: Date.now() - start,
+					err: error
+				}, "graphql execute threw");
+				span.setStatus({ code: SpanStatusCode.ERROR });
+				throw error;
+			} finally {
+				span.end();
+			}
+		});
+		try {
+			return await run();
+		} catch (error) {
+			log?.error({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start,
+				err: error
+			}, "graphql execute threw");
+			throw error;
+		}
+	};
+}
+function buildTracedSubscribe(subscribeFn, rumbleInput) {
+	return async (options) => {
+		let log = rumbleInput.logger?.enabled ? rumbleInput.logger.logger : void 0;
+		const operationName = options.operationName ?? "anonymous";
+		const start = Date.now();
+		const doSubscribe = async () => {
+			log?.info({ "graphql.operation.name": operationName }, "graphql subscribe start");
+			const result = await subscribeFn(options);
+			if (!isAsyncIterable(result)) {
+				const execResult = result;
+				if (execResult.errors?.length) log?.error({
+					"graphql.operation.name": operationName,
+					durationMs: Date.now() - start,
+					errors: execResult.errors.map((e) => e.message)
+				}, "graphql subscribe completed with errors");
+				return execResult;
+			}
+			log?.info({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start
+			}, "graphql subscription established");
+			if (log) return wrapSubscriptionIterator(result, log, operationName);
+			return result;
+		};
+		if (rumbleInput.otel?.enabled) return rumbleInput.otel.tracer.startActiveSpan("graphql.subscribe", async (span) => {
+			if (log && rumbleInput.logger?.injectTraceId !== false) {
+				const { traceId, spanId } = span.spanContext();
+				log = log.child({
+					traceId,
+					spanId
+				});
+			}
+			try {
+				const result = await doSubscribe();
+				if (!isAsyncIterable(result)) {
+					const execResult = result;
+					if (execResult.errors?.length) {
+						for (const error of execResult.errors) span.recordException(error);
+						span.setStatus({ code: SpanStatusCode.ERROR });
+					}
+				}
+				return result;
+			} catch (error) {
+				if (error instanceof Error) span.recordException(error);
+				log?.error({
+					"graphql.operation.name": operationName,
+					durationMs: Date.now() - start,
+					err: error
+				}, "graphql subscribe threw");
+				span.setStatus({ code: SpanStatusCode.ERROR });
+				throw error;
+			} finally {
+				span.end();
+			}
+		});
+		try {
+			return await doSubscribe();
+		} catch (error) {
+			log?.error({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start,
+				err: error
+			}, "graphql subscribe threw");
+			throw error;
+		}
+	};
+}
+//#endregion
 //#region lib/helpers/sqlTypes/mapDrizzleTypeToGraphQlType.ts
 function buildPothosResponseTypeFromGraphQLType({ builder, sqlType, fieldName, nullable, isArray }) {
 	const gqlType = mapSQLTypeToGraphQLType({
@@ -2309,59 +2468,7 @@ export const r = rumble({
 				...args?.plugins ?? [],
 				...enableApiDocs ? [] : [useDisableIntrospection(), EnvelopArmorPlugin(args?.armorConfig)],
 				rumbleInput.otel?.enabled || rumbleInput.logger?.enabled ? { onExecute: ({ setExecuteFn, executeFn }) => {
-					setExecuteFn(async (options) => {
-						const log = rumbleInput.logger?.enabled ? rumbleInput.logger.logger : void 0;
-						const operationName = options.operationName ?? "anonymous";
-						const start = Date.now();
-						log?.info({ "graphql.operation.name": operationName }, "graphql execute start");
-						const run = async () => {
-							const result = await executeFn(options);
-							if (result && "errors" in result && result.errors?.length) log?.error({
-								"graphql.operation.name": operationName,
-								durationMs: Date.now() - start,
-								errors: result.errors.map((e) => e.message)
-							}, "graphql execute completed with errors");
-							else log?.info({
-								"graphql.operation.name": operationName,
-								durationMs: Date.now() - start
-							}, "graphql execute completed");
-							return result;
-						};
-						if (rumbleInput.otel?.enabled) return rumbleInput.otel.tracer.startActiveSpan(SpanNames.EXECUTE, { attributes: {
-							[AttributeNames.OPERATION_NAME]: operationName,
-							[AttributeNames.SOURCE]: options.document
-						} }, async (span) => {
-							try {
-								const result = await run();
-								if (result && "errors" in result && result.errors?.length) {
-									for (const error of result.errors) span.recordException(error);
-									span.setStatus({ code: SpanStatusCode.ERROR });
-								}
-								return result;
-							} catch (error) {
-								if (error instanceof Error) span.recordException(error);
-								log?.error({
-									"graphql.operation.name": operationName,
-									durationMs: Date.now() - start,
-									err: error
-								}, "graphql execute threw");
-								span.setStatus({ code: SpanStatusCode.ERROR });
-								throw error;
-							} finally {
-								span.end();
-							}
-						});
-						try {
-							return await run();
-						} catch (error) {
-							log?.error({
-								"graphql.operation.name": operationName,
-								durationMs: Date.now() - start,
-								err: error
-							}, "graphql execute threw");
-							throw error;
-						}
-					});
+					setExecuteFn(buildTracedExecute(executeFn, rumbleInput));
 				} } : false
 			].filter(Boolean)
 		});
@@ -2382,6 +2489,8 @@ export const r = rumble({
 			...args,
 			schema: builtSchema(),
 			context,
+			execute: buildTracedExecute(args.execute ?? defaultExecute, rumbleInput),
+			subscribe: buildTracedSubscribe(args.subscribe ?? defaultSubscribe, rumbleInput),
 			errorHandler(errors) {
 				const span = trace.getActiveSpan();
 				for (const error of errors) span?.recordException(error);
@@ -2394,7 +2503,9 @@ export const r = rumble({
 		return implementation({
 			...args,
 			schema: builtSchema(),
-			context
+			context,
+			execute: buildTracedExecute(args.execute ?? defaultExecute, rumbleInput),
+			subscribe: buildTracedSubscribe(args.subscribe ?? defaultSubscribe, rumbleInput)
 		}, ...rest);
 	};
 	return {