Browse Source

refactor: modify data collection frequency limit

HeWang 6 tháng trước cách đây
mục cha
commit
24d021fddb

+ 14 - 3
amd64/web_server/src/infrastructure/persistence/mapper_collect_limit.go

@@ -28,7 +28,18 @@ func InsertNew(snCode string) {
 	}
 }
 
-func UpdateCollectLimit(snCode string) error {
+func UpdateCollectLimit(snCode string) bool {
+	// 查询记录
+	var resultPos []collect.LimitPo
+	selectSql, _ := util.ReadFile(c_db.SqlFilesMap["collect_limit-select-by-sn_code.sql"])
+	err := c_db.MysqlDb.Select(&resultPos, selectSql, snCode)
+	if err != nil {
+		c_log.GlobalLogger.Error("查询数据报错:", err)
+		return false
+	}
+	if len(resultPos) == 0 {
+		return false
+	}
 	// 更新记录结束时间为默认时间
 	sqlTemplate, _ := util.ReadFile(c_db.SqlFilesMap["collect_limit-add_one.sql"])
 	c_log.GlobalLogger.Error("执行 sql:", sqlTemplate)
@@ -36,9 +47,9 @@ func UpdateCollectLimit(snCode string) error {
 		snCode,
 	}); err != nil {
 		c_log.GlobalLogger.Error("插入数据报错:", err)
-		return err
+		return false
 	}
-	return nil
+	return true
 }
 
 //

+ 2 - 2
amd64/web_server/src/interfaces/api/h_collect_limit.go

@@ -78,8 +78,8 @@ func CanCollectPlus(c *gin.Context) {
 	}
 	c_log.GlobalLogger.Info("请求体为:", param)
 	// 所有值添加一
-	err := persistence.UpdateCollectLimit(param.SnCode)
-	if err != nil {
+	flag := persistence.UpdateCollectLimit(param.SnCode)
+	if !flag {
 		c.JSON(http.StatusOK, commonEntity.Response{
 			Code: 400,
 			Msg:  "设备采集数量更新失败。",