refactor: use separate functions for different policy types

uselagoon · Oct 18, 2024 · ac394de · ac394de
1 parent a1a7d6a
commit ac394de
Show file tree

Hide file tree

Showing 10 changed files with 497 additions and 413 deletions.
diff --git a/local-dev/api-data-watcher-pusher/api-data/03-populate-api-data-ci-local-control-k8s.gql b/local-dev/api-data-watcher-pusher/api-data/03-populate-api-data-ci-local-control-k8s.gql
@@ -128,92 +128,78 @@ mutation PopulateApi {
     id
   }
 
-  RetPol1: createRetentionPolicy(input:{
+  RetPol1: createHarborRetentionPolicy(input:{
     name: "harbor-policy"
-    type: HARBOR
-    harbor: {
-      enabled: true
-      rules: [
-        {
-          name: "all branches, excluding pullrequests"
-          pattern: "[^pr-]*/*"
-          latestPulled: 3
-        },
-        {
-          name: "pullrequests"
-          pattern: "pr-*"
-          latestPulled: 1
-        }
-      ]
-      schedule: "3 * * * *"
-    }
+    enabled: true
+    rules: [
+      {
+        name: "all branches, excluding pullrequests"
+        pattern: "[^pr-]*/*"
+        latestPulled: 3
+      },
+      {
+        name: "pullrequests"
+        pattern: "pr-*"
+        latestPulled: 1
+      }
+    ]
+    schedule: "3 * * * *"
   }) {
     id
     name
     configuration {
-      ... on HarborRetentionPolicy {
-        enabled
-        rules {
-          name
-          pattern
-          latestPulled
-        }
-        schedule
+      enabled
+      rules {
+        name
+        pattern
+        latestPulled
       }
+      schedule
     }
-    type
     created
     updated
   }
 
-  RetPol2: createRetentionPolicy(input:{
+  RetPol2: createHistoryRetentionPolicy(input:{
     name: "history-policy"
-    type: HISTORY
-    history: {
-      enabled: true
-      deploymentHistory: 5
-      deploymentType: COUNT
-      taskHistory: 10
-      taskType: COUNT
-    }
+    enabled: true
+    deploymentHistory: 5
+    deploymentType: COUNT
+    taskHistory: 10
+    taskType: COUNT
   }) {
     id
     name
     configuration {
-      ... on HistoryRetentionPolicy {
-        enabled
-        deploymentHistory
-        deploymentType
-        taskHistory
-        taskType
-      }
+      enabled
+      deploymentHistory
+      deploymentType
+      taskHistory
+      taskType
     }
-    type
     created
     updated
   }
 
-  RetPolLink1: addRetentionPolicyLink(input:{
-    id: 1
+  RetPolLink1: addHarborRetentionPolicyLink(input:{
+    name: "harbor-policy"
     scope: GLOBAL
     scopeName: "global",
   }) {
     id
     name
-    type
     source
     created
     updated
   }
 
-  RetPolLink2: addRetentionPolicyLink(input:{
-    id: 2
+  RetPolLink2: addHistoryRetentionPolicyLink(input:{
+    name: "history-policy"
     scope: GLOBAL
     scopeName: "global",
   }) {
     id
     name
-    type
     source
     created
     updated

diff --git a/services/api/database/migrations/20240708000000_retention_policy.js b/services/api/database/migrations/20240708000000_retention_policy.js
@@ -6,11 +6,12 @@ exports.up = async function(knex) {
     return knex.schema
     .createTable('retention_policy', function (table) {
         table.increments('id').notNullable().primary();
-        table.string('name', 300).unique({indexName: 'name'});
+        table.string('name', 300);
         table.enu('type',['harbor','history']).notNullable();
         table.text('configuration');
         table.timestamp('updated').notNullable().defaultTo(knex.fn.now());
         table.timestamp('created').notNullable().defaultTo(knex.fn.now());
+        table.unique(['name', 'type'], {indexName: 'retention_policy'});
     })
     .createTable('retention_policy_reference', function (table) {
         table.integer('retention_policy');

diff --git a/services/api/src/models/retentionpolicy.ts b/services/api/src/models/retentionpolicy.ts
diff --git a/services/api/src/resolvers.js b/services/api/src/resolvers.js
@@ -276,14 +276,22 @@ const {
 } = require('./resources/backup/resolvers');
 
 const {
-  createRetentionPolicy,
-  updateRetentionPolicy,
-  deleteRetentionPolicy,
-  getRetentionPoliciesByProjectId,
-  getRetentionPoliciesByOrganizationId,
-  listRetentionPolicies,
-  addRetentionPolicyLink,
-  removeRetentionPolicyLink,
+  createHarborRetentionPolicy,
+  updateHarborRetentionPolicy,
+  createHistoryRetentionPolicy,
+  updateHistoryRetentionPolicy,
+  deleteHarborRetentionPolicy,
+  deleteHistoryRetentionPolicy,
+  getHarborRetentionPoliciesByProjectId,
+  getHarborRetentionPoliciesByOrganizationId,
+  getHistoryRetentionPoliciesByProjectId,
+  getHistoryRetentionPoliciesByOrganizationId,
+  listHarborRetentionPolicies,
+  listHistoryRetentionPolicies,
+  addHarborRetentionPolicyLink,
+  addHistoryRetentionPolicyLink,
+  removeHarborRetentionPolicyLink,
+  removeHistoryRetentionPolicyLink,
 } = require('./resources/retentionpolicy/resolvers');
 
 const {
@@ -387,10 +395,6 @@ const resolvers = {
     ACTIVE: 'active',
     SUCCEEDED: 'succeeded',
   },
-  RetentionPolicyType: {
-    HARBOR: 'harbor',
-    HISTORY: 'history',
-  },
   RetentionPolicyScope: {
     GLOBAL: 'global',
     ORGANIZATION: 'organization',
@@ -423,7 +427,8 @@ const resolvers = {
     groups: getGroupsByProjectId,
     privateKey: getPrivateKey,
     publicKey: getProjectDeployKey,
-    retentionPolicies: getRetentionPoliciesByProjectId,
+    harborRetentionPolicies: getHarborRetentionPoliciesByProjectId,
+    historyRetentionPolicies: getHistoryRetentionPoliciesByProjectId,
   },
   GroupInterface: {
     __resolveType(group) {
@@ -475,13 +480,15 @@ const resolvers = {
     owners: getOwnersByOrganizationId,
     deployTargets: getDeployTargetsByOrganizationId,
     notifications: getNotificationsByOrganizationId,
-    retentionPolicies: getRetentionPoliciesByOrganizationId
+    harborRetentionPolicies: getHarborRetentionPoliciesByOrganizationId,
+    historyRetentionPolicies: getHistoryRetentionPoliciesByOrganizationId
   },
   OrgProject: {
     groups: getGroupsByOrganizationsProject,
     groupCount: getGroupCountByOrganizationProject,
     notifications: getNotificationsForOrganizationProjectId,
-    retentionPolicies: getRetentionPoliciesByProjectId,
+    harborRetentionPolicies: getHarborRetentionPoliciesByProjectId,
+    historyRetentionPolicies: getHistoryRetentionPoliciesByProjectId,
   },
   OrgEnvironment: {
     project: getProjectById,
@@ -529,18 +536,6 @@ const resolvers = {
       }
     }
   },
-  RetentionPolicyConfiguration: {
-    __resolveType(obj) {
-      switch (obj.type) {
-        case 'harbor':
-          return 'HarborRetentionPolicy';
-        case 'history':
-          return 'HistoryRetentionPolicy';
-        default:
-          return null;
-      }
-    }
-  },
   AdvancedTaskDefinition: {
     __resolveType (obj) {
       switch(obj.type) {
@@ -622,7 +617,8 @@ const resolvers = {
     getProjectGroupOrganizationAssociation,
     getEnvVariablesByProjectEnvironmentName,
     checkBulkImportProjectsAndGroupsToOrganization,
-    listRetentionPolicies
+    listHarborRetentionPolicies,
+    listHistoryRetentionPolicies
   },
   Mutation: {
     addProblem,
@@ -747,11 +743,16 @@ const resolvers = {
     bulkImportProjectsAndGroupsToOrganization,
     addOrUpdateEnvironmentService,
     deleteEnvironmentService,
-    createRetentionPolicy,
-    updateRetentionPolicy,
-    deleteRetentionPolicy,
-    addRetentionPolicyLink,
-    removeRetentionPolicyLink
+    createHarborRetentionPolicy,
+    updateHarborRetentionPolicy,
+    deleteHarborRetentionPolicy,
+    addHarborRetentionPolicyLink,
+    removeHarborRetentionPolicyLink,
+    createHistoryRetentionPolicy,
+    updateHistoryRetentionPolicy,
+    deleteHistoryRetentionPolicy,
+    addHistoryRetentionPolicyLink,
+    removeHistoryRetentionPolicyLink,
   },
   Subscription: {
     backupChanged: backupSubscriber,