mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-26 11:36:44 +08:00
106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
package mcache
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/cache/cachekey"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/database"
|
|
"github.com/openimsdk/tools/log"
|
|
"github.com/openimsdk/tools/s3/minio"
|
|
)
|
|
|
|
func NewMinioCache(cache database.Cache) minio.Cache {
|
|
return &minioCache{
|
|
cache: cache,
|
|
expireTime: time.Hour * 24 * 7,
|
|
}
|
|
}
|
|
|
|
type minioCache struct {
|
|
cache database.Cache
|
|
expireTime time.Duration
|
|
}
|
|
|
|
func (g *minioCache) getObjectImageInfoKey(key string) string {
|
|
return cachekey.GetObjectImageInfoKey(key)
|
|
}
|
|
|
|
func (g *minioCache) getMinioImageThumbnailKey(key string, format string, width int, height int) string {
|
|
return cachekey.GetMinioImageThumbnailKey(key, format, width, height)
|
|
}
|
|
|
|
func (g *minioCache) DelObjectImageInfoKey(ctx context.Context, keys ...string) error {
|
|
ks := make([]string, 0, len(keys))
|
|
for _, key := range keys {
|
|
ks = append(ks, g.getObjectImageInfoKey(key))
|
|
}
|
|
return g.cache.Del(ctx, ks)
|
|
}
|
|
|
|
func (g *minioCache) DelImageThumbnailKey(ctx context.Context, key string, format string, width int, height int) error {
|
|
return g.cache.Del(ctx, []string{g.getMinioImageThumbnailKey(key, format, width, height)})
|
|
}
|
|
|
|
func (g *minioCache) GetImageObjectKeyInfo(ctx context.Context, key string, fn func(ctx context.Context) (*minio.ImageInfo, error)) (*minio.ImageInfo, error) {
|
|
return getCache[*minio.ImageInfo](ctx, g.cache, g.getObjectImageInfoKey(key), g.expireTime, fn)
|
|
}
|
|
|
|
func (g *minioCache) GetThumbnailKey(ctx context.Context, key string, format string, width int, height int, minioCache func(ctx context.Context) (string, error)) (string, error) {
|
|
return getCache[string](ctx, g.cache, g.getMinioImageThumbnailKey(key, format, width, height), g.expireTime, minioCache)
|
|
}
|
|
|
|
func getCache[V any](ctx context.Context, cache database.Cache, key string, expireTime time.Duration, fn func(ctx context.Context) (V, error)) (V, error) {
|
|
getDB := func() (V, bool, error) {
|
|
res, err := cache.Get(ctx, []string{key})
|
|
if err != nil {
|
|
var val V
|
|
return val, false, err
|
|
}
|
|
var val V
|
|
if str, ok := res[key]; ok {
|
|
if json.Unmarshal([]byte(str), &val) != nil {
|
|
return val, false, err
|
|
}
|
|
return val, true, nil
|
|
}
|
|
return val, false, nil
|
|
}
|
|
dbVal, ok, err := getDB()
|
|
if err != nil {
|
|
return dbVal, err
|
|
}
|
|
if ok {
|
|
return dbVal, nil
|
|
}
|
|
lockValue, err := cache.Lock(ctx, key, time.Minute)
|
|
if err != nil {
|
|
return dbVal, err
|
|
}
|
|
defer func() {
|
|
if err := cache.Unlock(ctx, key, lockValue); err != nil {
|
|
log.ZError(ctx, "unlock cache key", err, "key", key, "value", lockValue)
|
|
}
|
|
}()
|
|
dbVal, ok, err = getDB()
|
|
if err != nil {
|
|
return dbVal, err
|
|
}
|
|
if ok {
|
|
return dbVal, nil
|
|
}
|
|
val, err := fn(ctx)
|
|
if err != nil {
|
|
return val, err
|
|
}
|
|
data, err := json.Marshal(val)
|
|
if err != nil {
|
|
return val, err
|
|
}
|
|
if err := cache.Set(ctx, key, string(data), expireTime); err != nil {
|
|
return val, err
|
|
}
|
|
return val, nil
|
|
}
|