refactor: modify data collection frequency limit

amd64/score_server/dao/mapper/mapper_collect_limit.go

@@ -28,7 +28,7 @@ func InsertNew(snCode string) {
 	}
 }
 
-func UpdateCollectLimit(snCode string) {
+func UpdateCollectLimit(snCode string) error {
 	// 更新记录结束时间为默认时间
 	sqlTemplate, _ := util.ReadFile(c_db.SqlFilesMap["collect_limit-add_one.sql"])
 	c_log.GlobalLogger.Error("执行 sql：", sqlTemplate)
@@ -36,7 +36,7 @@ func UpdateCollectLimit(snCode string) {
 		snCode,
 	}); err != nil {
 		c_log.GlobalLogger.Error("插入数据报错：", err)
-		return
+		return err
 	}
 }
 

amd64/score_server/handler/h_collect_limit.go

@@ -40,8 +40,6 @@ func TriggerCollect(c *gin.Context) {
 		if util.StringToInt(collectLimitPo.MonthCollectedNumber) < util.StringToInt(param.CollectLimitMonth) {
 			if util.StringToInt(collectLimitPo.WeekCollectedNumber) < util.StringToInt(param.CollectLimitWeek) {
 				if util.StringToInt(collectLimitPo.DayCollectedNumber) < util.StringToInt(param.CollectLimitDay) {
-					// 所有值添加一
-					mapper.UpdateCollectLimit(param.SnCode)
 					// 返回允许信号
 					c_log.GlobalLogger.Info("允许采集。")
 					c.JSON(http.StatusOK, commonEntity.Response{
@@ -66,6 +64,30 @@ func TriggerCollect(c *gin.Context) {
 		Msg:  "不允许采集。",
 	})
 }
-func CanCollectPlus(c *gin.Context) {
 
+// 给定设备sn，数据库中所有采集数（日、周、月、年）加一
+func CanCollectPlus(c *gin.Context) {
+	globalMutex.Lock()
+	defer globalMutex.Unlock()
+	param := new(CollectLimitPao)
+	// 映射到结构体
+	if err := c.ShouldBindJSON(&param); err != nil {
+		c_log.GlobalLogger.Error("接收请求参数报错：", err)
+		c.JSON(http.StatusBadRequest, commonEntity.Response{Code: 500, Msg: "请求体解析失败。"})
+		return
+	}
+	c_log.GlobalLogger.Info("请求体为：", param)
+	// 所有值添加一
+	err := mapper.UpdateCollectLimit(param.SnCode)
+	if err != nil {
+		c.JSON(http.StatusOK, commonEntity.Response{
+			Code: 400,
+			Msg:  "设备采集数量更新失败。",
+		})
+		return
+	}
+	c.JSON(http.StatusOK, commonEntity.Response{
+		Code: 200,
+		Msg:  "设备采集数量已更新。",
+	})
 }

amd64/score_server/main.go

@@ -60,7 +60,7 @@ func main() {
 		monitorPrefix := router.Group("/web_server/collect_limit") // todo web_server会改成score_server，后续修改
 		{
 			monitorPrefix.POST("/can_collect", handler.TriggerCollect)
-			monitorPrefix.POST("/can_collect_plus", handler.CanCollectPlus)
+			monitorPrefix.POST("/plus_collect_num", handler.CanCollectPlus)
 		}
 	}
 	// 端口