@geekmidas/constructs 10.0.0-alpha.1 → 10.0.0-alpha.100

package/dist/adaptors/aws.mjs

@@ -1,22 +1,246 @@
-import { __toESM } from "../Construct-CdhThxDI.mjs";
-import "../construct-interface-Z4CpS8tK.mjs";
-import "../functions-Cpff1uZH.mjs";
-import "../clone-DblgEZZa.mjs";
-import { Endpoint, ResponseBuilder, require_lodash } from "../Endpoint-BTj58Gob.mjs";
-import { envParserFor } from "../surfaceEnv-C3Ktahea.mjs";
-import { publishConstructEvents } from "../publisher-CGEo8UKD.mjs";
-import { AWSLambdaFunction, AWSScheduledFunction } from "../AWSScheduledFunction-BJjj8B5V.mjs";
-import { createAuditContext, executeWithAuditTransaction } from "../processAudits-D89tICM4.mjs";
-import { createApiGatewayCookies } from "../lazyAccessors-DM6HcXQk.mjs";
+import { o as __toESM } from "../Construct-B1vGBPme.mjs";
+import { n as recordCarrier } from "../tracing-CxaZ98IB.mjs";
+import { a as publishEvents, i as publishConstructEvents, t as FunctionBuilder } from "../functions-BFizikJU.mjs";
+import { i as require_lodash_set, n as ResponseBuilder, t as Endpoint } from "../Endpoint-BG4UJ9oA.mjs";
+import { n as envParserFor } from "../surfaceEnv-wMT5QzgZ.mjs";
+import { t as createApiGatewayCookies } from "../lazyAccessors-CFanua0O.mjs";
+import { n as subscribedEvents, r as subscriberContext, t as runSubscriber } from "../runSubscriber-BHvJ8POr.mjs";
+import { n as executeWithAuditTransaction, t as createAuditContext } from "../processAudits-BRRVVHUB.mjs";
+import { consumeBatchTraced } from "@geekmidas/events";
 import { ForbiddenError, UnprocessableEntityError, wrapError } from "@geekmidas/errors";
 import { ServiceDiscovery, runWithRequestContext } from "@geekmidas/services";
+import qs from "qs";
 import middy from "@middy/core";
+import { DefaultAuditor } from "@geekmidas/audit";
+import { withAuditableTransaction } from "@geekmidas/audit/kysely";
 import { withRlsContext } from "@geekmidas/db/rls";
-import qs from "qs";
-
+//#region src/functions/FunctionExecutionWrapper.ts
+var FunctionExecutionWrapper = class {
+	envParser;
+	fn;
+	constructor(envParser, fn) {
+		this.envParser = envParser;
+		this.fn = fn;
+	}
+	_logger;
+	get logger() {
+		return this._logger || this.fn.logger;
+	}
+	get serviceDiscovery() {
+		return ServiceDiscovery.getInstance(this.envParser);
+	}
+	getServices() {
+		return this.serviceDiscovery.register(this.fn.services);
+	}
+	async getDatabase() {
+		if (!this.fn.databaseService) return;
+		return (await this.serviceDiscovery.register([this.fn.databaseService]))[this.fn.databaseService.serviceName];
+	}
+	/**
+	* Get the audit storage service if configured.
+	* Returns undefined if no auditor storage is configured.
+	*/
+	async getAuditStorage() {
+		if (!this.fn.auditorStorageService) return;
+		return (await this.serviceDiscovery.register([this.fn.auditorStorageService]))[this.fn.auditorStorageService.serviceName];
+	}
+	/**
+	* Create an auditor instance for the function.
+	* Returns undefined if no auditor storage is configured.
+	*/
+	async createAuditor() {
+		const storage = await this.getAuditStorage();
+		if (!storage) return;
+		return new DefaultAuditor({
+			actor: {
+				id: "system",
+				type: "system"
+			},
+			storage,
+			metadata: { function: this.fn.type }
+		});
+	}
+	/**
+	* Execute handler with audit transaction support.
+	* If the audit storage has a database (via getDatabase()), wraps execution
+	* in a transaction so audits are atomic with handler's database operations.
+	*
+	* @param handler - The handler function to execute (receives auditor and db)
+	* @returns The handler result
+	*/
+	async executeWithAudit(handler) {
+		const auditor = await this.createAuditor();
+		const storage = await this.getAuditStorage();
+		if (!auditor || !storage) return handler({ db: await this.getDatabase() });
+		const storageDb = storage.getDatabase?.();
+		const databaseServiceName = this.fn.databaseService?.serviceName;
+		const auditDbServiceName = this.fn.auditorStorageService?.serviceName;
+		if (storageDb && databaseServiceName && auditDbServiceName) return withAuditableTransaction(storageDb, auditor, async (trx) => {
+			return await handler({
+				auditor,
+				db: trx
+			});
+		});
+		const response = await handler({
+			auditor,
+			db: await this.getDatabase()
+		});
+		await auditor.flush();
+		return response;
+	}
+	async getFunctionInput(event) {
+		return await FunctionBuilder.parseComposableStandardSchema(event, this.fn.input);
+	}
+	async publishEvents(response) {
+		await publishEvents(this.logger, this.serviceDiscovery, this.fn.events, response);
+	}
+	async parseComposableStandardSchema(data, schema) {
+		return FunctionBuilder.parseComposableStandardSchema(data, schema);
+	}
+};
+//#endregion
+//#region src/functions/AWSLambdaFunction.ts
+var AWSLambdaFunction = class extends FunctionExecutionWrapper {
+	fn;
+	constructor(envParser, fn) {
+		super(envParser, fn);
+		this.fn = fn;
+	}
+	error() {
+		return { onError: (req) => {
+			(req.event?.logger || this.fn.logger).error(req.error || {}, "Error processing function");
+			throw wrapError(req.error);
+		} };
+	}
+	baseInput() {
+		return { before: (_req) => {} };
+	}
+	input() {
+		return { before: async (req) => {
+			try {
+				if (this.fn.input) {
+					const parsedInput = await FunctionBuilder.parseComposableStandardSchema(req.event, this.fn.input);
+					req.event.parsedInput = parsedInput;
+				} else req.event.parsedInput = req.event;
+			} catch (error) {
+				this.logger.error({
+					err: error,
+					event: req.event
+				}, "Failed to parse input");
+				throw error;
+			}
+		} };
+	}
+	loggerMiddleware() {
+		return { before: (req) => {
+			this._logger = this.fn.logger.child({
+				fn: {
+					name: req.context.functionName,
+					version: req.context.functionVersion,
+					memory: req.context.memoryLimitInMB
+				},
+				req: { id: req.context.awsRequestId }
+			});
+			req.event.logger = this._logger;
+		} };
+	}
+	services() {
+		return { before: async (req) => {
+			req.event.services = await this.getServices();
+		} };
+	}
+	database() {
+		return { before: async (req) => {
+			req.event.db = await this.getDatabase();
+		} };
+	}
+	auditor() {
+		return {
+			before: async (req) => {
+				req.event.auditor = await this.createAuditor();
+			},
+			after: async (req) => {
+				if (req.event.auditor) {
+					const records = req.event.auditor.getRecords();
+					if (records.length > 0) {
+						this.logger.debug({ auditCount: records.length }, "Flushing function audits");
+						await req.event.auditor.flush();
+					}
+				}
+			}
+		};
+	}
+	events() {
+		return { after: async (req) => {
+			const response = req.response || void 0;
+			await this.publishEvents(response);
+		} };
+	}
+	async _handler(event) {
+		const result = await this.fn.fn({
+			input: event.parsedInput,
+			services: event.services,
+			logger: event.logger,
+			db: event.db,
+			auditor: event.auditor
+		});
+		return await this.fn.parseOutput(result);
+	}
+	get handler() {
+		const handler = this._handler.bind(this);
+		const chain = middy(handler).use(this.loggerMiddleware()).use(this.baseInput()).use(this.error()).use(this.services()).use(this.database()).use(this.auditor()).use(this.input()).use(this.events());
+		const wrappedHandler = async (event, context) => {
+			const startTime = Date.now();
+			const requestId = context.awsRequestId;
+			const logger = this.fn.logger.child({ requestId });
+			return runWithRequestContext({
+				logger,
+				requestId,
+				startTime,
+				operation: context.functionName
+			}, () => chain(event, context));
+		};
+		return wrappedHandler;
+	}
+};
+//#endregion
+//#region src/crons/AWSScheduledFunction.ts
+/**
+* AWS Lambda adaptor for {@link Cron} constructs.
+*
+* A `Cron` is a {@link import('../functions').Function} that is invoked on a
+* schedule (an EventBridge rule) rather than by an API request. Because of
+* that, the runtime execution pipeline is identical to a regular Lambda
+* function: input parsing, service/database/auditor wiring, event publishing,
+* error handling, and the `runWithRequestContext` wrapper that powers
+* request-scoped logging via `serviceContext.getLogger()`.
+*
+* This adaptor therefore reuses {@link AWSLambdaFunction} wholesale and only
+* narrows the constructor to accept a `Cron`. The schedule expression itself
+* (`cron.schedule`) is **deploy-time** infrastructure consumed by the CLI to
+* provision the EventBridge rule — it is not needed at runtime.
+*
+* @example
+* ```typescript
+* import { AWSScheduledFunction } from '@geekmidas/constructs/crons';
+* import { cancelPastBookings } from '../../../src/crons/cancelPastBookings.js';
+* import { envParser } from '../../../src/env';
+*
+* const adapter = new AWSScheduledFunction(envParser, cancelPastBookings);
+*
+* export const handler = adapter.handler;
+* ```
+*/
+var AWSScheduledFunction = class extends AWSLambdaFunction {
+	constructor(envParser, cron) {
+		super(envParser, cron);
+	}
+};
+//#endregion
 //#region src/endpoints/AmazonApiGatewayEndpointAdaptor.ts
-var import_lodash = __toESM(require_lodash(), 1);
+var import_lodash_set = /* @__PURE__ */ __toESM(require_lodash_set(), 1);
 var AmazonApiGatewayEndpoint = class {
+	endpoint;
 	options;
 	/**
 	* The parser this adaptor discovers services with.
@@ -54,8 +278,7 @@ var AmazonApiGatewayEndpoint = class {
 	static decodeBody(body, isBase64Encoded, contentType) {
 		if (!body) return void 0;
 		const raw = isBase64Encoded ? Buffer.from(body, "base64").toString("utf-8") : body;
-		const isJson = !contentType || contentType.toLowerCase().includes("application/json");
-		if (isJson) return JSON.parse(raw);
+		if (!contentType || contentType.toLowerCase().includes("application/json")) return JSON.parse(raw);
 		return raw;
 	}
 	getCookies(e) {
@@ -69,11 +292,11 @@ var AmazonApiGatewayEndpoint = class {
 				const headers = req.event.headers;
 				const header = Endpoint.createHeaders(headers);
 				const cookie = this.getCookies(req.event);
-				(0, import_lodash.default)(req.event, "body", await this.endpoint.parseInput(body, "body"));
-				(0, import_lodash.default)(req.event, "query", await this.endpoint.parseInput(query, "query"));
-				(0, import_lodash.default)(req.event, "params", await this.endpoint.parseInput(params, "params"));
-				(0, import_lodash.default)(req.event, "header", header);
-				(0, import_lodash.default)(req.event, "cookie", cookie);
+				(0, import_lodash_set.default)(req.event, "body", await this.endpoint.parseInput(body, "body"));
+				(0, import_lodash_set.default)(req.event, "query", await this.endpoint.parseInput(query, "query"));
+				(0, import_lodash_set.default)(req.event, "params", await this.endpoint.parseInput(params, "params"));
+				(0, import_lodash_set.default)(req.event, "header", header);
+				(0, import_lodash_set.default)(req.event, "cookie", cookie);
 			} catch (error) {
 				if (error && typeof error === "object" && Array.isArray(error)) throw new UnprocessableEntityError("Validation failed", error);
 				throw error;
@@ -92,9 +315,8 @@ var AmazonApiGatewayEndpoint = class {
 	}
 	services() {
 		return { before: async (req) => {
-			const _logger = req.event.logger;
-			const serviceDiscovery = ServiceDiscovery.getInstance(this.envParser);
-			const services = await serviceDiscovery.register(this.endpoint.services);
+			req.event.logger;
+			const services = await ServiceDiscovery.getInstance(this.envParser).register(this.endpoint.services);
 			req.event.services = services;
 		} };
 	}
@@ -105,7 +327,7 @@ var AmazonApiGatewayEndpoint = class {
 			const header = req.event.header;
 			const cookie = req.event.cookie;
 			const session = req.event.session;
-			const isAuthorized = await this.endpoint.authorize({
+			if (!await this.endpoint.authorize({
 				header,
 				cookie,
 				services,
@@ -115,8 +337,7 @@ var AmazonApiGatewayEndpoint = class {
 				body: req.event.body,
 				query: req.event.query,
 				params: req.event.params
-			});
-			if (!isAuthorized) {
+			})) {
 				logger.warn("Forbidden access attempt");
 				throw new ForbiddenError("Forbidden access to the endpoint", "You do not have permission to access this resource.");
 			}
@@ -125,9 +346,8 @@ var AmazonApiGatewayEndpoint = class {
 	database() {
 		return { before: async (req) => {
 			if (!this.endpoint.databaseService) return;
-			const _logger = req.event.logger;
-			const serviceDiscovery = ServiceDiscovery.getInstance(this.envParser);
-			const db = await serviceDiscovery.register([this.endpoint.databaseService]).then((s) => s[this.endpoint.databaseService?.serviceName]);
+			req.event.logger;
+			const db = await ServiceDiscovery.getInstance(this.envParser).register([this.endpoint.databaseService]).then((s) => s[this.endpoint.databaseService?.serviceName]);
 			req.event.db = db;
 		} };
 	}
@@ -178,9 +398,8 @@ var AmazonApiGatewayEndpoint = class {
 			cookie: event.cookie,
 			logger
 		}) : void 0;
-		const result = await executeWithAuditTransaction(auditContext, async (auditor) => {
-			const sameDatabase = auditContext?.storage?.databaseServiceName && auditContext.storage.databaseServiceName === this.endpoint.databaseService?.serviceName;
-			const baseDb = sameDatabase ? auditor?.getTransaction?.() ?? rawDb : rawDb;
+		const { output, metadata } = await executeWithAuditTransaction(auditContext, async (auditor) => {
+			const baseDb = auditContext?.storage?.databaseServiceName && auditContext.storage.databaseServiceName === this.endpoint.databaseService?.serviceName ? auditor?.getTransaction?.() ?? rawDb : rawDb;
 			const executeHandler = async (db) => {
 				const responseBuilder = new ResponseBuilder();
 				const response = await this.endpoint.handler({
@@ -194,33 +413,31 @@ var AmazonApiGatewayEndpoint = class {
 					...input
 				}, responseBuilder);
 				let data = response;
-				let metadata$1 = responseBuilder.getMetadata();
+				let metadata = responseBuilder.getMetadata();
 				if (Endpoint.hasMetadata(response)) {
 					data = response.data;
-					metadata$1 = response.metadata;
+					metadata = response.metadata;
 				}
-				const output$1 = this.endpoint.outputSchema ? await this.endpoint.parseOutput(data) : void 0;
 				return {
-					output: output$1,
-					metadata: metadata$1,
+					output: this.endpoint.outputSchema ? await this.endpoint.parseOutput(data) : void 0,
+					metadata,
 					responseBuilder
 				};
 			};
 			if (rlsActive && rlsContext && baseDb) return withRlsContext(baseDb, rlsContext, async (trx) => executeHandler(trx), { prefix: this.endpoint.rlsConfig?.prefix });
 			return executeHandler(baseDb);
-		}, async (result$1, auditor) => {
+		}, async (result, auditor) => {
 			if (!audits?.length) return;
 			for (const audit of audits) {
-				if (audit.when && !audit.when(result$1.output)) continue;
-				const payload = audit.payload(result$1.output);
-				const entityId = audit.entityId?.(result$1.output);
+				if (audit.when && !audit.when(result.output)) continue;
+				const payload = audit.payload(result.output);
+				const entityId = audit.entityId?.(result.output);
 				auditor.audit(audit.type, payload, {
 					table: audit.table,
 					entityId
 				});
 			}
 		}, { db: rawDb });
-		const { output, metadata } = result;
 		const isJsonResponse = this.endpoint.responseType === "application/json";
 		const body = output === void 0 ? void 0 : isJsonResponse ? JSON.stringify(output) : typeof output === "string" ? output : String(output);
 		event.__response = output;
@@ -279,16 +496,17 @@ var AmazonApiGatewayEndpoint = class {
 			const startTime = Date.now();
 			const requestId = context.awsRequestId;
 			const logger = this.endpoint.logger.child({ requestId });
+			const operation = `${this.endpoint.method} ${this.endpoint.route}`;
 			return runWithRequestContext({
 				logger,
 				requestId,
-				startTime
+				startTime,
+				operation
 			}, () => chain(event, context));
 		};
 		return wrappedHandler;
 	}
 };
-
 //#endregion
 //#region src/endpoints/parseQueryParams.ts
 /**
@@ -309,10 +527,10 @@ function parseQueryParams(queryParams) {
 	}
 	return qs.parse(parts.join("&"));
 }
-
 //#endregion
 //#region src/endpoints/AmazonApiGatewayV1EndpointAdaptor.ts
 var AmazonApiGatewayV1Endpoint = class extends AmazonApiGatewayEndpoint {
+	endpoint;
 	getInput(e) {
 		const multiValueParams = e.multiValueQueryStringParameters || {};
 		const singleValueParams = e.queryStringParameters || {};
@@ -345,17 +563,16 @@ var AmazonApiGatewayV1Endpoint = class extends AmazonApiGatewayEndpoint {
 		this.endpoint = endpoint;
 	}
 };
-
 //#endregion
 //#region src/endpoints/AmazonApiGatewayV2EndpointAdaptor.ts
 var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
+	endpoint;
 	getCookies(e) {
 		return createApiGatewayCookies(e.cookies, e.headers?.cookie);
 	}
 	getInput(e) {
-		const raw = AmazonApiGatewayEndpoint.decodeBody(e.body, e.isBase64Encoded, e.headers?.["content-type"]);
 		return {
-			body: raw,
+			body: AmazonApiGatewayEndpoint.decodeBody(e.body, e.isBase64Encoded, e.headers?.["content-type"]),
 			query: qs.parse(e.rawQueryString),
 			params: e.pathParameters || {}
 		};
@@ -380,7 +597,6 @@ var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
 		this.endpoint = endpoint;
 	}
 };
-
 //#endregion
 //#region src/queue/AWSLambdaQueueAdaptor.ts
 /**
@@ -395,6 +611,8 @@ var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
 * retries only those records.
 */
 var AWSLambdaQueue = class {
+	envParser;
+	queue;
 	_services;
 	constructor(envParser, queue) {
 		this.envParser = envParser;
@@ -409,6 +627,12 @@ var AWSLambdaQueue = class {
 		this._services = this.queue.services.length > 0 ? await serviceDiscovery.register(this.queue.services) : {};
 		return this._services;
 	}
+	/** The queue's database, registered as functions register theirs. */
+	async getDatabase() {
+		const service = this.queue.databaseService;
+		if (!service) return void 0;
+		return (await ServiceDiscovery.getInstance(this.envParser).register([service]))[service.serviceName];
+	}
 	safeJsonParse(value) {
 		try {
 			return JSON.parse(value);
@@ -434,6 +658,7 @@ var AWSLambdaQueue = class {
 			req: { id: context.awsRequestId }
 		});
 		const services = await this.getServices();
+		const db = await this.getDatabase();
 		const schema = this.queue.messageSchema;
 		const batchItemFailures = [];
 		const messages = [];
@@ -451,7 +676,7 @@ var AWSLambdaQueue = class {
 			messages.push(validation.value);
 		} catch (error) {
 			logger.error({
-				error,
+				err: error,
 				messageId: record.messageId
 			}, "Failed to parse SQS record");
 			batchItemFailures.push({ itemIdentifier: record.messageId });
@@ -461,7 +686,8 @@ var AWSLambdaQueue = class {
 			await this.queue.handler({
 				messages,
 				services,
-				logger
+				logger,
+				db
 			});
 		} catch (error) {
 			logger.error(wrapError(error), "Queue handler failed; retrying batch");
@@ -471,24 +697,28 @@ var AWSLambdaQueue = class {
 	}
 	get handler() {
 		const handler = this._handler.bind(this);
-		return async (event, context) => {
+		return (async (event, context) => {
 			const startTime = Date.now();
 			const requestId = context.awsRequestId;
 			const logger = this.queue.logger.child({ requestId });
 			return runWithRequestContext({
 				logger,
 				requestId,
-				startTime
-			}, () => handler(event, context));
-		};
+				startTime,
+				operation: `queue ${this.queue.name}`
+			}, () => consumeBatchTraced({
+				system: "aws_sqs",
+				destination: this.queue.name
+			}, event.Records.map(recordCarrier), () => handler(event, context)));
+		});
 	}
 };
-
 //#endregion
 //#region src/subscribers/AWSLambdaSubscriberAdaptor.ts
 var AWSLambdaSubscriber = class {
+	envParser;
+	subscriber;
 	_logger;
-	_services;
 	constructor(envParser, subscriber) {
 		this.envParser = envParser;
 		this.subscriber = subscriber;
@@ -497,19 +727,9 @@ var AWSLambdaSubscriber = class {
 	get logger() {
 		return this._logger;
 	}
-	async getServices() {
-		if (this._services) return this._services;
-		const serviceDiscovery = ServiceDiscovery.getInstance(this.envParser);
-		if (this.subscriber.services.length > 0) {
-			const registered = await serviceDiscovery.register(this.subscriber.services);
-			this._services = registered;
-		} else this._services = {};
-		return this._services;
-	}
 	error() {
 		return { onError: (req) => {
-			const logger = req.event?.logger || this.subscriber.logger;
-			logger.error(req.error || {}, "Error processing subscriber");
+			(req.event?.logger || this.subscriber.logger).error(req.error || {}, "Error processing subscriber");
 			throw wrapError(req.error);
 		} };
 	}
@@ -528,110 +748,26 @@ var AWSLambdaSubscriber = class {
 	}
 	services() {
 		return { before: async (req) => {
-			req.event.services = await this.getServices();
+			const { services, db } = await subscriberContext(this.subscriber, this.envParser);
+			req.event.services = services;
+			req.event.db = db;
 		} };
 	}
 	parseEvents() {
 		return { before: async (req) => {
-			const { logger,...e } = req.event;
-			const rawEvent = e;
+			const { logger, ...rawEvent } = req.event;
 			logger.info({ rawEvent });
-			const events = [];
-			if ("Records" in rawEvent) {
-				if (this.isSQSEvent(rawEvent)) for (const record of rawEvent.Records) try {
-					const event = this.parseSQSRecord(record);
-					if (this.shouldIncludeEvent(event)) events.push(event);
-				} catch (error) {
-					this.logger.error({
-						error,
-						record
-					}, "Failed to parse SQS record");
-				}
-				else if (this.isSNSEvent(rawEvent)) for (const record of rawEvent.Records) try {
-					const event = this.parseSNSRecord(record);
-					if (this.shouldIncludeEvent(event)) events.push(event);
-				} catch (error) {
-					this.logger.error({
-						error,
-						record
-					}, "Failed to parse SNS record");
-				}
-			}
-			req.event.events = events;
+			req.event.events = subscribedEvents(this.subscriber, rawEvent, this.logger);
 		} };
 	}
-	isSQSEvent(event) {
-		const firstRecord = event.Records[0];
-		return "Records" in event && event.Records.length > 0 && firstRecord !== void 0 && "eventSource" in firstRecord && firstRecord.eventSource === "aws:sqs";
-	}
-	isSNSEvent(event) {
-		const firstRecord = event.Records[0];
-		return "Records" in event && event.Records.length > 0 && firstRecord !== void 0 && "EventSource" in firstRecord && firstRecord.EventSource === "aws:sns";
-	}
-	parseSNSRecord(record) {
-		const message = this.safeJsonParse(record.Sns.Message);
-		const messageType = record.Sns.MessageAttributes?.type?.Value;
-		if (message === null) return messageType ? {
-			type: messageType,
-			payload: record.Sns.Message
-		} : record.Sns.Message;
-		const resolvedType = messageType ?? message.type;
-		if (message.type) return message;
-		return resolvedType ? {
-			type: resolvedType,
-			payload: message
-		} : message;
-	}
-	parseSQSRecord(record) {
-		const body = this.safeJsonParse(record.body);
-		if (body === null) return record.body;
-		if (body.Type === "Notification" && body.Message) {
-			const snsMessage = this.safeJsonParse(body.Message);
-			const messageType = body.MessageAttributes?.type?.Value;
-			if (snsMessage === null) return messageType ? {
-				type: messageType,
-				payload: body.Message
-			} : body.Message;
-			if (snsMessage.type) return snsMessage;
-			const resolvedType = messageType ?? snsMessage.type;
-			return resolvedType ? {
-				type: resolvedType,
-				payload: snsMessage
-			} : snsMessage;
-		}
-		return body;
-	}
-	safeJsonParse(value) {
-		try {
-			return JSON.parse(value);
-		} catch {
-			return null;
-		}
-	}
-	shouldIncludeEvent(event) {
-		if (typeof event !== "object" || !event?.type) return true;
-		if (!this.subscriber.subscribedEvents) return true;
-		return this.subscriber.subscribedEvents.includes(event.type);
-	}
 	async _handler(event) {
-		if (event.events.length === 0) {
-			this.logger.info("No subscribed events to process");
-			return { batchItemFailures: [] };
-		}
-		const result = await this.subscriber.handler({
-			events: event.events,
-			services: event.services,
-			logger: event.logger
+		const { events, services, logger, db } = event;
+		return runSubscriber(this.subscriber, {
+			events,
+			services,
+			logger,
+			db
 		});
-		if (this.subscriber.outputSchema && result) {
-			const validationResult = await this.subscriber.outputSchema["~standard"].validate(result);
-			if (validationResult.issues) {
-				this.logger.error({ issues: validationResult.issues }, "Subscriber output validation failed");
-				throw new Error("Subscriber output validation failed");
-			}
-			return validationResult.value;
-		}
-		return result;
 	}
 	get handler() {
 		const handler = this._handler.bind(this);
@@ -640,16 +776,22 @@ var AWSLambdaSubscriber = class {
 			const startTime = Date.now();
 			const requestId = context.awsRequestId;
 			const logger = this.subscriber.logger.child({ requestId });
+			const operation = `subscriber ${this.subscriber.topicName ?? context.functionName}`;
+			const records = event?.Records ?? [];
 			return runWithRequestContext({
 				logger,
 				requestId,
-				startTime
-			}, () => chain(event, context));
+				startTime,
+				operation
+			}, () => consumeBatchTraced({
+				system: records[0]?.Sns ? "aws_sns" : "aws_sqs",
+				destination: this.subscriber.topicName ?? context.functionName
+			}, records.map(recordCarrier), () => chain(event, context)));
 		};
 		return wrappedHandler;
 	}
 };
-
 //#endregion
 export { AWSLambdaFunction, AWSLambdaQueue, AWSLambdaSubscriber, AWSScheduledFunction, AmazonApiGatewayV1Endpoint, AmazonApiGatewayV2Endpoint };
+
 //# sourceMappingURL=aws.mjs.map