feat: 优化代码

internal/data/cache.go

@@ -6,6 +6,7 @@ import (
 	"slices"
 
 	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
 
 	"github.com/acepanel/panel/v3/internal/app"
 	"github.com/acepanel/panel/v3/internal/biz"
@@ -41,13 +42,10 @@ func (r *cacheRepo) Get(key biz.CacheKey, defaultValue ...string) (string, error
 }
 
 func (r *cacheRepo) Set(key biz.CacheKey, value string) error {
-	cache := new(biz.Cache)
-	if err := r.db.Where(biz.Cache{Key: key}).FirstOrInit(cache).Error; err != nil {
-		return err
-	}
-
-	cache.Value = value
-	return r.db.Save(cache).Error
+	return r.db.Clauses(clause.OnConflict{
+		Columns:   []clause.Column{{Name: "key"}},
+		DoUpdates: clause.AssignmentColumns([]string{"value"}),
+	}).Create(&biz.Cache{Key: key, Value: value}).Error
 }
 
 func (r *cacheRepo) UpdateCategories() error {

internal/data/setting.go

@@ -13,6 +13,7 @@ import (
 	"github.com/libtnb/cache"
 	"github.com/spf13/cast"
 	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
 
 	"github.com/acepanel/panel/v3/internal/app"
 	"github.com/acepanel/panel/v3/internal/biz"
@@ -112,16 +113,10 @@ func (r *settingRepo) GetSlice(key biz.SettingKey, defaultValue ...[]string) ([]
 }
 
 func (r *settingRepo) Set(key biz.SettingKey, value string) error {
-	setting := new(biz.Setting)
-	if err := r.db.Where("key = ?", key).First(setting).Error; err != nil {
-		if !errors.Is(err, gorm.ErrRecordNotFound) {
-			return err
-		}
-	}
-
-	setting.Key = key
-	setting.Value = value
-	if err := r.db.Save(setting).Error; err != nil {
+	if err := r.db.Clauses(clause.OnConflict{
+		Columns:   []clause.Column{{Name: "key"}},
+		DoUpdates: clause.AssignmentColumns([]string{"value"}),
+	}).Create(&biz.Setting{Key: key, Value: value}).Error; err != nil {
 		return err
 	}
 
@@ -130,30 +125,16 @@ func (r *settingRepo) Set(key biz.SettingKey, value string) error {
 }
 
 func (r *settingRepo) SetSlice(key biz.SettingKey, value []string) error {
-	setting := new(biz.Setting)
-	if err := r.db.Where("key = ?", key).First(setting).Error; err != nil {
-		if !errors.Is(err, gorm.ErrRecordNotFound) {
-			return err
-		}
-	}
-
-	setting.Key = key
-	if len(value) == 0 {
-		setting.Value = "[]"
-	} else {
+	v := "[]"
+	if len(value) > 0 {
 		b, err := json.Marshal(value)
 		if err != nil {
 			return err
 		}
-		setting.Value = string(b)
+		v = string(b)
 	}
 
-	if err := r.db.Save(setting).Error; err != nil {
-		return err
-	}
-
-	r.cache.Forget(r.cacheKey(key))
-	return nil
+	return r.Set(key, v)
 }
 
 func (r *settingRepo) Delete(key biz.SettingKey) error {