@event-driven-io/emmett-postgresql 0.43.0-beta.27 → 0.43.0-beta.29

package/dist/index.cjs

@@ -767,6 +767,8 @@ const getPostgreSQLEventStore = (connectionString, options = defaultPostgreSQLOp
 	let migrateSchema = void 0;
 	const autoGenerateSchema = options.schema?.autoMigration === void 0 || options.schema?.autoMigration !== "None";
 	const inlineProjections = (options.projections ?? []).filter(({ type }) => type === "inline").map(({ projection }) => projection);
+	const observability = (0, _event_driven_io_emmett.eventStoreObservability)(options);
+	const collector = (0, _event_driven_io_emmett.eventStoreCollector)(observability);
 	const migrate = async (migrationOptions) => {
 		if (!migrateSchema) {
 			migrateSchema = createEventStoreSchema(connectionString, pool, {
@@ -800,11 +802,19 @@ const getPostgreSQLEventStore = (connectionString, options = defaultPostgreSQLOp
 		if (!autoGenerateSchema) return Promise.resolve();
 		return migrate();
 	};
-	const beforeCommitHook = inlineProjections.length > 0 ? async (events, { transaction }) => handleProjections({
+	const readStreamFromPostgreSQL = (streamName, readOptions) => collector.instrumentRead(streamName, async () => {
+		await ensureSchemaExists();
+		return readStream(pool.execute, streamName, {
+			...readOptions,
+			serialization: options.serialization ?? readOptions?.serialization
+		});
+	}, readOptions?.observability);
+	const beforeCommitHook = (streamName, appendScope) => inlineProjections.length > 0 ? async (events, { transaction }) => collector.instrumentInlineProjection(streamName, appendScope, async (observabilityScope) => handleProjections({
 		projections: inlineProjections,
 		events,
-		...await transactionToPostgreSQLProjectionHandlerContext(connectionString, pool, transaction)
-	}) : void 0;
+		...await transactionToPostgreSQLProjectionHandlerContext(connectionString, pool, transaction),
+		observabilityScope
+	})) : void 0;
 	return {
 		schema: {
 			sql: () => _event_driven_io_dumbo.SQL.describe(schemaSQL, (0, _event_driven_io_dumbo.getFormatter)((0, _event_driven_io_dumbo.fromDatabaseDriverType)(pool.driverType).databaseType)),
@@ -823,36 +833,36 @@ const getPostgreSQLEventStore = (connectionString, options = defaultPostgreSQLOp
 			}) }
 		},
 		async aggregateStream(streamName, options) {
-			const { evolve, initialState, read } = options;
-			const expectedStreamVersion = read?.expectedStreamVersion;
-			let state = initialState();
-			const result = await this.readStream(streamName, read);
-			const currentStreamVersion = result.currentStreamVersion;
-			(0, _event_driven_io_emmett.assertExpectedVersionMatchesCurrent)(currentStreamVersion, expectedStreamVersion, PostgreSQLEventStoreDefaultStreamVersion);
-			for (const event of result.events) {
-				if (!event) continue;
-				state = evolve(state, event);
-			}
-			return {
-				currentStreamVersion,
-				state,
-				streamExists: result.streamExists
-			};
-		},
-		readStream: async (streamName, readOptions) => {
-			await ensureSchemaExists();
-			return readStream(pool.execute, streamName, {
-				...readOptions,
-				serialization: options.serialization ?? readOptions?.serialization
-			});
+			return collector.instrumentAggregate(streamName, async (scope) => {
+				const { evolve, initialState, read } = options;
+				const expectedStreamVersion = read?.expectedStreamVersion;
+				let state = initialState();
+				const result = await readStreamFromPostgreSQL(streamName, {
+					...read ?? {},
+					observability: (0, _event_driven_io_emmett.withOperationScope)(scope, read?.observability)
+				});
+				const currentStreamVersion = result.currentStreamVersion;
+				(0, _event_driven_io_emmett.assertExpectedVersionMatchesCurrent)(currentStreamVersion, expectedStreamVersion, PostgreSQLEventStoreDefaultStreamVersion);
+				for (const event of result.events) {
+					if (!event) continue;
+					state = evolve(state, event);
+				}
+				return {
+					currentStreamVersion,
+					state,
+					streamExists: result.streamExists
+				};
+			}, options.observability);
 		},
-		appendToStream: async (streamName, events, appendOptions) => {
+		readStream: async (streamName, readOptions) => readStreamFromPostgreSQL(streamName, readOptions),
+		appendToStream: async (streamName, events, appendOptions) => collector.instrumentAppend(streamName, events, async (scope) => {
 			await ensureSchemaExists();
 			const [firstPart, ...rest] = streamName.split("-");
 			const streamType = firstPart && rest.length > 0 ? firstPart : _event_driven_io_emmett.unknownTag;
 			const appendResult = await pool.withConnection(async (connection) => appendToStream(connection, streamName, streamType, (0, _event_driven_io_emmett.downcastRecordedMessages)(events, appendOptions?.schema?.versioning), {
 				...appendOptions,
-				beforeCommitHook
+				messageIdGenerator: () => observability.contextGenerator.generateMessageId(),
+				beforeCommitHook: beforeCommitHook(streamName, scope)
 			}));
 			if (!appendResult.success) throw new _event_driven_io_emmett.ExpectedVersionConflictError(-1n, appendOptions?.expectedStreamVersion ?? _event_driven_io_emmett.NO_CONCURRENCY_CHECK);
 			return {
@@ -860,7 +870,7 @@ const getPostgreSQLEventStore = (connectionString, options = defaultPostgreSQLOp
 				lastEventGlobalPosition: PostgreSQLEventStoreCheckpoint.toProcessorCheckpoint(appendResult.checkpoints[appendResult.checkpoints.length - 1]),
 				createdNewStream: appendResult.nextStreamPosition >= BigInt(events.length)
 			};
-		},
+		}, appendOptions?.observability),
 		streamExists: async (streamName, options) => {
 			await ensureSchemaExists();
 			return streamExists(pool.execute, streamName, options);
@@ -1041,7 +1051,7 @@ const handleProjections = async (options) => {
 				transaction
 			},
 			execute: transaction.execute,
-			observabilityScope: _event_driven_io_emmett.noopScope
+			observabilityScope: options.observabilityScope
 		});
 	}
 };
@@ -1277,7 +1287,7 @@ const appendToStream = (connection, streamName, streamType, messages, options) =
 			...e,
 			kind: e.kind ?? "Event",
 			metadata: {
-				messageId: (0, uuid.v4)(),
+				messageId: options?.messageIdGenerator?.() ?? (0, uuid.v4)(),
 				..."metadata" in e ? e.metadata ?? {} : {}
 			}
 		}));