add datasource api flush method

infinite-table · Sep 21, 2024 · 25505f1 · 25505f1
1 parent 9a87545
commit 25505f1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/source/src/components/DataSource/getDataSourceApi.ts b/source/src/components/DataSource/getDataSourceApi.ts
@@ -159,11 +159,16 @@ class DataSourceApiImpl<T> implements DataSourceApi<T> {
     this.actions.cache = cache;
   }
 
+  flush() {
+    return this.commit();
+  }
+
   commit() {
     this.commitOperations(this.pendingOperations);
     this.pendingOperations.length = 0;
 
-    if (this.pendingPromise && this.resolvePendingPromise) {
+    const pendingPromise = this.pendingPromise;
+    if (pendingPromise && this.resolvePendingPromise) {
       const resolve = this.resolvePendingPromise;
       // let's resolve the promise in the next frame
       // so we give the DataSource reducer the chance to pick up the commited operations
@@ -175,6 +180,8 @@ class DataSourceApiImpl<T> implements DataSourceApi<T> {
       this.pendingPromise = null;
       this.resolvePendingPromise = null;
     }
+
+    return pendingPromise || Promise.resolve(true);
   }
 
   getRowInfoArray = () => {

diff --git a/source/src/components/DataSource/types.ts b/source/src/components/DataSource/types.ts
@@ -434,6 +434,8 @@ export interface DataSourceApi<T> {
     options?: DataSourceCRUDParam,
   ): Promise<any>;
 
+  flush(): Promise<any>;
+
   removeDataByPrimaryKey(id: any, options?: DataSourceCRUDParam): Promise<any>;
   removeDataArrayByPrimaryKeys(
     id: any[],