@xata.io/client 0.20.1 → 0.20.2

package/dist/index.d.ts

@@ -1686,18 +1686,16 @@ declare type AggResponse$1 = (number | null) | {
  * A transaction operation
  */
 declare type TransactionOperation = {
-    insert: TransactionInsert;
+    insert: TransactionInsertOp;
 } | {
-    update: TransactionUpdate;
+    update: TransactionUpdateOp;
 } | {
-    ['delete']: TransactionDelete;
+    ['delete']: TransactionDeleteOp;
 };
 /**
  * Insert operation
- *
- * @x-go-type TxOperation
  */
-declare type TransactionInsert = {
+declare type TransactionInsertOp = {
     /**
      * The table name
      */
@@ -1726,10 +1724,8 @@ declare type TransactionInsert = {
 };
 /**
  * Update operation
- *
- * @x-go-type TxOperation
  */
-declare type TransactionUpdate = {
+declare type TransactionUpdateOp = {
     /**
      * The table name
      */
@@ -1752,10 +1748,8 @@ declare type TransactionUpdate = {
 };
 /**
  * A delete operation. The transaction will continue if no record matches the ID.
- *
- * @x-go-type TxOperation
  */
-declare type TransactionDelete = {
+declare type TransactionDeleteOp = {
     /**
      * The table name
      */
@@ -1766,6 +1760,10 @@ declare type TransactionDelete = {
  * A result from an insert operation.
  */
 declare type TransactionResultInsert = {
+    /**
+     * The type of operation who's result is being returned.
+     */
+    operation: 'insert';
     /**
      * The number of affected rows
      */
@@ -1777,7 +1775,11 @@ declare type TransactionResultInsert = {
  */
 declare type TransactionResultUpdate = {
     /**
-     * The number of affected rows
+     * The type of operation who's result is being returned.
+     */
+    operation: 'update';
+    /**
+     * The number of updated rows
      */
     rows: number;
     id: RecordID;
@@ -1787,12 +1789,18 @@ declare type TransactionResultUpdate = {
  */
 declare type TransactionResultDelete = {
     /**
-     * The number of affected rows
+     * The type of operation who's result is being returned.
+     */
+    operation: 'delete';
+    /**
+     * The number of deleted rows
      */
     rows: number;
 };
 /**
  * An error message from a failing transaction operation
+ *
+ * @x-go-type xata.ErrTxOp
  */
 declare type TransactionError = {
     /**
@@ -1881,18 +1889,18 @@ declare type SearchResponse = {
     warning?: string;
 };
 /**
- * @x-go-type TxResponse
+ * @x-go-type TxSuccess
  */
-declare type TransactionSucceeded = {
+declare type TransactionSuccess = {
     /**
      * An ordered array of results from the submitted operations that were executed
      */
     results: (TransactionResultInsert | TransactionResultUpdate | TransactionResultDelete)[];
 };
 /**
- * @x-go-type TxResponse
+ * @x-go-type TxFailure
  */
-declare type TransactionFailed = {
+declare type TransactionFailure = {
     /**
      * An array of errors from the submitted operations.
      */
@@ -2569,7 +2577,7 @@ declare type BranchTransactionPathParams = {
 };
 declare type BranchTransactionError = ErrorWrapper<{
     status: 400;
-    payload: TransactionFailed;
+    payload: TransactionFailure;
 } | {
     status: 401;
     payload: AuthError;
@@ -2584,7 +2592,7 @@ declare type BranchTransactionVariables = {
     body: BranchTransactionRequestBody;
     pathParams: BranchTransactionPathParams;
 } & DataPlaneFetcherExtraProps;
-declare const branchTransaction: (variables: BranchTransactionVariables, signal?: AbortSignal) => Promise<TransactionSucceeded>;
+declare const branchTransaction: (variables: BranchTransactionVariables, signal?: AbortSignal) => Promise<TransactionSuccess>;
 declare type QueryMigrationRequestsPathParams = {
     /**
      * The Database Name
@@ -4733,7 +4741,7 @@ declare const operationsByTag: {
         resolveBranch: (variables: ResolveBranchVariables, signal?: AbortSignal | undefined) => Promise<ResolveBranchResponse>;
     };
     records: {
-        branchTransaction: (variables: BranchTransactionVariables, signal?: AbortSignal | undefined) => Promise<TransactionSucceeded>;
+        branchTransaction: (variables: BranchTransactionVariables, signal?: AbortSignal | undefined) => Promise<TransactionSuccess>;
         insertRecord: (variables: InsertRecordVariables, signal?: AbortSignal | undefined) => Promise<RecordUpdateResponse>;
         getRecord: (variables: GetRecordVariables, signal?: AbortSignal | undefined) => Promise<XataRecord$1>;
         insertRecordWithID: (variables: InsertRecordWithIDVariables, signal?: AbortSignal | undefined) => Promise<RecordUpdateResponse>;
@@ -4855,8 +4863,8 @@ type responses_SchemaUpdateResponse = SchemaUpdateResponse;
 type responses_SummarizeResponse = SummarizeResponse;
 type responses_AggResponse = AggResponse;
 type responses_SearchResponse = SearchResponse;
-type responses_TransactionSucceeded = TransactionSucceeded;
-type responses_TransactionFailed = TransactionFailed;
+type responses_TransactionSuccess = TransactionSuccess;
+type responses_TransactionFailure = TransactionFailure;
 declare namespace responses {
   export {
     responses_AuthError as AuthError,
@@ -4873,8 +4881,8 @@ declare namespace responses {
     responses_SummarizeResponse as SummarizeResponse,
     responses_AggResponse as AggResponse,
     responses_SearchResponse as SearchResponse,
-    responses_TransactionSucceeded as TransactionSucceeded,
-    responses_TransactionFailed as TransactionFailed,
+    responses_TransactionSuccess as TransactionSuccess,
+    responses_TransactionFailure as TransactionFailure,
   };
 }
 
@@ -4953,9 +4961,9 @@ type schemas_RecordID = RecordID;
 type schemas_TableRename = TableRename;
 type schemas_RecordsMetadata = RecordsMetadata;
 type schemas_TransactionOperation = TransactionOperation;
-type schemas_TransactionInsert = TransactionInsert;
-type schemas_TransactionUpdate = TransactionUpdate;
-type schemas_TransactionDelete = TransactionDelete;
+type schemas_TransactionInsertOp = TransactionInsertOp;
+type schemas_TransactionUpdateOp = TransactionUpdateOp;
+type schemas_TransactionDeleteOp = TransactionDeleteOp;
 type schemas_TransactionResultInsert = TransactionResultInsert;
 type schemas_TransactionResultUpdate = TransactionResultUpdate;
 type schemas_TransactionResultDelete = TransactionResultDelete;
@@ -5058,9 +5066,9 @@ declare namespace schemas {
     schemas_RecordsMetadata as RecordsMetadata,
     AggResponse$1 as AggResponse,
     schemas_TransactionOperation as TransactionOperation,
-    schemas_TransactionInsert as TransactionInsert,
-    schemas_TransactionUpdate as TransactionUpdate,
-    schemas_TransactionDelete as TransactionDelete,
+    schemas_TransactionInsertOp as TransactionInsertOp,
+    schemas_TransactionUpdateOp as TransactionUpdateOp,
+    schemas_TransactionDeleteOp as TransactionDeleteOp,
     schemas_TransactionResultInsert as TransactionResultInsert,
     schemas_TransactionResultUpdate as TransactionResultUpdate,
     schemas_TransactionResultDelete as TransactionResultDelete,
@@ -5411,6 +5419,13 @@ declare class RecordsApi {
         records: Record<string, any>[];
         columns?: ColumnsProjection;
     }): Promise<BulkInsertResponse>;
+    branchTransaction({ workspace, region, database, branch, operations }: {
+        workspace: WorkspaceID;
+        region: string;
+        database: DBName;
+        branch: BranchName;
+        operations: TransactionOperation[];
+    }): Promise<TransactionSuccess>;
 }
 declare class SearchAndFilterApi {
     private extraProps;

package/dist/index.mjs

@@ -60,6 +60,13 @@ function deepMerge(a, b) {
   }
   return result;
 }
+function chunk(array, chunkSize) {
+  const result = [];
+  for (let i = 0; i < array.length; i += chunkSize) {
+    result.push(array.slice(i, i + chunkSize));
+  }
+  return result;
+}
 
 function getEnvironment() {
   try {
@@ -183,7 +190,7 @@ function getFetchImplementation(userFetch) {
   return fetchImpl;
 }
 
-const VERSION = "0.20.1";
+const VERSION = "0.20.2";
 
 class ErrorWithCause extends Error {
   constructor(message, options) {
@@ -1366,6 +1373,19 @@ class RecordsApi {
       ...this.extraProps
     });
   }
+  branchTransaction({
+    workspace,
+    region,
+    database,
+    branch,
+    operations
+  }) {
+    return operationsByTag.records.branchTransaction({
+      pathParams: { workspace, region, dbBranchName: `${database}:${branch}` },
+      body: { operations },
+      ...this.extraProps
+    });
+  }
 }
 class SearchAndFilterApi {
   constructor(extraProps) {
@@ -2131,7 +2151,8 @@ var __privateMethod$2 = (obj, member, method) => {
   __accessCheck$4(obj, member, "access private method");
   return method;
 };
-var _table, _getFetchProps, _db, _cache, _schemaTables$2, _trace, _insertRecordWithoutId, insertRecordWithoutId_fn, _insertRecordWithId, insertRecordWithId_fn, _bulkInsertTableRecords, bulkInsertTableRecords_fn, _updateRecordWithID, updateRecordWithID_fn, _upsertRecordWithID, upsertRecordWithID_fn, _deleteRecord, deleteRecord_fn, _setCacheQuery, setCacheQuery_fn, _getCacheQuery, getCacheQuery_fn, _getSchemaTables$1, getSchemaTables_fn$1;
+var _table, _getFetchProps, _db, _cache, _schemaTables$2, _trace, _insertRecordWithoutId, insertRecordWithoutId_fn, _insertRecordWithId, insertRecordWithId_fn, _insertRecords, insertRecords_fn, _updateRecordWithID, updateRecordWithID_fn, _updateRecords, updateRecords_fn, _upsertRecordWithID, upsertRecordWithID_fn, _deleteRecord, deleteRecord_fn, _deleteRecords, deleteRecords_fn, _setCacheQuery, setCacheQuery_fn, _getCacheQuery, getCacheQuery_fn, _getSchemaTables$1, getSchemaTables_fn$1;
+const BULK_OPERATION_MAX_SIZE = 1e3;
 class Repository extends Query {
 }
 class RestRepository extends Query {
@@ -2143,10 +2164,12 @@ class RestRepository extends Query {
     );
     __privateAdd$4(this, _insertRecordWithoutId);
     __privateAdd$4(this, _insertRecordWithId);
-    __privateAdd$4(this, _bulkInsertTableRecords);
+    __privateAdd$4(this, _insertRecords);
     __privateAdd$4(this, _updateRecordWithID);
+    __privateAdd$4(this, _updateRecords);
     __privateAdd$4(this, _upsertRecordWithID);
     __privateAdd$4(this, _deleteRecord);
+    __privateAdd$4(this, _deleteRecords);
     __privateAdd$4(this, _setCacheQuery);
     __privateAdd$4(this, _getCacheQuery);
     __privateAdd$4(this, _getSchemaTables$1);
@@ -2180,20 +2203,22 @@ class RestRepository extends Query {
       if (Array.isArray(a)) {
         if (a.length === 0)
           return [];
-        const columns = isStringArray(b) ? b : void 0;
-        return __privateMethod$2(this, _bulkInsertTableRecords, bulkInsertTableRecords_fn).call(this, a, columns);
+        const ids = await __privateMethod$2(this, _insertRecords, insertRecords_fn).call(this, a, { ifVersion, createOnly: true });
+        const columns = isStringArray(b) ? b : ["*"];
+        const result = await this.read(ids, columns);
+        return result;
       }
       if (isString(a) && isObject(b)) {
         if (a === "")
           throw new Error("The id can't be empty");
         const columns = isStringArray(c) ? c : void 0;
-        return __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a, b, columns, { createOnly: true, ifVersion });
+        return await __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a, b, columns, { createOnly: true, ifVersion });
       }
       if (isObject(a) && isString(a.id)) {
         if (a.id === "")
           throw new Error("The id can't be empty");
         const columns = isStringArray(b) ? b : void 0;
-        return __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a.id, { ...a, id: void 0 }, columns, { createOnly: true, ifVersion });
+        return await __privateMethod$2(this, _insertRecordWithId, insertRecordWithId_fn).call(this, a.id, { ...a, id: void 0 }, columns, { createOnly: true, ifVersion });
       }
       if (isObject(a)) {
         const columns = isStringArray(b) ? b : void 0;
@@ -2268,11 +2293,15 @@ class RestRepository extends Query {
       if (Array.isArray(a)) {
         if (a.length === 0)
           return [];
-        if (a.length > 100) {
-          console.warn("Bulk update operation is not optimized in the Xata API yet, this request might be slow");
-        }
+        const existing = await this.read(a, ["id"]);
+        const updates = a.filter((_item, index) => existing[index] !== null);
+        await __privateMethod$2(this, _updateRecords, updateRecords_fn).call(this, updates, {
+          ifVersion,
+          upsert: false
+        });
         const columns = isStringArray(b) ? b : ["*"];
-        return Promise.all(a.map((object) => this.update(object, columns)));
+        const result = await this.read(a, columns);
+        return result;
       }
       if (isString(a) && isObject(b)) {
         const columns = isStringArray(c) ? c : void 0;
@@ -2310,11 +2339,13 @@ class RestRepository extends Query {
       if (Array.isArray(a)) {
         if (a.length === 0)
           return [];
-        if (a.length > 100) {
-          console.warn("Bulk update operation is not optimized in the Xata API yet, this request might be slow");
-        }
+        await __privateMethod$2(this, _updateRecords, updateRecords_fn).call(this, a, {
+          ifVersion,
+          upsert: true
+        });
         const columns = isStringArray(b) ? b : ["*"];
-        return Promise.all(a.map((object) => this.createOrUpdate(object, columns)));
+        const result = await this.read(a, columns);
+        return result;
       }
       if (isString(a) && isObject(b)) {
         const columns = isStringArray(c) ? c : void 0;
@@ -2333,8 +2364,10 @@ class RestRepository extends Query {
       if (Array.isArray(a)) {
         if (a.length === 0)
           return [];
+        const ids = await __privateMethod$2(this, _insertRecords, insertRecords_fn).call(this, a, { ifVersion, createOnly: false });
         const columns = isStringArray(b) ? b : ["*"];
-        return __privateMethod$2(this, _bulkInsertTableRecords, bulkInsertTableRecords_fn).call(this, a, columns);
+        const result = await this.read(ids, columns);
+        return result;
       }
       if (isString(a) && isObject(b)) {
         const columns = isStringArray(c) ? c : void 0;
@@ -2352,10 +2385,17 @@ class RestRepository extends Query {
       if (Array.isArray(a)) {
         if (a.length === 0)
           return [];
-        if (a.length > 100) {
-          console.warn("Bulk delete operation is not optimized in the Xata API yet, this request might be slow");
-        }
-        return Promise.all(a.map((id) => this.delete(id, b)));
+        const ids = a.map((o) => {
+          if (isString(o))
+            return o;
+          if (isString(o.id))
+            return o.id;
+          throw new Error("Invalid arguments for delete method");
+        });
+        const columns = isStringArray(b) ? b : ["*"];
+        const result = await this.read(a, columns);
+        await __privateMethod$2(this, _deleteRecords, deleteRecords_fn).call(this, ids);
+        return result;
       }
       if (isString(a)) {
         return __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a, b);
@@ -2523,31 +2563,40 @@ insertRecordWithId_fn = async function(recordId, object, columns = ["*"], { crea
   const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
   return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response, columns);
 };
-_bulkInsertTableRecords = new WeakSet();
-bulkInsertTableRecords_fn = async function(objects, columns = ["*"]) {
+_insertRecords = new WeakSet();
+insertRecords_fn = async function(objects, { createOnly, ifVersion }) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
-  const records = objects.map((object) => transformObjectLinks(object));
-  const response = await bulkInsertTableRecords({
-    pathParams: {
-      workspace: "{workspaceId}",
-      dbBranchName: "{dbBranch}",
-      region: "{region}",
-      tableName: __privateGet$4(this, _table)
-    },
-    queryParams: { columns },
-    body: { records },
-    ...fetchProps
-  });
-  if (!isResponseWithRecords(response)) {
-    throw new Error("Request included columns but server didn't include them");
+  const chunkedOperations = chunk(
+    objects.map((object) => ({
+      insert: { table: __privateGet$4(this, _table), record: transformObjectLinks(object), createOnly, ifVersion }
+    })),
+    BULK_OPERATION_MAX_SIZE
+  );
+  const ids = [];
+  for (const operations of chunkedOperations) {
+    const { results } = await branchTransaction({
+      pathParams: {
+        workspace: "{workspaceId}",
+        dbBranchName: "{dbBranch}",
+        region: "{region}"
+      },
+      body: { operations },
+      ...fetchProps
+    });
+    for (const result of results) {
+      if (result.operation === "insert") {
+        ids.push(result.id);
+      } else {
+        ids.push(null);
+      }
+    }
   }
-  const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-  return response.records?.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item, columns));
+  return ids;
 };
 _updateRecordWithID = new WeakSet();
 updateRecordWithID_fn = async function(recordId, object, columns = ["*"], { ifVersion }) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
-  const record = transformObjectLinks(object);
+  const { id: _id, ...record } = transformObjectLinks(object);
   try {
     const response = await updateRecordWithID({
       pathParams: {
@@ -2570,6 +2619,36 @@ updateRecordWithID_fn = async function(recordId, object, columns = ["*"], { ifVe
     throw e;
   }
 };
+_updateRecords = new WeakSet();
+updateRecords_fn = async function(objects, { ifVersion, upsert }) {
+  const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
+  const chunkedOperations = chunk(
+    objects.map(({ id, ...object }) => ({
+      update: { table: __privateGet$4(this, _table), id, ifVersion, upsert, fields: transformObjectLinks(object) }
+    })),
+    BULK_OPERATION_MAX_SIZE
+  );
+  const ids = [];
+  for (const operations of chunkedOperations) {
+    const { results } = await branchTransaction({
+      pathParams: {
+        workspace: "{workspaceId}",
+        dbBranchName: "{dbBranch}",
+        region: "{region}"
+      },
+      body: { operations },
+      ...fetchProps
+    });
+    for (const result of results) {
+      if (result.operation === "update") {
+        ids.push(result.id);
+      } else {
+        ids.push(null);
+      }
+    }
+  }
+  return ids;
+};
 _upsertRecordWithID = new WeakSet();
 upsertRecordWithID_fn = async function(recordId, object, columns = ["*"], { ifVersion }) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
@@ -2612,6 +2691,25 @@ deleteRecord_fn = async function(recordId, columns = ["*"]) {
     throw e;
   }
 };
+_deleteRecords = new WeakSet();
+deleteRecords_fn = async function(recordIds) {
+  const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
+  const chunkedOperations = chunk(
+    recordIds.map((id) => ({ delete: { table: __privateGet$4(this, _table), id } })),
+    BULK_OPERATION_MAX_SIZE
+  );
+  for (const operations of chunkedOperations) {
+    await branchTransaction({
+      pathParams: {
+        workspace: "{workspaceId}",
+        dbBranchName: "{dbBranch}",
+        region: "{region}"
+      },
+      body: { operations },
+      ...fetchProps
+    });
+  }
+};
 _setCacheQuery = new WeakSet();
 setCacheQuery_fn = async function(query, meta, records) {
   await __privateGet$4(this, _cache).set(`query_${__privateGet$4(this, _table)}:${query.key()}`, { date: new Date(), meta, records });
@@ -2722,9 +2820,6 @@ const initObject = (db, schemaTables, table, object, selectedColumns) => {
   Object.freeze(result);
   return result;
 };
-function isResponseWithRecords(value) {
-  return isObject(value) && Array.isArray(value.records);
-}
 function extractId(value) {
   if (isString(value))
     return value;