Initial commit

models/dailystatmod/helper.go

@@ -0,0 +1,24 @@
+package dailystatmod
+
+import "time"
+
+var (
+	fixTime time.Time
+)
+
+func init() {
+	time.Local, _ = time.LoadLocation("Asia/Shanghai")
+	fixTime, _ = time.ParseInLocation("2006-01-02", "2023-07-14", time.Local)
+}
+
+func (s *DailyStatMod) CorrectValidData() *DailyStatMod {
+	if s.SumDate.Before(fixTime) {
+		s.TotalAmount = s.DirectAmount + s.FissionAmount + s.DeductionAmount
+		s.NewTotalCount = s.NewDirectCount + s.NewFissionCount + s.NewDeductionCount
+		s.TotalVisitCount = s.DirectVisitCount + s.FissionVisitCount + s.DeductionVisitCount
+		s.LightAddCount = s.LightDirectCount + s.LightFissionCount + s.LightDeductionCount
+		s.LightRechargeCount = s.LightDirectRechargeCount + s.LightFissionRechargeCount + s.LightDeductionRechargeCount
+		s.LightRechargeAmount = s.LightDirectRechargeAmount + s.LightFissionRechargeAmount + s.LightDeductionRechargeAmount
+	}
+	return s
+}