45 lines
1.5 KiB
45 lines
1.5 KiB
package redis
|
|
|
|
import (
|
|
"context"
|
|
"github.com/dtm-labs/rockscache"
|
|
"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/open-im-server/v3/pkg/common/storage/model"
|
|
"github.com/redis/go-redis/v9"
|
|
"time"
|
|
)
|
|
|
|
func NewApplicationRedisCache(db database.Application, rdb redis.UniversalClient) *ApplicationRedisCache {
|
|
return &ApplicationRedisCache{
|
|
db: db,
|
|
rcClient: rockscache.NewClient(rdb, *GetRocksCacheOptions()),
|
|
deleter: NewBatchDeleterRedis(rdb, GetRocksCacheOptions(), nil),
|
|
expireTime: time.Hour * 24 * 7,
|
|
}
|
|
}
|
|
|
|
type ApplicationRedisCache struct {
|
|
db database.Application
|
|
rcClient *rockscache.Client
|
|
deleter *BatchDeleterRedis
|
|
expireTime time.Duration
|
|
}
|
|
|
|
func (a *ApplicationRedisCache) LatestVersion(ctx context.Context, platform string, hot bool) (*model.Application, error) {
|
|
return getCache(ctx, a.rcClient, cachekey.GetApplicationLatestVersionKey(platform, hot), a.expireTime, func(ctx context.Context) (*model.Application, error) {
|
|
return a.db.LatestVersion(ctx, platform, hot)
|
|
})
|
|
}
|
|
|
|
func (a *ApplicationRedisCache) DeleteCache(ctx context.Context, platforms []string) error {
|
|
if len(platforms) == 0 {
|
|
return nil
|
|
}
|
|
keys := make([]string, 0, len(platforms)*2)
|
|
for _, platform := range platforms {
|
|
keys = append(keys, cachekey.GetApplicationLatestVersionKey(platform, true), cachekey.GetApplicationLatestVersionKey(platform, false))
|
|
}
|
|
return a.deleter.ExecDelWithKeys(ctx, keys)
|
|
}
|