GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-05-11 18:49:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: refactor caching logic to use gocache for better compatibility

Browse Source
This commit is contained in:
krau
2025-04-12 15:07:43 +08:00
parent 166c27c70f
commit 725acd0199
4 changed files with 90 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
bot/utils.go
View File

@@ -180,12 +180,11 @@ func FileFromMedia(media tg.MessageMediaClass, customFileName string) (*types.Fi
func FileFromMessage(ctx *ext.Context, chatID int64, messageID int, customFileName string) (*types.File, error) {
key := fmt.Sprintf("file:%d:%d", chatID, messageID)
common.Log.Debugf("Getting file: %s", key)
var cachedFile types.File
err := common.Cache.Get(key, &cachedFile)
cachedFile, err := common.CacheGet[*types.File](ctx, key)
if err == nil {
return &cachedFile, nil
return cachedFile, nil
}
common.Log.Debugf("Getting file: %s", key)
message, err := GetTGMessage(ctx, chatID, messageID)
if err != nil {
return nil, err
@@ -194,13 +193,18 @@ func FileFromMessage(ctx *ext.Context, chatID int64, messageID int, customFileNa
if err != nil {
return nil, err
}
if err := common.Cache.Set(key, file, 3600); err != nil {
if err := common.CacheSet(ctx, key, file); err != nil {
common.Log.Errorf("Failed to cache file: %s", err)
}
return file, nil
}
func GetTGMessage(ctx *ext.Context, chatId int64, messageID int) (*tg.Message, error) {
key := fmt.Sprintf("message:%d:%d", chatId, messageID)
cacheMessage, err := common.CacheGet[*tg.Message](ctx, key)
if err == nil {
return cacheMessage, nil
}
common.Log.Debugf("Fetching message: %d", messageID)
messages, err := ctx.GetMessages(chatId, []tg.InputMessageClass{&tg.InputMessageID{ID: messageID}})
if err != nil {
@@ -214,6 +218,9 @@ func GetTGMessage(ctx *ext.Context, chatId int64, messageID int) (*tg.Message, e
if !ok {
return nil, fmt.Errorf("unexpected message type: %T", msg)
}
if err := common.CacheSet(ctx, key, tgMessage); err != nil {
common.Log.Errorf("Failed to cache message: %s", err)
}
return tgMessage, nil
}