95 lines
3.2 KiB
Go
95 lines
3.2 KiB
Go
package cache
|
||
|
||
import (
|
||
"electricity_bill_calc/logger"
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
var log = logger.Named("Cache")
|
||
|
||
func assembleSearchKey(entityName string, additional ...string) string {
|
||
var keys = make([]string, 0)
|
||
keys = append(keys, strings.ToUpper(entityName))
|
||
keys = append(keys, additional...)
|
||
var b strings.Builder
|
||
b.WriteString(TAG_SEARCH)
|
||
for _, s := range keys {
|
||
fmt.Fprintf(&b, ":%s", s)
|
||
}
|
||
return b.String()
|
||
}
|
||
|
||
// 缓存模型名称明确的，使用或者包含非ID检索条件的实体内容。
|
||
func CacheSearch[T any](instance T, relationNames []string, entityName string, conditions ...string) error {
|
||
searchKey := assembleSearchKey(entityName, conditions...)
|
||
err := Cache(searchKey, &instance, 5*time.Minute)
|
||
for _, relationName := range relationNames {
|
||
CacheRelation(relationName, STORE_TYPE_KEY, searchKey)
|
||
}
|
||
return err
|
||
}
|
||
|
||
// 从缓存中取得模型名称明确的，使用或者包含非ID检索条件的实体内容。
|
||
func RetrieveSearch[T any](entityName string, conditions ...string) (*T, error) {
|
||
searchKey := assembleSearchKey(entityName, conditions...)
|
||
instance, err := Retrieve[T](searchKey)
|
||
return instance, err
|
||
}
|
||
|
||
// 从缓存中删除全部指定模型名称的实体内容。
|
||
func AbolishSearch(entityName string) error {
|
||
pattern := fmt.Sprintf("%s:%s:*", TAG_SEARCH, strings.ToUpper(entityName))
|
||
return DeleteAll(pattern)
|
||
}
|
||
|
||
// 向缓存中保存指定模型名称的分页检索结果，会同时采用`CacheCount`中的方法保存检索结果的总数量。
|
||
func CachePagedSearch[T any](instance T, total int64, relationNames []string, entityName string, conditions ...string) error {
|
||
searchKey := assembleSearchKey(entityName, conditions...)
|
||
countKey := assembleCountKey(entityName, conditions...)
|
||
err := Cache(searchKey, &instance, 5*time.Minute)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
cacheInstance := &_CountRecord{Count: total}
|
||
err = Cache(countKey, cacheInstance, 5*time.Minute)
|
||
for _, relationName := range relationNames {
|
||
CacheRelation(relationName, STORE_TYPE_KEY, searchKey)
|
||
CacheRelation(relationName, STORE_TYPE_KEY, countKey)
|
||
}
|
||
return err
|
||
}
|
||
|
||
// 从缓存中获取指定模型名称的分页检索结果，会同时采用`RetrieveCount`中的方法获取检索结果的总数量。
|
||
func RetrievePagedSearch[T any](entityName string, conditions ...string) (*T, int64, error) {
|
||
searchKey := assembleSearchKey(entityName, conditions...)
|
||
countKey := assembleCountKey(entityName, conditions...)
|
||
instance, err := Retrieve[T](searchKey)
|
||
if err != nil {
|
||
return nil, -1, err
|
||
}
|
||
count, err := Retrieve[_CountRecord](countKey)
|
||
if err != nil {
|
||
return nil, -1, err
|
||
}
|
||
if instance == nil || count == nil {
|
||
log.Warn("检索结果或者检索总数为空。", zap.String("searchKey", searchKey), zap.String("countKey", countKey))
|
||
return nil, -1, nil
|
||
}
|
||
return instance, count.Count, nil
|
||
}
|
||
|
||
// 从缓存中删除指定模型名称的分页检索结果，会同时采用`AbolishCountEntity`中的方法删除检索结果的总数量。
|
||
func AbolishPagedSearch(entityName string) error {
|
||
pattern := fmt.Sprintf("%s:%s:*", TAG_SEARCH, strings.ToUpper(entityName))
|
||
err := DeleteAll(pattern)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
pattern = fmt.Sprintf("%s:%s:*", TAG_COUNT, strings.ToUpper(entityName))
|
||
return DeleteAll(pattern)
|
||
}
|