From a6e84625b91e19220933ed3bc8ef91435281ab26 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期二, 16 四月 2024 16:52:35 +0800
Subject: [PATCH] 添加服务器是否可以连接的判断

---
 models/cache.go |    8 ++++++--
 1 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/models/cache.go b/models/cache.go
index a4516e8..54becc2 100644
--- a/models/cache.go
+++ b/models/cache.go
@@ -20,10 +20,14 @@
 	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).Error
+	return db.Table(c.TableName()).Where("id = ?", c.Id).Update("retry", c.Retry+1).Error
 }
 
 func (c *Cache) Delete() error {
-	return db.Table(c.TableName()).Delete("id = ?", c.Id).Error
+	return db.Table(c.TableName()).Where("id = ?", c.Id).Delete(c).Error
 }

--
Gitblit v1.8.0