重置日为0则关闭流量重置

lynkco01 · Dec 24, 2020 · e8f999b · e8f999b
1 parent 6c055d0
commit e8f999b
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/web/controller/data.go b/web/controller/data.go
@@ -56,7 +56,7 @@ func GetResetDay() *ResponseBody {
 func UpdateResetDay(day uint) *ResponseBody {
 	responseBody := ResponseBody{Msg: "success"}
 	defer TimeCost(time.Now(), &responseBody)
-	if day > 31 || day < 1 {
+	if day > 31 || day < 0 {
 		responseBody.Msg = fmt.Sprintf("%d为非正常日期", day)
 		return &responseBody
 	}
@@ -65,10 +65,14 @@ func UpdateResetDay(day uint) *ResponseBody {
 	if day == uint(oldDay) {
 		return &responseBody
 	}
-	c.Remove(c.Entries()[len(c.Entries())-1].ID)
-	c.AddFunc(fmt.Sprintf("0 0 %d * *", day), func() {
-		monthlyResetJob()
-	})
+	if len(c.Entries()) > 1 {
+		c.Remove(c.Entries()[len(c.Entries())-1].ID)
+	}
+	if day != 0 {
+		c.AddFunc(fmt.Sprintf("0 0 %d * *", day), func() {
+			monthlyResetJob()
+		})
+	}
 	core.SetValue("reset_day", strconv.Itoa(int(day)))
 	return &responseBody
 }
@@ -92,8 +96,10 @@ func SheduleTask() {
 		core.SetValue("reset_day", dayStr)
 	}
 	day, _ := strconv.Atoi(dayStr)
-	c.AddFunc(fmt.Sprintf("0 0 %d * *", day), func() {
-		monthlyResetJob()
-	})
+	if day != 0 {
+		c.AddFunc(fmt.Sprintf("0 0 %d * *", day), func() {
+			monthlyResetJob()
+		})
+	}
 	c.Start()
 }