feat(client): enable ITX support for DataProxy (#16005)

Closes prisma/client-planning#92
Closes #16177

Co-authored-by: Daniel Starns <danielstarns@hotmail.com>
Co-authored-by: Sergey Tatarintsev <tatarintsev@prisma.io>

packages/client/package.json

@@ -79,7 +79,7 @@
     "@prisma/instrumentation": "workspace:*",
     "@prisma/internals": "workspace:*",
     "@prisma/migrate": "workspace:*",
-    "@prisma/mini-proxy": "0.2.0",
+    "@prisma/mini-proxy": "0.3.0",
     "@swc-node/register": "1.5.4",
     "@swc/core": "1.3.14",
     "@swc/jest": "0.2.23",

packages/client/src/runtime/RequestHandler.ts

@@ -69,7 +69,7 @@ function getRequestInfo(request: Request) {
   }
 
   return {
-    batchTransaction: transaction?.kind === 'batch' ? transaction : undefined,
+    transaction,
     headers,
   }
 }
@@ -92,13 +92,16 @@ export class RequestHandler {
         // TODO: pass the child information to QE for it to issue links to queries
         // const links = requests.map((r) => trace.getSpanContext(r.otelChildCtx!))
 
-        return this.client._engine.requestBatch(queries, info.headers, info.batchTransaction)
+        const batchTransaction = info.transaction?.kind === 'batch' ? info.transaction : undefined
+
+        return this.client._engine.requestBatch(queries, info.headers, batchTransaction)
       },
       singleLoader: (request) => {
         const info = getRequestInfo(request)
         const query = String(request.document)
+        const interactiveTransaction = info.transaction?.kind === 'itx' ? info.transaction : undefined
 
-        return this.client._engine.request(query, info.headers)
+        return this.client._engine.request(query, info.headers, interactiveTransaction)
       },
       batchBy: (request) => {
         if (request.transaction?.id) {

packages/client/src/runtime/core/request/PrismaPromise.ts

@@ -9,6 +9,7 @@ export type PrismaPromiseBatchTransaction = {
 export type PrismaPromiseInteractiveTransaction = {
   kind: 'itx'
   id: string
+  payload: unknown
 }
 
 export type PrismaPromiseTransaction = PrismaPromiseBatchTransaction | PrismaPromiseInteractiveTransaction

packages/client/src/runtime/getPrismaClient.ts

@@ -974,7 +974,7 @@ new PrismaClient({
       let result: unknown
       try {
         // execute user logic with a proxied the client
-        result = await callback(transactionProxy(this, { id: info.id }))
+        result = await callback(transactionProxy(this, { id: info.id, payload: info.payload }))
 
         // it went well, then we commit the transaction
         await this._engine.transaction('commit', headers, info)

packages/client/tests/functional/interactive-transactions/__snapshots__/tests.ts.snap

@@ -13,8 +13,8 @@ exports[`interactive-transactions (provider=cockroachdb) batching rollback 1`] =
 Invalid \`prisma.user.create()\` invocation in
 /client/tests/functional/interactive-transactions/tests.ts:0:0
 
-  XX  */
-  XX testIf(getClientEngineType() === ClientEngineType.Library)('batching rollback', async () => {
+  XX 'batching rollback',
+  XX async () => {
   XX   const result = prisma.$transaction([
 → XX     prisma.user.create(
 Unique constraint failed on the fields: (\`email\`)
@@ -37,8 +37,8 @@ exports[`interactive-transactions (provider=mongodb) batching rollback 1`] = `
 Invalid \`prisma.user.create()\` invocation in
 /client/tests/functional/interactive-transactions/tests.ts:0:0
 
-  XX  */
-  XX testIf(getClientEngineType() === ClientEngineType.Library)('batching rollback', async () => {
+  XX 'batching rollback',
+  XX async () => {
   XX   const result = prisma.$transaction([
 → XX     prisma.user.create(
 Unique constraint failed on the constraint: \`User_email_key\`
@@ -69,8 +69,8 @@ exports[`interactive-transactions (provider=mysql) batching rollback 1`] = `
 Invalid \`prisma.user.create()\` invocation in
 /client/tests/functional/interactive-transactions/tests.ts:0:0
 
-  XX  */
-  XX testIf(getClientEngineType() === ClientEngineType.Library)('batching rollback', async () => {
+  XX 'batching rollback',
+  XX async () => {
   XX   const result = prisma.$transaction([
 → XX     prisma.user.create(
 Unique constraint failed on the constraint: \`User_email_key\`
@@ -101,8 +101,8 @@ exports[`interactive-transactions (provider=postgresql) batching rollback 1`] =
 Invalid \`prisma.user.create()\` invocation in
 /client/tests/functional/interactive-transactions/tests.ts:0:0
 
-  XX  */
-  XX testIf(getClientEngineType() === ClientEngineType.Library)('batching rollback', async () => {
+  XX 'batching rollback',
+  XX async () => {
   XX   const result = prisma.$transaction([
 → XX     prisma.user.create(
 Unique constraint failed on the fields: (\`email\`)
@@ -133,8 +133,8 @@ exports[`interactive-transactions (provider=sqlite) batching rollback 1`] = `
 Invalid \`prisma.user.create()\` invocation in
 /client/tests/functional/interactive-transactions/tests.ts:0:0
 
-  XX  */
-  XX testIf(getClientEngineType() === ClientEngineType.Library)('batching rollback', async () => {
+  XX 'batching rollback',
+  XX async () => {
   XX   const result = prisma.$transaction([
 → XX     prisma.user.create(
 Unique constraint failed on the fields: (\`email\`)
@@ -165,8 +165,8 @@ exports[`interactive-transactions (provider=sqlserver) batching rollback 1`] = `
 Invalid \`prisma.user.create()\` invocation in
 /client/tests/functional/interactive-transactions/tests.ts:0:0
 
-  XX  */
-  XX testIf(getClientEngineType() === ClientEngineType.Library)('batching rollback', async () => {
+  XX 'batching rollback',
+  XX async () => {
   XX   const result = prisma.$transaction([
 → XX     prisma.user.create(
 Unique constraint failed on the constraint: \`dbo.User\`