fix billing total pages and count (#3709)

labring · Aug 18, 2023 · 016c05e · 016c05e
1 parent 32f3843
commit 016c05e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/controllers/pkg/database/mongodb.go b/controllers/pkg/database/mongodb.go
@@ -361,6 +361,8 @@ func (m *MongoDB) queryBillingRecordsByOrderID(billingRecordQuery *accountv1.Bil
 	}
 
 	billingRecordQuery.Status.Items = billingRecords
+	billingRecordQuery.Status.PageLength = 1
+	billingRecordQuery.Status.TotalCount = len(billingRecords)
 	return nil
 }
 
@@ -530,6 +532,10 @@ func (m *MongoDB) QueryBillingRecords(billingRecordQuery *accountv1.BillingRecor
 	}
 
 	totalPages := (totalCount + billingRecordQuery.Spec.PageSize - 1) / billingRecordQuery.Spec.PageSize
+	if totalCount == 0 {
+		totalPages = 1
+		totalCount = len(billingRecords)
+	}
 	billingRecordQuery.Status.Items, billingRecordQuery.Status.PageLength, billingRecordQuery.Status.TotalCount,
 		billingRecordQuery.Status.RechargeAmount, billingRecordQuery.Status.DeductionAmount = billingRecords, totalPages, totalCount, totalRechargeAmount, totalDeductionAmount
 	return nil