PLT-724:Fixed Back up day 2 operation

spectrocloud · Oct 9, 2023 · 31c76af · 31c76af
1 parent b76840d
commit 31c76af
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/client/cluster_backup_config.go b/client/cluster_backup_config.go
@@ -58,13 +58,28 @@ func (h *V1Client) CreateClusterBackupConfig(uid string, config *models.V1Cluste
  return err
 }
 
-func (h *V1Client) UpdateClusterBackupConfig(uid string, config *models.V1ClusterBackupConfig) error {
+func (h *V1Client) UpdateClusterBackupConfig(uid string, config *models.V1ClusterBackupConfig, ClusterContext string) error {
  client, err := h.GetClusterClient()
  if err != nil {
  return err
  }
-
- params := clusterC.NewV1ClusterFeatureBackupUpdateParamsWithContext(h.Ctx).WithUID(uid).WithBody(config)
+ var params *clusterC.V1ClusterFeatureBackupUpdateParams
+ switch ClusterContext {
+ case "project":
+ params = clusterC.NewV1ClusterFeatureBackupUpdateParamsWithContext(h.Ctx).WithUID(uid).WithBody(config)
+ case "tenant":
+ params = clusterC.NewV1ClusterFeatureBackupUpdateParams().WithUID(uid).WithBody(config)
+ }
  _, err = client.V1ClusterFeatureBackupUpdate(params)
  return err
 }
+
+func (h *V1Client) ApplyClusterBackupConfig(uid string, config *models.V1ClusterBackupConfig, ClusterContext string) error {
+ if policy, err := h.GetClusterBackupConfig(uid, ClusterContext); err != nil {
+ return err
+ } else if policy == nil {
+ return h.CreateClusterBackupConfig(uid, config, ClusterContext)
+ } else {
+ return h.UpdateClusterBackupConfig(uid, config, ClusterContext)
+ }
+}