Invalidate OLAP table list when a new source/model is added (#4724)

rilldata · Apr 26, 2024 · 039d5a9 · 039d5a9 · github-actions · Apr 26, 2024
1 parent f6f1dc3
commit 039d5a9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/web-common/src/features/entity-management/file-artifacts.ts b/web-common/src/features/entity-management/file-artifacts.ts
@@ -44,6 +44,8 @@ export class FileArtifact {
    */
   public lastStateUpdatedOn: string | undefined;
 
+  public isNew = true;
+
   public constructor(filePath: string) {
     this.path = filePath;
   }
@@ -58,6 +60,7 @@ export class FileArtifact {
         V1ReconcileStatus.RECONCILE_STATUS_RUNNING,
     );
     this.renaming = !!resource.meta?.renamedFrom;
+    this.isNew = false;
   }
 
   public updateReconciling(resource: V1Resource) {
@@ -284,6 +287,14 @@ export class FileArtifacts {
     });
   }
 
+  public isNew(resource: V1Resource) {
+    return (
+      resource.meta?.filePaths?.some((filePath) => {
+        return this.artifacts[filePath].isNew;
+      }) ?? false
+    );
+  }
+
   public wasRenaming(resource: V1Resource) {
     const finishedRename = !resource.meta?.renamedFrom;
     return (

diff --git a/web-common/src/features/entity-management/resource-invalidations.ts b/web-common/src/features/entity-management/resource-invalidations.ts
@@ -104,7 +104,12 @@ async function invalidateResource(
   )
     return;
 
-  if (fileArtifacts.wasRenaming(resource)) {
+  if (
+    fileArtifacts.wasRenaming(resource) ||
+    (fileArtifacts.isNew(resource) &&
+      (resource.meta.name?.kind === ResourceKind.Source ||
+        resource.meta.name?.kind === ResourceKind.Model))
+  ) {
     void queryClient.invalidateQueries(
       getConnectorServiceOLAPListTablesQueryKey(),
     );