Initial commit

models/dailyhourstatmod/helper.go

@@ -0,0 +1,24 @@
+package dailyhourstatmod
+
+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 *DailyHourStat) CorrectValidData() *DailyHourStat {
+	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.NewLightCount = s.NewLightDirectCount + s.NewLightFissionCount + s.NewLightDeductionCount
+		s.LightAmount = s.LightDirectAmount + s.LightFissionAmount + s.LightDeductionAmount
+		s.LightAmountCount = s.LightDirectAmountCount + s.LightFissionAmountCount + s.LightDeductionAmountCount
+	}
+	return s
+}