@prisma-next/mongo-runtime 0.3.0-dev.127 → 0.3.0-dev.162

package/README.md

@@ -4,15 +4,16 @@ MongoDB runtime executor for Prisma Next.
 
 ## Responsibilities
 
-- **Runtime executor**: `createMongoRuntime()` composes adapter and driver into a `MongoRuntime` that executes query plans
-- **Command lowering**: Translates ORM query plans into MongoDB commands (`FindCommand`, `AggregateCommand`)
-- **`$lookup` pipeline**: Builds aggregation pipelines for reference-relation includes
+- **Runtime executor**: `createMongoRuntime()` composes adapter and driver into a `MongoRuntime` with a single `execute(plan)` entry point accepting `MongoQueryPlan<Row>` from `@prisma-next/mongo-query-ast`. Execution is one path for reads and writes: `adapter.lower(plan)` produces a wire command, then the driver runs it.
+- **Unified flow**: There is no separate `execute` vs `executeCommand`; all operations use `execute(plan)`.
+- **Lowering**: Happens in the adapter (`lower(plan)`), not inside the runtime.
 - **Lifecycle management**: Connection lifecycle via `close()`
 
 ## Dependencies
 
 - **Depends on**:
-  - `@prisma-next/mongo-core` (contract types, query plan types, lowering context)
+  - `@prisma-next/mongo-lowering` (`MongoAdapter`, `MongoDriver` interfaces)
+  - `@prisma-next/mongo-query-ast` (`MongoQueryPlan`, `AnyMongoCommand` — the typed plan shape)
   - `@prisma-next/runtime-executor` (`AsyncIterableResult` return type)
 - **Depended on by**:
   - Integration tests (`test/integration/test/mongo/`)

package/dist/index.d.mts

@@ -1,11 +1,24 @@
-import { AsyncIterableResult } from "@prisma-next/runtime-executor";
-import { MongoAdapter, MongoDriver, MongoLoweringContext, MongoQueryPlan } from "@prisma-next/mongo-core";
+import { AfterExecuteResult, AsyncIterableResult, RuntimeMiddleware, RuntimeMiddlewareContext } from "@prisma-next/framework-components/runtime";
+import { MongoQueryPlan } from "@prisma-next/mongo-query-ast/execution";
+import { MongoAdapter, MongoDriver } from "@prisma-next/mongo-lowering";
 
+//#region src/mongo-middleware.d.ts
+interface MongoMiddlewareContext extends RuntimeMiddlewareContext {}
+interface MongoMiddleware extends RuntimeMiddleware {
+  readonly familyId: 'mongo';
+  beforeExecute?(plan: MongoQueryPlan, ctx: MongoMiddlewareContext): Promise<void>;
+  onRow?(row: Record<string, unknown>, plan: MongoQueryPlan, ctx: MongoMiddlewareContext): Promise<void>;
+  afterExecute?(plan: MongoQueryPlan, result: AfterExecuteResult, ctx: MongoMiddlewareContext): Promise<void>;
+}
+//#endregion
 //#region src/mongo-runtime.d.ts
 interface MongoRuntimeOptions {
   readonly adapter: MongoAdapter;
   readonly driver: MongoDriver;
-  readonly loweringContext: MongoLoweringContext;
+  readonly contract: unknown;
+  readonly targetId: string;
+  readonly middleware?: readonly RuntimeMiddleware[];
+  readonly mode?: 'strict' | 'permissive';
 }
 interface MongoRuntime {
   execute<Row>(plan: MongoQueryPlan<Row>): AsyncIterableResult<Row>;
@@ -13,5 +26,5 @@ interface MongoRuntime {
 }
 declare function createMongoRuntime(options: MongoRuntimeOptions): MongoRuntime;
 //#endregion
-export { type MongoRuntime, type MongoRuntimeOptions, createMongoRuntime };
+export { type MongoMiddleware, type MongoMiddlewareContext, type MongoRuntime, type MongoRuntimeOptions, createMongoRuntime };
 //# sourceMappingURL=index.d.mts.map

package/dist/index.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.mts","names":[],"sources":["../src/mongo-runtime.ts"],"sourcesContent":[],"mappings":";;;;UAQiB,mBAAA;oBACG;EADH,SAAA,MAAA,EAEE,WAFiB;EAChB,SAAA,eAAA,EAEQ,oBAFR;;AAEQ,UAGX,YAAA,CAHW;EAAoB,OAAA,CAAA,GAAA,CAAA,CAAA,IAAA,EAI3B,cAJ2B,CAIZ,GAJY,CAAA,CAAA,EAIL,mBAJK,CAIe,GAJf,CAAA;EAG/B,KAAA,EAAA,EAEN,OAFkB,CAAA,IAAA,CAAA;;AACR,iBA+BL,kBAAA,CA/BK,OAAA,EA+BuB,mBA/BvB,CAAA,EA+B6C,YA/B7C"}
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/mongo-middleware.ts","../src/mongo-runtime.ts"],"sourcesContent":[],"mappings":";;;;;UAOiB,sBAAA,SAA+B;UAE/B,eAAA,SAAwB;EAFxB,SAAA,QAAA,EAAA,OAAuB;EAEvB,aAAA,EAAA,IAAgB,EAEV,cAFU,EAAA,GAAA,EAEW,sBAFX,CAAA,EAEoC,OAFpC,CAAA,IAAA,CAAA;EAEV,KAAA,EAAA,GAAA,EAEd,MAFc,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,IAAA,EAGb,cAHa,EAAA,GAAA,EAId,sBAJc,CAAA,EAKlB,OALkB,CAAA,IAAA,CAAA;EAAqB,YAAA,EAAA,IAAA,EAOlC,cAPkC,EAAA,MAAA,EAQhC,kBARgC,EAAA,GAAA,EASnC,sBATmC,CAAA,EAUvC,OAVuC,CAAA,IAAA,CAAA;;;;UCK3B,mBAAA;EDTA,SAAA,OAAA,ECUG,YDVoB;EAEvB,SAAA,MAAA,ECSE,WDTc;EAEV,SAAA,QAAA,EAAA,OAAA;EAAqB,SAAA,QAAA,EAAA,MAAA;EAAyB,SAAA,UAAA,CAAA,EAAA,SCUpC,iBDVoC,EAAA;EAE5D,SAAA,IAAA,CAAA,EAAA,QAAA,GAAA,YAAA;;AAEA,UCUQ,YAAA,CDVR;EACJ,OAAA,CAAA,GAAA,CAAA,CAAA,IAAA,ECUgB,cDVhB,CCU+B,GDV/B,CAAA,CAAA,ECUsC,mBDVtC,CCU0D,GDV1D,CAAA;EAEK,KAAA,EAAA,ECSC,ODTD,CAAA,IAAA,CAAA;;AAED,iBCgGO,kBAAA,CDhGP,OAAA,ECgGmC,mBDhGnC,CAAA,ECgGyD,YDhGzD"}

package/dist/index.mjs

@@ -1,22 +1,77 @@
-import { AsyncIterableResult } from "@prisma-next/runtime-executor";
+import { AsyncIterableResult, checkMiddlewareCompatibility } from "@prisma-next/framework-components/runtime";
 
 //#region src/mongo-runtime.ts
+function noop() {}
+function now() {
+	return Date.now();
+}
 var MongoRuntimeImpl = class {
 	#adapter;
 	#driver;
-	#loweringContext;
+	#middleware;
+	#middlewareContext;
 	constructor(options) {
 		this.#adapter = options.adapter;
 		this.#driver = options.driver;
-		this.#loweringContext = options.loweringContext;
+		const middleware = options.middleware ? [...options.middleware] : [];
+		for (const mw of middleware) checkMiddlewareCompatibility(mw, "mongo", options.targetId);
+		this.#middleware = middleware;
+		this.#middlewareContext = {
+			contract: options.contract,
+			mode: options.mode ?? "strict",
+			now,
+			log: {
+				info: noop,
+				warn: noop,
+				error: noop
+			}
+		};
 	}
 	execute(plan) {
-		const executionPlan = this.#adapter.lower(plan, this.#loweringContext);
-		const iterable = this.#driver.execute(executionPlan.wireCommand);
-		async function* toGenerator() {
-			yield* iterable;
-		}
-		return new AsyncIterableResult(toGenerator());
+		const adapter = this.#adapter;
+		const driver = this.#driver;
+		const middleware = this.#middleware;
+		const ctx = this.#middlewareContext;
+		const iterator = async function* () {
+			const startedAt = ctx.now();
+			let rowCount = 0;
+			let completed = false;
+			let failed = false;
+			try {
+				for (const mw of middleware) if (mw.beforeExecute) await mw.beforeExecute(plan, ctx);
+				const wireCommand = adapter.lower(plan);
+				for await (const row of driver.execute(wireCommand)) {
+					for (const mw of middleware) if (mw.onRow) await mw.onRow(row, plan, ctx);
+					rowCount++;
+					yield row;
+				}
+				completed = true;
+			} catch (error) {
+				failed = true;
+				throw error;
+			} finally {
+				const latencyMs = ctx.now() - startedAt;
+				for (const mw of middleware) {
+					if (!mw.afterExecute) continue;
+					if (failed) {
+						try {
+							await mw.afterExecute(plan, {
+								rowCount,
+								latencyMs,
+								completed
+							}, ctx);
+						} catch {}
+						continue;
+					}
+					await mw.afterExecute(plan, {
+						rowCount,
+						latencyMs,
+						completed
+					}, ctx);
+				}
+			}
+		};
+		return new AsyncIterableResult(iterator());
 	}
 	async close() {
 		await this.#driver.close();

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","names":["#adapter","#driver","#loweringContext"],"sources":["../src/mongo-runtime.ts"],"sourcesContent":["import type {\n  MongoAdapter,\n  MongoDriver,\n  MongoLoweringContext,\n  MongoQueryPlan,\n} from '@prisma-next/mongo-core';\nimport { AsyncIterableResult } from '@prisma-next/runtime-executor';\n\nexport interface MongoRuntimeOptions {\n  readonly adapter: MongoAdapter;\n  readonly driver: MongoDriver;\n  readonly loweringContext: MongoLoweringContext;\n}\n\nexport interface MongoRuntime {\n  execute<Row>(plan: MongoQueryPlan<Row>): AsyncIterableResult<Row>;\n  close(): Promise<void>;\n}\n\nclass MongoRuntimeImpl implements MongoRuntime {\n  readonly #adapter: MongoAdapter;\n  readonly #driver: MongoDriver;\n  readonly #loweringContext: MongoLoweringContext;\n\n  constructor(options: MongoRuntimeOptions) {\n    this.#adapter = options.adapter;\n    this.#driver = options.driver;\n    this.#loweringContext = options.loweringContext;\n  }\n\n  execute<Row>(plan: MongoQueryPlan<Row>): AsyncIterableResult<Row> {\n    const executionPlan = this.#adapter.lower(plan, this.#loweringContext);\n    const iterable = this.#driver.execute<Row>(executionPlan.wireCommand);\n\n    async function* toGenerator(): AsyncGenerator<Row, void, unknown> {\n      yield* iterable;\n    }\n\n    return new AsyncIterableResult(toGenerator());\n  }\n\n  async close(): Promise<void> {\n    await this.#driver.close();\n  }\n}\n\nexport function createMongoRuntime(options: MongoRuntimeOptions): MongoRuntime {\n  return new MongoRuntimeImpl(options);\n}\n"],"mappings":";;;AAmBA,IAAM,mBAAN,MAA+C;CAC7C,CAASA;CACT,CAASC;CACT,CAASC;CAET,YAAY,SAA8B;AACxC,QAAKF,UAAW,QAAQ;AACxB,QAAKC,SAAU,QAAQ;AACvB,QAAKC,kBAAmB,QAAQ;;CAGlC,QAAa,MAAqD;EAChE,MAAM,gBAAgB,MAAKF,QAAS,MAAM,MAAM,MAAKE,gBAAiB;EACtE,MAAM,WAAW,MAAKD,OAAQ,QAAa,cAAc,YAAY;EAErE,gBAAgB,cAAkD;AAChE,UAAO;;AAGT,SAAO,IAAI,oBAAoB,aAAa,CAAC;;CAG/C,MAAM,QAAuB;AAC3B,QAAM,MAAKA,OAAQ,OAAO;;;AAI9B,SAAgB,mBAAmB,SAA4C;AAC7E,QAAO,IAAI,iBAAiB,QAAQ"}
+{"version":3,"file":"index.mjs","names":["#adapter","#driver","#middleware","#middlewareContext"],"sources":["../src/mongo-runtime.ts"],"sourcesContent":["import type {\n  RuntimeMiddleware,\n  RuntimeMiddlewareContext,\n} from '@prisma-next/framework-components/runtime';\nimport {\n  AsyncIterableResult,\n  checkMiddlewareCompatibility,\n} from '@prisma-next/framework-components/runtime';\nimport type { MongoAdapter, MongoDriver } from '@prisma-next/mongo-lowering';\nimport type { MongoQueryPlan } from '@prisma-next/mongo-query-ast/execution';\n\nfunction noop() {}\nfunction now() {\n  return Date.now();\n}\n\nexport interface MongoRuntimeOptions {\n  readonly adapter: MongoAdapter;\n  readonly driver: MongoDriver;\n  readonly contract: unknown;\n  readonly targetId: string;\n  readonly middleware?: readonly RuntimeMiddleware[];\n  readonly mode?: 'strict' | 'permissive';\n}\n\nexport interface MongoRuntime {\n  execute<Row>(plan: MongoQueryPlan<Row>): AsyncIterableResult<Row>;\n  close(): Promise<void>;\n}\n\nclass MongoRuntimeImpl implements MongoRuntime {\n  readonly #adapter: MongoAdapter;\n  readonly #driver: MongoDriver;\n  readonly #middleware: readonly RuntimeMiddleware[];\n  readonly #middlewareContext: RuntimeMiddlewareContext;\n\n  constructor(options: MongoRuntimeOptions) {\n    this.#adapter = options.adapter;\n    this.#driver = options.driver;\n\n    const middleware = options.middleware ? [...options.middleware] : [];\n    for (const mw of middleware) {\n      checkMiddlewareCompatibility(mw, 'mongo', options.targetId);\n    }\n    this.#middleware = middleware;\n\n    this.#middlewareContext = {\n      contract: options.contract,\n      mode: options.mode ?? 'strict',\n      now,\n      log: { info: noop, warn: noop, error: noop },\n    };\n  }\n\n  execute<Row>(plan: MongoQueryPlan<Row>): AsyncIterableResult<Row> {\n    const adapter = this.#adapter;\n    const driver = this.#driver;\n    const middleware = this.#middleware;\n    const ctx = this.#middlewareContext;\n\n    const iterator = async function* (): AsyncGenerator<Row, void, unknown> {\n      const startedAt = ctx.now();\n      let rowCount = 0;\n      let completed = false;\n      let failed = false;\n\n      try {\n        for (const mw of middleware) {\n          if (mw.beforeExecute) {\n            await mw.beforeExecute(plan, ctx);\n          }\n        }\n\n        const wireCommand = adapter.lower(plan);\n\n        for await (const row of driver.execute<Row>(wireCommand)) {\n          for (const mw of middleware) {\n            if (mw.onRow) {\n              await mw.onRow(row as Record<string, unknown>, plan, ctx);\n            }\n          }\n          rowCount++;\n          yield row;\n        }\n\n        completed = true;\n      } catch (error) {\n        failed = true;\n        throw error;\n      } finally {\n        const latencyMs = ctx.now() - startedAt;\n        for (const mw of middleware) {\n          if (!mw.afterExecute) continue;\n\n          if (failed) {\n            try {\n              await mw.afterExecute(plan, { rowCount, latencyMs, completed }, ctx);\n            } catch {\n              // Ignore errors from afterExecute during error handling\n            }\n            continue;\n          }\n\n          await mw.afterExecute(plan, { rowCount, latencyMs, completed }, ctx);\n        }\n      }\n    };\n\n    return new AsyncIterableResult(iterator());\n  }\n\n  async close(): Promise<void> {\n    await this.#driver.close();\n  }\n}\n\nexport function createMongoRuntime(options: MongoRuntimeOptions): MongoRuntime {\n  return new MongoRuntimeImpl(options);\n}\n"],"mappings":";;;AAWA,SAAS,OAAO;AAChB,SAAS,MAAM;AACb,QAAO,KAAK,KAAK;;AAiBnB,IAAM,mBAAN,MAA+C;CAC7C,CAASA;CACT,CAASC;CACT,CAASC;CACT,CAASC;CAET,YAAY,SAA8B;AACxC,QAAKH,UAAW,QAAQ;AACxB,QAAKC,SAAU,QAAQ;EAEvB,MAAM,aAAa,QAAQ,aAAa,CAAC,GAAG,QAAQ,WAAW,GAAG,EAAE;AACpE,OAAK,MAAM,MAAM,WACf,8BAA6B,IAAI,SAAS,QAAQ,SAAS;AAE7D,QAAKC,aAAc;AAEnB,QAAKC,oBAAqB;GACxB,UAAU,QAAQ;GAClB,MAAM,QAAQ,QAAQ;GACtB;GACA,KAAK;IAAE,MAAM;IAAM,MAAM;IAAM,OAAO;IAAM;GAC7C;;CAGH,QAAa,MAAqD;EAChE,MAAM,UAAU,MAAKH;EACrB,MAAM,SAAS,MAAKC;EACpB,MAAM,aAAa,MAAKC;EACxB,MAAM,MAAM,MAAKC;EAEjB,MAAM,WAAW,mBAAuD;GACtE,MAAM,YAAY,IAAI,KAAK;GAC3B,IAAI,WAAW;GACf,IAAI,YAAY;GAChB,IAAI,SAAS;AAEb,OAAI;AACF,SAAK,MAAM,MAAM,WACf,KAAI,GAAG,cACL,OAAM,GAAG,cAAc,MAAM,IAAI;IAIrC,MAAM,cAAc,QAAQ,MAAM,KAAK;AAEvC,eAAW,MAAM,OAAO,OAAO,QAAa,YAAY,EAAE;AACxD,UAAK,MAAM,MAAM,WACf,KAAI,GAAG,MACL,OAAM,GAAG,MAAM,KAAgC,MAAM,IAAI;AAG7D;AACA,WAAM;;AAGR,gBAAY;YACL,OAAO;AACd,aAAS;AACT,UAAM;aACE;IACR,MAAM,YAAY,IAAI,KAAK,GAAG;AAC9B,SAAK,MAAM,MAAM,YAAY;AAC3B,SAAI,CAAC,GAAG,aAAc;AAEtB,SAAI,QAAQ;AACV,UAAI;AACF,aAAM,GAAG,aAAa,MAAM;QAAE;QAAU;QAAW;QAAW,EAAE,IAAI;cAC9D;AAGR;;AAGF,WAAM,GAAG,aAAa,MAAM;MAAE;MAAU;MAAW;MAAW,EAAE,IAAI;;;;AAK1E,SAAO,IAAI,oBAAoB,UAAU,CAAC;;CAG5C,MAAM,QAAuB;AAC3B,QAAM,MAAKF,OAAQ,OAAO;;;AAI9B,SAAgB,mBAAmB,SAA4C;AAC7E,QAAO,IAAI,iBAAiB,QAAQ"}

package/package.json

@@ -1,25 +1,34 @@
 {
   "name": "@prisma-next/mongo-runtime",
-  "version": "0.3.0-dev.127",
+  "version": "0.3.0-dev.162",
   "type": "module",
   "sideEffects": false,
   "description": "MongoDB runtime implementation for Prisma Next",
   "dependencies": {
-    "@prisma-next/contract": "0.3.0-dev.127",
-    "@prisma-next/mongo-core": "0.3.0-dev.127",
-    "@prisma-next/runtime-executor": "0.3.0-dev.127"
+    "@prisma-next/contract": "0.3.0-dev.162",
+    "@prisma-next/mongo-lowering": "0.3.0-dev.162",
+    "@prisma-next/mongo-query-ast": "0.3.0-dev.162",
+    "@prisma-next/framework-components": "0.3.0-dev.162"
   },
   "devDependencies": {
     "mongodb": "^6.16.0",
-    "mongodb-memory-server": "^10.4.0",
+    "mongodb-memory-server": "10.4.3",
     "tsdown": "0.18.4",
     "typescript": "5.9.3",
     "vitest": "4.0.17",
-    "@prisma-next/adapter-mongo": "0.3.0-dev.127",
-    "@prisma-next/tsconfig": "0.0.0",
+    "@prisma-next/adapter-mongo": "0.3.0-dev.162",
+    "@prisma-next/middleware-telemetry": "0.3.0-dev.162",
+    "@prisma-next/runtime-executor": "0.3.0-dev.162",
+    "@prisma-next/mongo-contract-ts": "0.3.0-dev.162",
+    "@prisma-next/mongo-pipeline-builder": "0.3.0-dev.162",
+    "@prisma-next/mongo-value": "0.3.0-dev.162",
+    "@prisma-next/family-mongo": "0.3.0-dev.162",
+    "@prisma-next/driver-mongo": "0.3.0-dev.162",
+    "@prisma-next/mongo-contract": "0.3.0-dev.162",
     "@prisma-next/test-utils": "0.0.1",
     "@prisma-next/tsdown": "0.0.0",
-    "@prisma-next/driver-mongo": "0.3.0-dev.127"
+    "@prisma-next/tsconfig": "0.0.0",
+    "@prisma-next/target-mongo": "0.3.0-dev.162"
   },
   "files": [
     "dist",
@@ -35,7 +44,7 @@
   "repository": {
     "type": "git",
     "url": "https://github.com/prisma/prisma-next.git",
-    "directory": "packages/2-mongo-family/5-runtime"
+    "directory": "packages/2-mongo-family/7-runtime"
   },
   "scripts": {
     "build": "tsdown",

package/src/exports/index.ts

@@ -1,2 +1,3 @@
+export type { MongoMiddleware, MongoMiddlewareContext } from '../mongo-middleware';
 export type { MongoRuntime, MongoRuntimeOptions } from '../mongo-runtime';
 export { createMongoRuntime } from '../mongo-runtime';

package/src/mongo-middleware.ts

@@ -0,0 +1,23 @@
+import type {
+  AfterExecuteResult,
+  RuntimeMiddleware,
+  RuntimeMiddlewareContext,
+} from '@prisma-next/framework-components/runtime';
+import type { MongoQueryPlan } from '@prisma-next/mongo-query-ast/execution';
+
+export interface MongoMiddlewareContext extends RuntimeMiddlewareContext {}
+
+export interface MongoMiddleware extends RuntimeMiddleware {
+  readonly familyId: 'mongo';
+  beforeExecute?(plan: MongoQueryPlan, ctx: MongoMiddlewareContext): Promise<void>;
+  onRow?(
+    row: Record<string, unknown>,
+    plan: MongoQueryPlan,
+    ctx: MongoMiddlewareContext,
+  ): Promise<void>;
+  afterExecute?(
+    plan: MongoQueryPlan,
+    result: AfterExecuteResult,
+    ctx: MongoMiddlewareContext,
+  ): Promise<void>;
+}

package/src/mongo-runtime.ts

@@ -1,15 +1,26 @@
 import type {
-  MongoAdapter,
-  MongoDriver,
-  MongoLoweringContext,
-  MongoQueryPlan,
-} from '@prisma-next/mongo-core';
-import { AsyncIterableResult } from '@prisma-next/runtime-executor';
+  RuntimeMiddleware,
+  RuntimeMiddlewareContext,
+} from '@prisma-next/framework-components/runtime';
+import {
+  AsyncIterableResult,
+  checkMiddlewareCompatibility,
+} from '@prisma-next/framework-components/runtime';
+import type { MongoAdapter, MongoDriver } from '@prisma-next/mongo-lowering';
+import type { MongoQueryPlan } from '@prisma-next/mongo-query-ast/execution';
+
+function noop() {}
+function now() {
+  return Date.now();
+}
 
 export interface MongoRuntimeOptions {
   readonly adapter: MongoAdapter;
   readonly driver: MongoDriver;
-  readonly loweringContext: MongoLoweringContext;
+  readonly contract: unknown;
+  readonly targetId: string;
+  readonly middleware?: readonly RuntimeMiddleware[];
+  readonly mode?: 'strict' | 'permissive';
 }
 
 export interface MongoRuntime {
@@ -20,23 +31,82 @@ export interface MongoRuntime {
 class MongoRuntimeImpl implements MongoRuntime {
   readonly #adapter: MongoAdapter;
   readonly #driver: MongoDriver;
-  readonly #loweringContext: MongoLoweringContext;
+  readonly #middleware: readonly RuntimeMiddleware[];
+  readonly #middlewareContext: RuntimeMiddlewareContext;
 
   constructor(options: MongoRuntimeOptions) {
     this.#adapter = options.adapter;
     this.#driver = options.driver;
-    this.#loweringContext = options.loweringContext;
+
+    const middleware = options.middleware ? [...options.middleware] : [];
+    for (const mw of middleware) {
+      checkMiddlewareCompatibility(mw, 'mongo', options.targetId);
+    }
+    this.#middleware = middleware;
+
+    this.#middlewareContext = {
+      contract: options.contract,
+      mode: options.mode ?? 'strict',
+      now,
+      log: { info: noop, warn: noop, error: noop },
+    };
   }
 
   execute<Row>(plan: MongoQueryPlan<Row>): AsyncIterableResult<Row> {
-    const executionPlan = this.#adapter.lower(plan, this.#loweringContext);
-    const iterable = this.#driver.execute<Row>(executionPlan.wireCommand);
+    const adapter = this.#adapter;
+    const driver = this.#driver;
+    const middleware = this.#middleware;
+    const ctx = this.#middlewareContext;
 
-    async function* toGenerator(): AsyncGenerator<Row, void, unknown> {
-      yield* iterable;
-    }
+    const iterator = async function* (): AsyncGenerator<Row, void, unknown> {
+      const startedAt = ctx.now();
+      let rowCount = 0;
+      let completed = false;
+      let failed = false;
+
+      try {
+        for (const mw of middleware) {
+          if (mw.beforeExecute) {
+            await mw.beforeExecute(plan, ctx);
+          }
+        }
+
+        const wireCommand = adapter.lower(plan);
+
+        for await (const row of driver.execute<Row>(wireCommand)) {
+          for (const mw of middleware) {
+            if (mw.onRow) {
+              await mw.onRow(row as Record<string, unknown>, plan, ctx);
+            }
+          }
+          rowCount++;
+          yield row;
+        }
+
+        completed = true;
+      } catch (error) {
+        failed = true;
+        throw error;
+      } finally {
+        const latencyMs = ctx.now() - startedAt;
+        for (const mw of middleware) {
+          if (!mw.afterExecute) continue;
+
+          if (failed) {
+            try {
+              await mw.afterExecute(plan, { rowCount, latencyMs, completed }, ctx);
+            } catch {
+              // Ignore errors from afterExecute during error handling
+            }
+            continue;
+          }
+
+          await mw.afterExecute(plan, { rowCount, latencyMs, completed }, ctx);
+        }
+      }
+    };
 
-    return new AsyncIterableResult(toGenerator());
+    return new AsyncIterableResult(iterator());
   }
 
   async close(): Promise<void> {