From 8ba0a125eb7302d8c51d3c85cf8ea8a4de5413db Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期六, 25 五月 2024 17:56:27 +0800
Subject: [PATCH] 修复ape类型判断

---
 service/resend.go |   24 +++++++++++++++++++-----
 1 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/service/resend.go b/service/resend.go
index 6342ce7..ade60a8 100644
--- a/service/resend.go
+++ b/service/resend.go
@@ -9,12 +9,26 @@
 	"gat1400Exchange/vo"
 )
 
-func ResendImageData() {
-	var cacheMod models.Cache
-	cacheItems, _ := cacheMod.FindAll()
-	logger.Debug("Start resend task. cache len:%d", len(cacheItems))
+var resendTaskRunning bool
 
-	for _, c := range cacheItems {
+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()

--
Gitblit v1.8.0