envio 2.26.0-alpha.9 → 2.26.0-rc.0

package/src/db/Table.res.js

@@ -72,10 +72,11 @@ function getFieldType(field) {
         );
 }
 
-function mkTable(tableName, compositeIndicesOpt, fields) {
+function mkTable(tableName, schemaName, compositeIndicesOpt, fields) {
   var compositeIndices = compositeIndicesOpt !== undefined ? compositeIndicesOpt : [];
   return {
           tableName: tableName,
+          schemaName: schemaName,
           fields: fields,
           compositeIndices: compositeIndices
         };
@@ -183,7 +184,7 @@ function getUnfilteredCompositeIndicesUnsafe(table) {
               }));
 }
 
-function toSqlParams(table, schema, pgSchema) {
+function toSqlParams(table, schema) {
   var quotedFieldNames = [];
   var quotedNonPrimaryFieldNames = [];
   var arrayFieldTypes = [];
@@ -261,7 +262,7 @@ function toSqlParams(table, schema, pgSchema) {
                   var fieldType = f.fieldType;
                   tmp = fieldType === "TIMESTAMP" || fieldType === "TIMESTAMP WITH TIME ZONE" || fieldType === "JSONB" || fieldType === "SERIAL" || fieldType === "TEXT" || fieldType === "DOUBLE PRECISION" || fieldType === "NUMERIC" || fieldType === "BOOLEAN" || fieldType === "INTEGER" || fieldType === "TIMESTAMP WITH TIME ZONE NULL" ? (
                       fieldType === "BOOLEAN" ? "INTEGER[]::" + f.fieldType : fieldType
-                    ) : "TEXT[]::\"" + pgSchema + "\"." + fieldType;
+                    ) : "TEXT[]::" + fieldType;
                 } else {
                   tmp = "TEXT";
                 }
@@ -301,6 +302,43 @@ function getCompositeIndices(table) {
               }));
 }
 
+function makeBatchSetFnString(table) {
+  var fieldNamesInQuotes = Belt_Array.map(getNonDefaultFieldNames(table), (function (fieldName) {
+          return "\"" + fieldName + "\"";
+        }));
+  return "(sql, rows) => {\n      return sql\`\n        INSERT INTO \"" + table.schemaName + "\".\"" + table.tableName + "\"\n        \${sql(rows, " + fieldNamesInQuotes.join(", ") + ")}\n        ON CONFLICT(" + getPrimaryKeyFieldNames(table).join(", ") + ") DO UPDATE\n        SET\n        " + Belt_Array.map(fieldNamesInQuotes, (function (fieldNameInQuotes) {
+                  return fieldNameInQuotes + " = EXCLUDED." + fieldNameInQuotes;
+                })).join(", ") + ";\`\n    }";
+}
+
+function chunkBatchQuery(sql, entityDataArray, queryToExecute, maxItemsPerQueryOpt) {
+  var maxItemsPerQuery = maxItemsPerQueryOpt !== undefined ? maxItemsPerQueryOpt : 500;
+  var responses = [];
+  var i = 0;
+  while(i < entityDataArray.length) {
+    var chunk = entityDataArray.slice(i, i + maxItemsPerQuery | 0);
+    var response = queryToExecute(sql, chunk);
+    responses.push(response);
+    i = i + maxItemsPerQuery | 0;
+  };
+  return Promise.all(responses);
+}
+
+function makeBatchSetFn(table, schema) {
+  var batchSetFn = eval(makeBatchSetFnString(table));
+  var parseOrThrow = S$RescriptSchema.compile(S$RescriptSchema.array(schema), "Output", "Json", "Sync", true);
+  return async function (sql, rows) {
+    var rowsJson = parseOrThrow(rows);
+    await chunkBatchQuery(sql, rowsJson, batchSetFn, undefined);
+  };
+}
+
+var PostgresInterop = {
+  makeBatchSetFnString: makeBatchSetFnString,
+  chunkBatchQuery: chunkBatchQuery,
+  makeBatchSetFn: makeBatchSetFn
+};
+
 exports.mkField = mkField;
 exports.mkDerivedFromField = mkDerivedFromField;
 exports.getUserDefinedFieldName = getUserDefinedFieldName;
@@ -323,4 +361,5 @@ exports.getUnfilteredCompositeIndicesUnsafe = getUnfilteredCompositeIndicesUnsaf
 exports.toSqlParams = toSqlParams;
 exports.getSingleIndices = getSingleIndices;
 exports.getCompositeIndices = getCompositeIndices;
+exports.PostgresInterop = PostgresInterop;
 /* BigInt Not a pure module */