Merge pull request backstage#28518 from backstage/blam/remove-things

catalog: cleanup `refresh_state_references` for old providers
miqdigital · Jan 16, 2025 · 8467532 · 8467532
2 parents eb39a6e + 67bf279
commit 8467532
Show file tree

Hide file tree

Showing 3 changed files with 159 additions and 3 deletions.
diff --git a/.changeset/fair-mangos-sleep.md b/.changeset/fair-mangos-sleep.md
@@ -0,0 +1,5 @@
+---
+'@backstage/plugin-catalog-backend': patch
+---
+
+Cleanup `refresh_state_references` for providers that are no longer in control of a `refresh_state` row for entity
diff --git a/plugins/catalog-backend/src/database/DefaultProviderDatabase.ts b/plugins/catalog-backend/src/database/DefaultProviderDatabase.ts
@@ -169,6 +169,10 @@ export class DefaultProviderDatabase implements ProviderDatabase {
             .delete();
 
           if (ok) {
+            await tx<DbRefreshStateReferencesRow>('refresh_state_references')
+              .where('target_entity_ref', entityRef)
+              .delete();
+
             await tx<DbRefreshStateReferencesRow>(
               'refresh_state_references',
             ).insert({

diff --git a/plugins/catalog-backend/src/tests/integration.test.ts b/plugins/catalog-backend/src/tests/integration.test.ts
@@ -31,7 +31,7 @@ import {
 } from '@backstage/plugin-catalog-node';
 import { PermissionEvaluator } from '@backstage/plugin-permission-common';
 import { createHash } from 'crypto';
-import { Knex } from 'knex';
+import knexFactory, { Knex } from 'knex';
 import { EntitiesCatalog } from '../catalog/types';
 import { DefaultCatalogDatabase } from '../database/DefaultCatalogDatabase';
 import { DefaultProcessingDatabase } from '../database/DefaultProcessingDatabase';
@@ -55,6 +55,7 @@ import { mockServices } from '@backstage/backend-test-utils';
 import { LoggerService } from '@backstage/backend-plugin-api';
 import { DatabaseManager } from '@backstage/backend-common';
 import { entitiesResponseToObjects } from '../service/response';
+import { DbRefreshStateReferencesRow } from '../database/tables';
 
 const voidLogger = mockServices.logger.mock();
 
@@ -202,6 +203,7 @@ class TestHarness {
     logger?: LoggerService;
     db?: Knex;
     permissions?: PermissionEvaluator;
+    additionalProviders?: EntityProvider[];
     processEntity?(
       entity: Entity,
       location: LocationSpec,
@@ -301,8 +303,13 @@ class TestHarness {
     const refresh = new DefaultRefreshService({ database: catalogDatabase });
 
     const provider = new TestProvider();
+    const providers: EntityProvider[] = [provider];
 
-    await connectEntityProviders(providerDatabase, [provider]);
+    if (options?.additionalProviders) {
+      providers.push(...options.additionalProviders);
+    }
+
+    await connectEntityProviders(providerDatabase, providers);
 
     return new TestHarness(
       catalog,
@@ -353,7 +360,7 @@ class TestHarness {
   }
 
   async setInputEntities(entities: (Entity & { locationKey?: string })[]) {
-    return this.#provider.getConnection().applyMutation({
+    await this.#provider.getConnection().applyMutation({
       type: 'full',
       entities: entities.map(({ locationKey, ...entity }) => ({
         entity,
@@ -838,4 +845,144 @@ describe('Catalog Backend Integration', () => {
       },
     });
   });
+
+  it('should replace any refresh_state_references that are dangling after claiming an entityRef with locationKey', async () => {
+    const db = knexFactory({
+      client: 'better-sqlite3',
+      connection: ':memory:',
+      useNullAsDefault: true,
+    });
+
+    const firstProvider = new TestProvider();
+    firstProvider.getProviderName = () => 'first';
+
+    const secondProvider = new TestProvider();
+    secondProvider.getProviderName = () => 'second';
+
+    const harness = await TestHarness.create({
+      additionalProviders: [firstProvider, secondProvider],
+      db,
+    });
+
+    await firstProvider.getConnection().applyMutation({
+      type: 'full',
+      entities: [
+        {
+          entity: {
+            apiVersion: 'backstage.io/v1alpha1',
+            kind: 'Component',
+            metadata: {
+              name: 'component-1',
+              annotations: {
+                'backstage.io/managed-by-location': 'url:.',
+                'backstage.io/managed-by-origin-location': 'url:.',
+              },
+            },
+            spec: {
+              type: 'service',
+              owner: 'no-location-key',
+            },
+          },
+        },
+      ],
+    });
+
+    await expect(harness.process()).resolves.toEqual({});
+
+    await expect(harness.getOutputEntities()).resolves.toEqual({
+      'component:default/component-1': expect.objectContaining({
+        spec: {
+          type: 'service',
+          owner: 'no-location-key',
+        },
+      }),
+    });
+
+    await secondProvider.getConnection().applyMutation({
+      type: 'full',
+      entities: [
+        {
+          locationKey: 'takeover',
+          entity: {
+            apiVersion: 'backstage.io/v1alpha1',
+            kind: 'Component',
+            metadata: {
+              name: 'component-1',
+              annotations: {
+                'backstage.io/managed-by-location': 'url:.',
+                'backstage.io/managed-by-origin-location': 'url:.',
+              },
+            },
+            spec: {
+              type: 'service',
+              owner: 'location-key',
+            },
+          },
+        },
+      ],
+    });
+
+    await expect(harness.process()).resolves.toEqual({});
+
+    await expect(harness.getOutputEntities()).resolves.toEqual({
+      'component:default/component-1': expect.objectContaining({
+        spec: {
+          type: 'service',
+          owner: 'location-key',
+        },
+      }),
+    });
+
+    await expect(
+      db<DbRefreshStateReferencesRow>('refresh_state_references')
+        .where({ target_entity_ref: 'component:default/component-1' })
+        .select(),
+    ).resolves.toEqual([
+      expect.objectContaining({
+        source_key: 'second',
+        target_entity_ref: 'component:default/component-1',
+      }),
+    ]);
+
+    await secondProvider.getConnection().applyMutation({
+      type: 'full',
+      entities: [
+        {
+          locationKey: 'takeover',
+          entity: {
+            apiVersion: 'backstage.io/v1alpha1',
+            kind: 'Component',
+            metadata: {
+              name: 'component-2',
+              annotations: {
+                'backstage.io/managed-by-location': 'url:.',
+                'backstage.io/managed-by-origin-location': 'url:.',
+              },
+            },
+            spec: {
+              type: 'service',
+              owner: 'location-key',
+            },
+          },
+        },
+      ],
+    });
+
+    await expect(harness.process()).resolves.toEqual({});
+
+    await expect(
+      db<DbRefreshStateReferencesRow>('refresh_state_references')
+        .where({ target_entity_ref: 'component:default/component-1' })
+        .select(),
+    ).resolves.toEqual([]);
+
+    await expect(harness.getOutputEntities()).resolves.toEqual({
+      'component:default/component-2': expect.objectContaining({
+        spec: {
+          type: 'service',
+          owner: 'location-key',
+        },
+      }),
+    });
+  });
 });