From dcd003ff3e7c674198a19fd4e47b913fd5d9cbb8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=99=88=E5=85=94=E5=AD=90?= <49477914+racns@users.noreply.github.com> Date: Tue, 17 Mar 2026 17:21:44 +0800 Subject: [PATCH] v1.9.4 --- dto/cache.go | 41 +++++++++++++++++++++ facade/cache.go | 95 ++++++++++++++++++------------------------------- 2 files changed, 75 insertions(+), 61 deletions(-) create mode 100644 dto/cache.go diff --git a/dto/cache.go b/dto/cache.go new file mode 100644 index 0000000..bf9c119 --- /dev/null +++ b/dto/cache.go @@ -0,0 +1,41 @@ +package dto + +// CacheConfig - 缓存总配置（由调用方传入） +type CacheConfig struct { + // 配置哈希（可选，不传会自动计算） + Hash string + // 引擎：redis / file + Engine string + // Redis 配置 + Redis CacheRedisConfig + // 文件缓存配置 + File CacheFileConfig +} + +// CacheRedisConfig - Redis 配置 +type CacheRedisConfig struct { + // Host - 主机地址 - 不限制 host，因为 docker 的地址不正常 + Host string `json:"host" comment:"主机地址" default:"localhost"` + // Port - 端口号 + Port int `json:"port" comment:"端口号" validate:"numeric" default:"6379"` + // Password - 密码 + Password string `json:"password" comment:"密码"` + // Expired - 过期时间 + Expired int `json:"expired" comment:"过期时间" validate:"numeric" default:"7200"` + // Prefix - 前缀 + Prefix string `json:"prefix" comment:"前缀" validate:"alphaDash,max=12" default:"INIS"` + // Database - 数据库 + Database int `json:"database" comment:"数据库索引" validate:"numeric"` +} + +// CacheFileConfig - 文件缓存配置 +type CacheFileConfig struct { + // Expired - 过期时间 + Expired int `json:"expired" comment:"过期时间" validate:"numeric" default:"7200"` + // Prefix - 前缀 + Prefix string `json:"prefix" comment:"前缀" validate:"alphaDash,max=12" default:"INIS"` + // Root - 文件缓存根目录 + Root string `json:"root" comment:"文件缓存根目录" default:"/runtime/cache"` + // Suffix - 文件后缀 + Suffix string `json:"suffix" comment:"文件后缀" default:"json"` +} diff --git a/facade/cache.go b/facade/cache.go index 1f7dbe9..d407152 100644 --- a/facade/cache.go +++ b/facade/cache.go @@ -7,7 +7,8 @@ import ( "os" "strings" "time" - + + "github.com/inis-io/aide/dto" "github.com/inis-io/aide/utils" "github.com/redis/go-redis/v9" "github.com/redis/go-redis/v9/maintnotifications" @@ -37,43 +38,15 @@ facade.CacheInst.Watcher(newConfig) type CacheClass struct { // 记录配置 Hash 值，用于检测配置文件是否有变化 - Hash string + Hash string // 当前缓存配置（由调用方注入） - config CacheConfig + config dto.CacheConfig // 是否已经注入过配置 hasConfig bool } -// CacheConfig - 缓存总配置（由调用方传入） -type CacheConfig struct { - // 配置哈希（可选，不传会自动计算） - Hash string - // 引擎：redis / file - Engine string - // Redis 配置 - Redis CacheRedisConfig - // 文件缓存配置 - File CacheFileConfig -} - -type CacheRedisConfig struct { - Host string - Port int - Database int - Password string - Prefix string - Expired int64 -} - -type CacheFileConfig struct { - Prefix string - Expired int64 - Root string - Suffix string -} - // normalizeCacheConfig 统一配置默认值，避免不同项目接入时行为不一致 -func normalizeCacheConfig(config CacheConfig) CacheConfig { +func normalizeCacheConfig(config dto.CacheConfig) dto.CacheConfig { config.Engine = strings.ToLower(strings.TrimSpace(config.Engine)) if utils.Is.Empty(config.Engine) { @@ -114,22 +87,20 @@ func normalizeCacheConfig(config CacheConfig) CacheConfig { } // SetConfig - 注入缓存配置 -func (this *CacheClass) SetConfig(config CacheConfig) *CacheClass { +func (this *CacheClass) SetConfig(config dto.CacheConfig) *CacheClass { this.config = normalizeCacheConfig(config) this.hasConfig = true return this } // Watcher - 监听器 -func (this *CacheClass) Watcher(config ...CacheConfig) { +func (this *CacheClass) Watcher(config ...dto.CacheConfig) { if len(config) > 0 { this.SetConfig(config[0]) } - if !this.hasConfig { - return - } + if !this.hasConfig { return } // hash 变化，说明配置有更新 if this.Hash != this.config.Hash { @@ -138,7 +109,7 @@ func (this *CacheClass) Watcher(config ...CacheConfig) { } // Init 初始化 -func (this *CacheClass) Init(config ...CacheConfig) { +func (this *CacheClass) Init(config ...dto.CacheConfig) { if len(config) > 0 { this.SetConfig(config[0]) @@ -260,7 +231,7 @@ type CacheAPI interface { */ Expired(second any) CacheAPI // NewCache - 新建缓存 - NewCache(config CacheConfig) CacheAPI + NewCache(config dto.CacheConfig) CacheAPI } type CacheBody struct { @@ -280,27 +251,27 @@ type CacheBody struct { type RedisClass struct { Client *redis.Client Body CacheBody - Config CacheRedisConfig + Config dto.CacheRedisConfig } -func (this *RedisClass) NewCache(config CacheConfig) CacheAPI { +func (this *RedisClass) NewCache(config dto.CacheConfig) CacheAPI { - item := normalizeCacheConfig(config) + conf := normalizeCacheConfig(config) - switch item.Engine { + switch conf.Engine { case "file": cache := &FileClass{} - cache.Init(item.File) + cache.Init(conf.File) return cache default: cache := &RedisClass{} - cache.Init(item.Redis) + cache.Init(conf.Redis) return cache } } // Init - 初始化 Redis 缓存 -func (this *RedisClass) Init(config CacheRedisConfig) { +func (this *RedisClass) Init(config dto.CacheRedisConfig) { this.Config = config @@ -581,28 +552,29 @@ func (this *RedisClass) DelTags() { type FileClass struct { // 文件客户端 - Fs afero.Fs + Fs afero.Fs // 缓存参数 - Body CacheBody + Body CacheBody // 当前配置 - Config CacheFileConfig + Config dto.CacheFileConfig // 存储目录 - Root string + Root string // 文件后缀 Suffix string } -func (this *FileClass) NewCache(config CacheConfig) CacheAPI { - item := normalizeCacheConfig(config) +func (this *FileClass) NewCache(config dto.CacheConfig) CacheAPI { + + conf := normalizeCacheConfig(config) - switch item.Engine { + switch strings.ToLower(conf.Engine) { case "redis": cache := &RedisClass{} - cache.Init(item.Redis) + cache.Init(conf.Redis) return cache default: cache := &FileClass{} - cache.Init(item.File) + cache.Init(conf.File) return cache } } @@ -616,10 +588,11 @@ type FileCacheResp struct { } // Init 初始化 文件缓存 -func (this *FileClass) Init(config CacheFileConfig) { +func (this *FileClass) Init(config dto.CacheFileConfig) { + this.Config = config - this.Root = this.Config.Root + this.Root = this.Config.Root // 设置文件后缀 this.Suffix = this.Config.Suffix @@ -701,7 +674,7 @@ func (this *FileClass) SetTags(key string) { for _, tag := range this.Body.Tags { var value []byte - var keys []string + var keys []string // 文件名 name := fmt.Sprintf("%v-%v.%s", this.Body.Prefix, tag, this.Suffix) @@ -719,12 +692,12 @@ func (this *FileClass) SetTags(key string) { // 获取标签下的所有成员 read, _ := this.Read(dest) - keys = cast.ToStringSlice(utils.Json.Decode(read)) + keys = cast.ToStringSlice(utils.Json.Decode(read)) // 添加新成员 - keys = append(keys, this.Name(key)) + keys = append(keys, this.Name(key)) // 去重 - keys = cast.ToStringSlice(utils.ArrayUnique(keys)) + keys = cast.ToStringSlice(utils.ArrayUnique(keys)) // 重新设置 value = []byte(utils.Json.Encode(keys)) // 写入文件