From 4fa01e289643bc1b91878e28976b494ebaa148f0 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期一, 27 五月 2024 17:46:31 +0800
Subject: [PATCH] 修改更新订阅的模块

---
 service/resend.go |   47 ++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 36 insertions(+), 11 deletions(-)

diff --git a/service/resend.go b/service/resend.go
index c803c0d..ade60a8 100644
--- a/service/resend.go
+++ b/service/resend.go
@@ -1,25 +1,50 @@
 package service
 
 import (
+	"gat1400Exchange/client"
 	"gat1400Exchange/config"
 	"gat1400Exchange/models"
 	"gat1400Exchange/pkg/logger"
 	"gat1400Exchange/util"
+	"gat1400Exchange/vo"
 )
 
-func ResendImageData() {
-	var cacheMod models.Cache
-	cacheItems, _ := cacheMod.FindAll()
+var resendTaskRunning bool
 
-	logger.Debug("Start resend task. cache len:%d", len(cacheItems))
-	for _, c := range cacheItems {
-		if !util.SendData([]byte(c.Data), config.ForwardConf.SyncServer) {
-			c.UpdateRetryCount()
-			logger.Warn("The data resend failed. retry count %d", c.Retry+1)
-		} else {
-			c.Delete()
-			logger.Debug("The data resend successful.")
+func ResendImageData() {
+	if resendTaskRunning {
+		return
+	}
+	resendTaskRunning = true
+	defer func() { resendTaskRunning = false }()
+
+	var cacheMod models.Cache
+	total, _ := cacheMod.Count()
+	logger.Debug("Start resend task. cache len:%d", total)
+
+	for i := 0; i < int(total); i++ {
+		var c models.Cache
+		err := c.First()
+		if err != nil {
+			logger.Error(err.Error())
+			return
 		}
+		if c.Type == "1400" {
+			if client.FaceCapture([]byte(c.Data)) != vo.StatusSuccess {
+				c.UpdateRetryCount()
+				logger.Warn("The data resend failed. retry count %d", c.Retry+1)
+				return
+			}
+		} else {
+			if !util.SendData([]byte(c.Data), config.ForwardConf.SyncServer) {
+				c.UpdateRetryCount()
+				logger.Warn("The data resend failed. retry count %d", c.Retry+1)
+				return
+			}
+		}
+
+		c.Delete()
+		logger.Debug("The data resend successful.")
 	}
 
 }

--
Gitblit v1.8.0