add debt-ctr backoff version func (labring#3615)

* add debt-ctr backoff version func
cuisongliu · Aug 2, 2023 · 36f8008 · 36f8008
1 parent 49aaf13
commit 36f8008
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/controllers/account/api/v1/debt_types.go b/controllers/account/api/v1/debt_types.go
@@ -32,6 +32,10 @@ const (
 	ImminentDeletionPeriod    DebtStatusType = "ImminentDeletionPeriod"
 	FinalDeletionPeriod       DebtStatusType = "FinalDeletionPeriod"
 
+	PreWarningPeriod DebtStatusType = "PreWarningPeriod"
+	SuspendPeriod    DebtStatusType = "SuspendPeriod"
+	RemovedPeriod    DebtStatusType = "RemovedPeriod"
+
 	DebtPrefix = "debt-"
 	DaySecond  = 24 * 60 * 60
 )

diff --git a/controllers/account/controllers/debt_controller.go b/controllers/account/controllers/debt_controller.go
@@ -254,7 +254,7 @@ func (r *DebtReconciler) reconcileDebtStatus(ctx context.Context, debt *accountv
 		}
 	//兼容老版本
 	default:
-		update = oldStatusConversion(debt)
+		update = newStatusConversion(debt)
 	}
 
 	if update {
@@ -284,13 +284,14 @@ func SetDebtStatus(debt *accountv1.Debt, status accountv1.DebtStatusType) bool {
 	return true
 }
 
-// 旧版本转换
-func oldStatusConversion(debt *accountv1.Debt) bool {
+func newStatusConversion(debt *accountv1.Debt) bool {
 	switch debt.Status.AccountDebtStatus {
-	case accountv1.DebtStatusNormal:
+	case accountv1.PreWarningPeriod:
 		debt.Status.AccountDebtStatus = accountv1.NormalPeriod
-	default:
+	case accountv1.SuspendPeriod:
 		debt.Status.AccountDebtStatus = accountv1.ImminentDeletionPeriod
+	case accountv1.RemovedPeriod:
+		debt.Status.AccountDebtStatus = accountv1.FinalDeletionPeriod
 	}
 	return true
 }