From aeb73a502613ce3eb76b84232d212884be9d7f45 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期二, 16 四月 2024 14:51:47 +0800
Subject: [PATCH] 修复缓存保存的问题

---
 models/cache.go           |    4 ++++
 repository/captureRepo.go |    3 +--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/models/cache.go b/models/cache.go
index a8e4ec7..7e54016 100644
--- a/models/cache.go
+++ b/models/cache.go
@@ -20,6 +20,10 @@
 	return caches, nil
 }
 
+func (c *Cache) Save() error {
+	return db.Table(c.TableName()).Save(c).Error
+}
+
 func (c *Cache) UpdateRetryCount() error {
 	return db.Table(c.TableName()).Update("retry", c.Retry+1).Where("id = ?", c.Id).Error
 }
diff --git a/repository/captureRepo.go b/repository/captureRepo.go
index 3654d8c..d1086d8 100644
--- a/repository/captureRepo.go
+++ b/repository/captureRepo.go
@@ -167,6 +167,5 @@
 		Retry:      0,
 	}
 
-	models.GetDB().Table(cacheItem.TableName()).Create(&c)
-
+	cacheItem.Save()
 }

--
Gitblit v1.8.0