@prisma/client-engine-runtime 6.14.0-dev.2 → 6.14.0-dev.21

package/dist/index.js

@@ -43,6 +43,7 @@ __export(index_exports, {
   isPrismaValuePlaceholder: () => isPrismaValuePlaceholder,
   noopTracingHelper: () => noopTracingHelper,
   normalizeJsonProtocolValues: () => normalizeJsonProtocolValues,
+  normalizeRawJsonProtocolResponse: () => normalizeRawJsonProtocolResponse,
   safeJsonStringify: () => safeJsonStringify
 });
 module.exports = __toCommonJS(index_exports);
@@ -318,8 +319,10 @@ function mapValue(value, columnName, resultType, enums) {
       return values.map((v, i) => mapValue(v, `${columnName}[${i}]`, resultType.inner, enums));
     }
     case "Object": {
-      const jsonValue = typeof value === "string" ? value : safeJsonStringify(value);
-      return { $type: "Json", value: jsonValue };
+      return { $type: "Json", value: safeJsonStringify(value) };
+    }
+    case "Json": {
+      return { $type: "Json", value: `${value}` };
     }
     case "Bytes": {
       if (typeof value === "string" && value.startsWith("\\x")) {
@@ -340,7 +343,7 @@ function mapValue(value, columnName, resultType, enums) {
       }
       const enumValue = enumDef[`${value}`];
       if (enumValue === void 0) {
-        throw new DataMapperError(`Unknown enum value '${value}' for enum '${resultType.inner}'`);
+        throw new DataMapperError(`Value '${value}' not found in enum '${resultType.inner}'`);
       }
       return enumValue;
     }
@@ -450,6 +453,16 @@ function rethrowAsUserFacing(error) {
   }
   throw new UserFacingError(message, code, { driverAdapterError: error });
 }
+function rethrowAsUserFacingRawError(error) {
+  if (!(0, import_driver_adapter_utils.isDriverAdapterError)(error)) {
+    throw error;
+  }
+  throw new UserFacingError(
+    `Raw query failed. Code: ${error.cause.originalCode ?? "N/A"}. Message: ${error.cause.originalMessage ?? renderErrorMessage(error)}`,
+    "P2010",
+    { driverAdapterError: error }
+  );
+}
 function getErrorCode(err) {
   switch (err.cause.kind) {
     case "AuthenticationFailed":
@@ -708,17 +721,21 @@ function isPrismaValueBigInt(value) {
 }
 
 // src/interpreter/renderQuery.ts
-function renderQuery(dbQuery, scope, generators) {
+function renderQuery(dbQuery, scope, generators, maxChunkSize) {
   const queryType = dbQuery.type;
+  const params = evaluateParams(dbQuery.params, scope, generators);
   switch (queryType) {
     case "rawSql":
-      return renderRawSql(dbQuery.sql, evaluateParams(dbQuery.params, scope, generators));
-    case "templateSql":
-      return renderTemplateSql(
-        dbQuery.fragments,
-        dbQuery.placeholderFormat,
-        evaluateParams(dbQuery.params, scope, generators)
-      );
+      return [renderRawSql(dbQuery.sql, evaluateParams(dbQuery.params, scope, generators))];
+    case "templateSql": {
+      const chunks = dbQuery.chunkable ? chunkParams(dbQuery.fragments, params, maxChunkSize) : [params];
+      return chunks.map((params2) => {
+        if (maxChunkSize !== void 0 && params2.length > maxChunkSize) {
+          throw new UserFacingError("The query parameter limit supported by your database is exceeded.", "P2029");
+        }
+        return renderTemplateSql(dbQuery.fragments, dbQuery.placeholderFormat, params2);
+      });
+    }
     default:
       assertNever(queryType, `Invalid query type`);
   }
@@ -756,61 +773,36 @@ function evaluateParam(param, scope, generators) {
   return value;
 }
 function renderTemplateSql(fragments, placeholderFormat, params) {
-  let paramIndex = 0;
-  let placeholderNumber = 1;
+  let sql = "";
+  const ctx = { placeholderNumber: 1 };
   const flattenedParams = [];
-  const sql = fragments.map((fragment) => {
-    const fragmentType = fragment.type;
-    switch (fragmentType) {
-      case "parameter":
-        if (paramIndex >= params.length) {
-          throw new Error(`Malformed query template. Fragments attempt to read over ${params.length} parameters.`);
-        }
-        flattenedParams.push(params[paramIndex++]);
-        return formatPlaceholder(placeholderFormat, placeholderNumber++);
-      case "stringChunk":
-        return fragment.chunk;
-      case "parameterTuple": {
-        if (paramIndex >= params.length) {
-          throw new Error(`Malformed query template. Fragments attempt to read over ${params.length} parameters.`);
-        }
-        const paramValue = params[paramIndex++];
-        const paramArray = Array.isArray(paramValue) ? paramValue : [paramValue];
-        const placeholders = paramArray.length == 0 ? "NULL" : paramArray.map((value) => {
-          flattenedParams.push(value);
-          return formatPlaceholder(placeholderFormat, placeholderNumber++);
-        }).join(",");
-        return `(${placeholders})`;
-      }
-      case "parameterTupleList": {
-        if (paramIndex >= params.length) {
-          throw new Error(`Malformed query template. Fragments attempt to read over ${params.length} parameters.`);
-        }
-        const paramValue = params[paramIndex++];
-        if (!Array.isArray(paramValue)) {
-          throw new Error(`Malformed query template. Tuple list expected.`);
-        }
-        if (paramValue.length === 0) {
-          throw new Error(`Malformed query template. Tuple list cannot be empty.`);
-        }
-        const tupleList = paramValue.map((tuple) => {
-          if (!Array.isArray(tuple)) {
-            throw new Error(`Malformed query template. Tuple expected.`);
-          }
-          const elements = tuple.map((value) => {
-            flattenedParams.push(value);
-            return formatPlaceholder(placeholderFormat, placeholderNumber++);
-          }).join(fragment.itemSeparator);
-          return `${fragment.itemPrefix}${elements}${fragment.itemSuffix}`;
-        }).join(fragment.groupSeparator);
-        return tupleList;
-      }
-      default:
-        assertNever(fragmentType, "Invalid fragment type");
-    }
-  }).join("");
+  for (const fragment of pairFragmentsWithParams(fragments, params)) {
+    flattenedParams.push(...flattenedFragmentParams(fragment));
+    sql += renderFragment(fragment, placeholderFormat, ctx);
+  }
   return renderRawSql(sql, flattenedParams);
 }
+function renderFragment(fragment, placeholderFormat, ctx) {
+  const fragmentType = fragment.type;
+  switch (fragmentType) {
+    case "parameter":
+      return formatPlaceholder(placeholderFormat, ctx.placeholderNumber++);
+    case "stringChunk":
+      return fragment.chunk;
+    case "parameterTuple": {
+      const placeholders = fragment.value.length == 0 ? "NULL" : fragment.value.map(() => formatPlaceholder(placeholderFormat, ctx.placeholderNumber++)).join(",");
+      return `(${placeholders})`;
+    }
+    case "parameterTupleList": {
+      return fragment.value.map((tuple) => {
+        const elements = tuple.map(() => formatPlaceholder(placeholderFormat, ctx.placeholderNumber++)).join(fragment.itemSeparator);
+        return `${fragment.itemPrefix}${elements}${fragment.itemSuffix}`;
+      }).join(fragment.groupSeparator);
+    }
+    default:
+      assertNever(fragmentType, "Invalid fragment type");
+  }
+}
 function formatPlaceholder(placeholderFormat, placeholderNumber) {
   return placeholderFormat.hasNumbering ? `${placeholderFormat.prefix}${placeholderNumber}` : placeholderFormat.prefix;
 }
@@ -843,6 +835,143 @@ function toArgType(value) {
 function doesRequireEvaluation(param) {
   return isPrismaValuePlaceholder(param) || isPrismaValueGenerator(param);
 }
+function* pairFragmentsWithParams(fragments, params) {
+  let index = 0;
+  for (const fragment of fragments) {
+    switch (fragment.type) {
+      case "parameter": {
+        if (index >= params.length) {
+          throw new Error(`Malformed query template. Fragments attempt to read over ${params.length} parameters.`);
+        }
+        yield { ...fragment, value: params[index++] };
+        break;
+      }
+      case "stringChunk": {
+        yield fragment;
+        break;
+      }
+      case "parameterTuple": {
+        if (index >= params.length) {
+          throw new Error(`Malformed query template. Fragments attempt to read over ${params.length} parameters.`);
+        }
+        const value = params[index++];
+        yield { ...fragment, value: Array.isArray(value) ? value : [value] };
+        break;
+      }
+      case "parameterTupleList": {
+        if (index >= params.length) {
+          throw new Error(`Malformed query template. Fragments attempt to read over ${params.length} parameters.`);
+        }
+        const value = params[index++];
+        if (!Array.isArray(value)) {
+          throw new Error(`Malformed query template. Tuple list expected.`);
+        }
+        if (value.length === 0) {
+          throw new Error(`Malformed query template. Tuple list cannot be empty.`);
+        }
+        for (const tuple of value) {
+          if (!Array.isArray(tuple)) {
+            throw new Error(`Malformed query template. Tuple expected.`);
+          }
+        }
+        yield { ...fragment, value };
+        break;
+      }
+    }
+  }
+}
+function* flattenedFragmentParams(fragment) {
+  switch (fragment.type) {
+    case "parameter":
+      yield fragment.value;
+      break;
+    case "stringChunk":
+      break;
+    case "parameterTuple":
+      yield* fragment.value;
+      break;
+    case "parameterTupleList":
+      for (const tuple of fragment.value) {
+        yield* tuple;
+      }
+      break;
+  }
+}
+function chunkParams(fragments, params, maxChunkSize) {
+  let totalParamCount = 0;
+  let maxParamsPerFragment = 0;
+  for (const fragment of pairFragmentsWithParams(fragments, params)) {
+    let paramSize = 0;
+    for (const _ of flattenedFragmentParams(fragment)) {
+      void _;
+      paramSize++;
+    }
+    maxParamsPerFragment = Math.max(maxParamsPerFragment, paramSize);
+    totalParamCount += paramSize;
+  }
+  let chunkedParams = [[]];
+  for (const fragment of pairFragmentsWithParams(fragments, params)) {
+    switch (fragment.type) {
+      case "parameter": {
+        for (const params2 of chunkedParams) {
+          params2.push(fragment.value);
+        }
+        break;
+      }
+      case "stringChunk": {
+        break;
+      }
+      case "parameterTuple": {
+        const thisParamCount = fragment.value.length;
+        let chunks = [];
+        if (maxChunkSize && // Have we split the parameters into chunks already?
+        chunkedParams.length === 1 && // Is this the fragment that has the most parameters?
+        thisParamCount === maxParamsPerFragment && // Do we need chunking to fit the parameters?
+        totalParamCount > maxChunkSize && // Would chunking enable us to fit the parameters?
+        totalParamCount - thisParamCount < maxChunkSize) {
+          const availableSize = maxChunkSize - (totalParamCount - thisParamCount);
+          chunks = chunkArray(fragment.value, availableSize);
+        } else {
+          chunks = [fragment.value];
+        }
+        chunkedParams = chunkedParams.flatMap((params2) => chunks.map((chunk) => [...params2, chunk]));
+        break;
+      }
+      case "parameterTupleList": {
+        const thisParamCount = fragment.value.reduce((acc, tuple) => acc + tuple.length, 0);
+        const completeChunks = [];
+        let currentChunk = [];
+        let currentChunkParamCount = 0;
+        for (const tuple of fragment.value) {
+          if (maxChunkSize && // Have we split the parameters into chunks already?
+          chunkedParams.length === 1 && // Is this the fragment that has the most parameters?
+          thisParamCount === maxParamsPerFragment && // Is there anything in the current chunk?
+          currentChunk.length > 0 && // Will adding this tuple exceed the max chunk size?
+          totalParamCount - thisParamCount + currentChunkParamCount + tuple.length > maxChunkSize) {
+            completeChunks.push(currentChunk);
+            currentChunk = [];
+            currentChunkParamCount = 0;
+          }
+          currentChunk.push(tuple);
+          currentChunkParamCount += tuple.length;
+        }
+        if (currentChunk.length > 0) {
+          completeChunks.push(currentChunk);
+        }
+        chunkedParams = chunkedParams.flatMap((params2) => completeChunks.map((chunk) => [...params2, chunk]));
+        break;
+      }
+    }
+  }
+  return chunkedParams;
+}
+function chunkArray(array, chunkSize) {
+  const result = [];
+  for (let i = 0; i < array.length; i += chunkSize) {
+    result.push(array.slice(i, i + chunkSize));
+  }
+  return result;
+}
 
 // src/interpreter/serializeSql.ts
 var import_driver_adapter_utils2 = require("@prisma/driver-adapter-utils");
@@ -875,29 +1004,90 @@ function serializeSql(resultSet) {
   );
 }
 function serializeRawSql(resultSet) {
-  const types = resultSet.columnTypes.map((type) => serializeColumnType(type));
-  const mappers = types.map((type) => {
-    switch (type) {
-      case "bytes":
-        return (value) => Array.isArray(value) ? new Uint8Array(value) : value;
-      case "int":
-        return (value) => value === null ? null : typeof value === "number" ? value : parseInt(`${value}`, 10);
-      case "bigint":
-        return (value) => value === null ? null : typeof value === "bigint" ? value : BigInt(`${value}`);
-      case "json":
-        return (value) => typeof value === "string" ? JSON.parse(value) : value;
-      case "bool":
-        return (value) => typeof value === "string" ? value === "true" || value === "1" : typeof value === "number" ? value === 1 : value;
-      default:
-        return (value) => value;
-    }
-  });
   return {
     columns: resultSet.columnNames,
     types: resultSet.columnTypes.map((type) => serializeColumnType(type)),
-    rows: resultSet.rows.map((row) => row.map((value, index) => mappers[index](value)))
+    rows: resultSet.rows.map(
+      (row) => row.map((value, index) => serializeRawValue(value, resultSet.columnTypes[index]))
+    )
   };
 }
+function serializeRawValue(value, type) {
+  if (value === null) {
+    return null;
+  }
+  switch (type) {
+    case import_driver_adapter_utils2.ColumnTypeEnum.Int32:
+      switch (typeof value) {
+        case "number":
+          return Math.trunc(value);
+        case "string":
+          return Math.trunc(Number(value));
+        default:
+          throw new Error(`Cannot serialize value of type ${typeof value} as Int32`);
+      }
+    case import_driver_adapter_utils2.ColumnTypeEnum.Int32Array:
+      if (!Array.isArray(value)) {
+        throw new Error(`Cannot serialize value of type ${typeof value} as Int32Array`);
+      }
+      return value.map((v) => serializeRawValue(v, import_driver_adapter_utils2.ColumnTypeEnum.Int32));
+    case import_driver_adapter_utils2.ColumnTypeEnum.Int64:
+      switch (typeof value) {
+        case "number":
+          return BigInt(Math.trunc(value));
+        case "string":
+          return value;
+        default:
+          throw new Error(`Cannot serialize value of type ${typeof value} as Int64`);
+      }
+    case import_driver_adapter_utils2.ColumnTypeEnum.Int64Array:
+      if (!Array.isArray(value)) {
+        throw new Error(`Cannot serialize value of type ${typeof value} as Int64Array`);
+      }
+      return value.map((v) => serializeRawValue(v, import_driver_adapter_utils2.ColumnTypeEnum.Int64));
+    case import_driver_adapter_utils2.ColumnTypeEnum.Json:
+      switch (typeof value) {
+        case "string":
+          return JSON.parse(value);
+        default:
+          throw new Error(`Cannot serialize value of type ${typeof value} as Json`);
+      }
+    case import_driver_adapter_utils2.ColumnTypeEnum.JsonArray:
+      if (!Array.isArray(value)) {
+        throw new Error(`Cannot serialize value of type ${typeof value} as JsonArray`);
+      }
+      return value.map((v) => serializeRawValue(v, import_driver_adapter_utils2.ColumnTypeEnum.Json));
+    case import_driver_adapter_utils2.ColumnTypeEnum.Bytes:
+      if (Array.isArray(value)) {
+        return new Uint8Array(value);
+      } else {
+        throw new Error(`Cannot serialize value of type ${typeof value} as Bytes`);
+      }
+    case import_driver_adapter_utils2.ColumnTypeEnum.BytesArray:
+      if (!Array.isArray(value)) {
+        throw new Error(`Cannot serialize value of type ${typeof value} as BytesArray`);
+      }
+      return value.map((v) => serializeRawValue(v, import_driver_adapter_utils2.ColumnTypeEnum.Bytes));
+    case import_driver_adapter_utils2.ColumnTypeEnum.Boolean:
+      switch (typeof value) {
+        case "boolean":
+          return value;
+        case "string":
+          return value === "true" || value === "1";
+        case "number":
+          return value === 1;
+        default:
+          throw new Error(`Cannot serialize value of type ${typeof value} as Boolean`);
+      }
+    case import_driver_adapter_utils2.ColumnTypeEnum.BooleanArray:
+      if (!Array.isArray(value)) {
+        throw new Error(`Cannot serialize value of type ${typeof value} as BooleanArray`);
+      }
+      return value.map((v) => serializeRawValue(v, import_driver_adapter_utils2.ColumnTypeEnum.Boolean));
+    default:
+      return value;
+  }
+}
 function serializeColumnType(columnType) {
   switch (columnType) {
     case import_driver_adapter_utils2.ColumnTypeEnum.Int32:
@@ -1052,6 +1242,7 @@ var QueryInterpreter = class _QueryInterpreter {
   #serializer;
   #rawSerializer;
   #provider;
+  #connectioInfo;
   constructor({
     transactionManager,
     placeholderValues,
@@ -1059,7 +1250,8 @@ var QueryInterpreter = class _QueryInterpreter {
     tracingHelper,
     serializer,
     rawSerializer,
-    provider
+    provider,
+    connectionInfo
   }) {
     this.#transactionManager = transactionManager;
     this.#placeholderValues = placeholderValues;
@@ -1068,6 +1260,7 @@ var QueryInterpreter = class _QueryInterpreter {
     this.#serializer = serializer;
     this.#rawSerializer = rawSerializer ?? serializer;
     this.#provider = provider;
+    this.#connectioInfo = connectionInfo;
   }
   static forSql(options) {
     return new _QueryInterpreter({
@@ -1077,7 +1270,8 @@ var QueryInterpreter = class _QueryInterpreter {
       tracingHelper: options.tracingHelper,
       serializer: serializeSql,
       rawSerializer: serializeRawSql,
-      provider: options.provider
+      provider: options.provider,
+      connectionInfo: options.connectionInfo
     });
   }
   async run(queryPlan, queryable) {
@@ -1138,21 +1332,41 @@ var QueryInterpreter = class _QueryInterpreter {
         };
       }
       case "execute": {
-        const query = renderQuery(node.args, scope, generators);
-        return this.#withQuerySpanAndEvent(query, queryable, async () => {
-          return { value: await queryable.executeRaw(query) };
-        });
+        const queries = renderQuery(node.args, scope, generators, this.#maxChunkSize());
+        let sum = 0;
+        for (const query of queries) {
+          sum += await this.#withQuerySpanAndEvent(
+            query,
+            queryable,
+            () => queryable.executeRaw(query).catch(
+              (err) => node.args.type === "rawSql" ? rethrowAsUserFacingRawError(err) : rethrowAsUserFacing(err)
+            )
+          );
+        }
+        return { value: sum };
       }
       case "query": {
-        const query = renderQuery(node.args, scope, generators);
-        return this.#withQuerySpanAndEvent(query, queryable, async () => {
-          const result = await queryable.queryRaw(query);
-          if (node.args.type === "rawSql") {
-            return { value: this.#rawSerializer(result), lastInsertId: result.lastInsertId };
+        const queries = renderQuery(node.args, scope, generators, this.#maxChunkSize());
+        let results;
+        for (const query of queries) {
+          const result = await this.#withQuerySpanAndEvent(
+            query,
+            queryable,
+            () => queryable.queryRaw(query).catch(
+              (err) => node.args.type === "rawSql" ? rethrowAsUserFacingRawError(err) : rethrowAsUserFacing(err)
+            )
+          );
+          if (results === void 0) {
+            results = result;
           } else {
-            return { value: this.#serializer(result), lastInsertId: result.lastInsertId };
+            results.rows.push(...result.rows);
+            results.lastInsertId = result.lastInsertId;
           }
-        });
+        }
+        return {
+          value: node.args.type === "rawSql" ? this.#rawSerializer(results) : this.#serializer(results),
+          lastInsertId: results?.lastInsertId
+        };
       }
       case "reverse": {
         const { value, lastInsertId } = await this.interpretNode(node.args, queryable, scope, generators);
@@ -1289,6 +1503,34 @@ var QueryInterpreter = class _QueryInterpreter {
         assertNever(node, `Unexpected node type: ${node.type}`);
     }
   }
+  #maxChunkSize() {
+    if (this.#connectioInfo?.maxBindValues !== void 0) {
+      return this.#connectioInfo.maxBindValues;
+    }
+    return this.#providerMaxChunkSize();
+  }
+  #providerMaxChunkSize() {
+    if (this.#provider === void 0) {
+      return void 0;
+    }
+    switch (this.#provider) {
+      case "cockroachdb":
+      case "postgres":
+      case "postgresql":
+      case "prisma+postgres":
+        return 32766;
+      case "mysql":
+        return 65535;
+      case "sqlite":
+        return 999;
+      case "sqlserver":
+        return 2098;
+      case "mongodb":
+        return void 0;
+      default:
+        assertNever(this.#provider, `Unexpected provider: ${this.#provider}`);
+    }
+  }
   #withQuerySpanAndEvent(query, queryable, execute) {
     return withQuerySpanAndEvent({
       query,
@@ -1458,6 +1700,35 @@ function mapObjectValues(object, mapper) {
   return result;
 }
 
+// src/raw-json-protocol.ts
+var import_decimal4 = __toESM(require("decimal.js"));
+function normalizeRawJsonProtocolResponse(response) {
+  for (let i = 0; i < response.rows.length; i++) {
+    const row = response.rows[i];
+    for (let j = 0; j < row.length; j++) {
+      row[j] = normalizeValue(response.types[j], row[j]);
+    }
+  }
+  return response;
+}
+function normalizeValue(type, value) {
+  if (value === null) {
+    return value;
+  }
+  switch (type) {
+    case "bigint":
+      return String(BigInt(value));
+    case "decimal":
+      return String(new import_decimal4.default(value));
+    case "bigint-array":
+      return value.map((v) => normalizeValue("bigint", v));
+    case "decimal-array":
+      return value.map((v) => normalizeValue("decimal", v));
+    default:
+      return value;
+  }
+}
+
 // src/transactionManager/TransactionManager.ts
 var import_debug = require("@prisma/debug");
 
@@ -1734,5 +2005,6 @@ var TransactionManager = class {
   isPrismaValuePlaceholder,
   noopTracingHelper,
   normalizeJsonProtocolValues,
+  normalizeRawJsonProtocolResponse,
   safeJsonStringify
 });