From 858833a4d5d5bbf4b2d1d5418f6ae5a92ec0d5d3 Mon Sep 17 00:00:00 2001
From: Kiss Miklos <miklos@roadie.io>
Date: Wed, 7 Sep 2022 13:00:48 +0200
Subject: [PATCH] add refresh to connection

Signed-off-by: Kiss Miklos <miklos@roadie.io>
---
 plugins/catalog-backend/src/database/types.ts          |  8 ++++++++
 .../src/processing/connectEntityProviders.ts           | 10 ++++++++++
 2 files changed, 18 insertions(+)

diff --git a/plugins/catalog-backend/src/database/types.ts b/plugins/catalog-backend/src/database/types.ts
index 07241f0682..2d30f8bd86 100644
--- a/plugins/catalog-backend/src/database/types.ts
+++ b/plugins/catalog-backend/src/database/types.ts
@@ -157,6 +157,14 @@ export interface ProcessingDatabase {
    */
   refresh(txOpaque: Transaction, options: RefreshOptions): Promise<void>;
 
+  /**
+   * Schedules a refresh for every entity that has a matching set of refresh key stored for it.
+   */
+  refreshByRefreshKeys(
+    txOpaque: Transaction,
+    options: RefreshByKeyOptions,
+  ): Promise<void>;
+
   /**
    * Lists all ancestors of a given entityRef.
    *
diff --git a/plugins/catalog-backend/src/processing/connectEntityProviders.ts b/plugins/catalog-backend/src/processing/connectEntityProviders.ts
index 70a43de589..8325e2bcac 100644
--- a/plugins/catalog-backend/src/processing/connectEntityProviders.ts
+++ b/plugins/catalog-backend/src/processing/connectEntityProviders.ts
@@ -61,6 +61,16 @@ class Connection implements EntityProviderConnection {
     }
   }
 
+  async refresh(keys: string[]): Promise<void> {
+    const db = this.config.processingDatabase;
+
+    await db.transaction(async (tx: any) => {
+      return db.refreshByRefreshKeys(tx, {
+        keys,
+      });
+    });
+  }
+
   private check(entities: Entity[]) {
     for (const entity of entities) {
       try {