Skip to content

Commit c28c2d6

Browse files
committed
refactor(dataprotection): simplify target version env name
1 parent 4882105 commit c28c2d6

5 files changed

Lines changed: 10 additions & 10 deletions

File tree

pkg/dataprotection/restore/builder.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ func (r *restoreJobBuilder) overridePostReadyTargetEnv(targetEnv []corev1.EnvVar
317317
env := make([]corev1.EnvVar, 0, len(r.env)+len(targetEnv))
318318
for i := range r.env {
319319
if r.env[i].Name == dptypes.DPTargetClusterTopology ||
320-
r.env[i].Name == dptypes.DPTargetComponentServiceVersion {
320+
r.env[i].Name == dptypes.DPTargetServiceVersion {
321321
continue
322322
}
323323
env = append(env, r.env[i])

pkg/dataprotection/restore/builder_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,11 +33,11 @@ func TestRestoreJobBuilderOverridePostReadyTargetEnv(t *testing.T) {
3333
{Name: "KEEP", Value: "kept"},
3434
{Name: dptypes.DPTargetClusterTopology, Value: "restore-value"},
3535
{Name: dptypes.DPTargetClusterTopology, Value: "pod-value"},
36-
{Name: dptypes.DPTargetComponentServiceVersion, Value: "spoofed-version"},
36+
{Name: dptypes.DPTargetServiceVersion, Value: "spoofed-version"},
3737
}}
3838
builder.overridePostReadyTargetEnv([]corev1.EnvVar{
3939
{Name: dptypes.DPTargetClusterTopology, Value: "shared-nothing"},
40-
{Name: dptypes.DPTargetComponentServiceVersion, Value: "3.4.0"},
40+
{Name: dptypes.DPTargetServiceVersion, Value: "3.4.0"},
4141
})
4242

4343
values := func(name string) []string {
@@ -51,5 +51,5 @@ func TestRestoreJobBuilderOverridePostReadyTargetEnv(t *testing.T) {
5151
}
5252
require.Equal(t, []string{"kept"}, values("KEEP"))
5353
require.Equal(t, []string{"shared-nothing"}, values(dptypes.DPTargetClusterTopology))
54-
require.Equal(t, []string{"3.4.0"}, values(dptypes.DPTargetComponentServiceVersion))
54+
require.Equal(t, []string{"3.4.0"}, values(dptypes.DPTargetServiceVersion))
5555
}

pkg/dataprotection/restore/manager.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -702,7 +702,7 @@ func (r *RestoreManager) postReadyTargetEnv(reqCtx intctrlutil.RequestCtx, cli c
702702
}
703703
return []corev1.EnvVar{
704704
{Name: dptypes.DPTargetClusterTopology, Value: cluster.Spec.Topology},
705-
{Name: dptypes.DPTargetComponentServiceVersion, Value: serviceVersion},
705+
{Name: dptypes.DPTargetServiceVersion, Value: serviceVersion},
706706
}, nil
707707
}
708708

pkg/dataprotection/restore/manager_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ func TestRestoreManagerPostReadyTargetEnv(t *testing.T) {
119119
require.NoError(t, err)
120120
require.Equal(t, []corev1.EnvVar{
121121
{Name: dptypes.DPTargetClusterTopology, Value: "shared-nothing"},
122-
{Name: dptypes.DPTargetComponentServiceVersion, Value: tt.wantVersion},
122+
{Name: dptypes.DPTargetServiceVersion, Value: tt.wantVersion},
123123
}, env)
124124
})
125125
}
@@ -623,7 +623,7 @@ var _ = Describe("RestoreManager Test", func() {
623623
})
624624
restoreMGR.Restore.Spec.Env = append(restoreMGR.Restore.Spec.Env,
625625
corev1.EnvVar{Name: dptypes.DPTargetClusterTopology, Value: "spoofed-topology"},
626-
corev1.EnvVar{Name: dptypes.DPTargetComponentServiceVersion, Value: "spoofed-version"})
626+
corev1.EnvVar{Name: dptypes.DPTargetServiceVersion, Value: "spoofed-version"})
627627

628628
By("create cluster to restore")
629629
clusterInfo := testdp.NewFakeCluster(&testCtx)
@@ -656,7 +656,7 @@ var _ = Describe("RestoreManager Test", func() {
656656
// count of job should equal to 1
657657
Expect(len(jobs)).Should(Equal(1))
658658
Expect(postReadyJobEnvValues(jobs[0], dptypes.DPTargetClusterTopology)).Should(Equal([]string{"shared-nothing"}))
659-
Expect(postReadyJobEnvValues(jobs[0], dptypes.DPTargetComponentServiceVersion)).Should(Equal([]string{"3.4.0"}))
659+
Expect(postReadyJobEnvValues(jobs[0], dptypes.DPTargetServiceVersion)).Should(Equal([]string{"3.4.0"}))
660660
// test timeZone transform
661661
var backupStopTimeEnv string
662662
for _, v := range jobs[0].Spec.Template.Spec.Containers[0].Env {

pkg/dataprotection/types/constant.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,8 +122,8 @@ const (
122122
DPTargetPodRole = "DP_TARGET_POD_ROLE"
123123
// DPTargetClusterTopology is the topology of the restore target Cluster.
124124
DPTargetClusterTopology = "DP_TARGET_CLUSTER_TOPOLOGY"
125-
// DPTargetComponentServiceVersion is the expected serviceVersion of the restore target Component instance.
126-
DPTargetComponentServiceVersion = "DP_TARGET_COMPONENT_SERVICE_VERSION"
125+
// DPTargetServiceVersion is the expected serviceVersion of the restore target instance.
126+
DPTargetServiceVersion = "DP_TARGET_SERVICE_VERSION"
127127
// DPBackupBasePath the base path for backup data in the storage
128128
// In a backup action pod, it equals ${DP_BACKUP_ROOT_PATH}/${DP_BACKUP_NAME}/${DP_TARGET_RELATIVE_PATH}
129129
DPBackupBasePath = "DP_BACKUP_BASE_PATH"

0 commit comments

Comments
 (0)