修复当关闭缓存功能时异常报错的bug

rroy233 · Feb 3, 2024 · 8e54427 · 8e54427
1 parent 49163d5
commit 8e54427
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/handler/DownloadStickerSetQuery.go b/handler/DownloadStickerSetQuery.go
@@ -276,8 +276,10 @@ func downloadWorker(ctx context.Context, queue chan tgbotapi.Sticker, task *down
 					atomic.AddInt32(&task.failed, 1)
 					continue
 				}
-				if _, err := db.CacheSticker(sticker, convertTask.OutputFilePath); err != nil {
-					logger.Error.Printf("DownloadStickerSetQuery[%d/%d]-failed to Save Cache:%s,%s", i, sum, err.Error(), stickerInfo)
+				if config.Get().Cache.Enabled == true {
+					if _, err := db.CacheSticker(sticker, convertTask.OutputFilePath); err != nil {
+						logger.Error.Printf("DownloadStickerSetQuery[%d/%d]-failed to Save Cache:%s,%s", i, sum, err.Error(), stickerInfo)
+					}
 				}
 			}
 			atomic.AddInt32(&task.finished, 1)

diff --git a/handler/StickersMessage.go b/handler/StickersMessage.go
@@ -111,13 +111,15 @@ func StickerMessage(update tgbotapi.Update) {
 		}
 
 		//CacheSticker
-		cacheItem, err = db.CacheSticker(*update.Message.Sticker, convertTask.OutputFilePath)
-		if err != nil {
-			logger.Error.Println(userInfo+"CacheSticker Error ", err)
-		} else {
-			cacheItem.ConvertedFileID = sentMsg.Document.FileID
-			if err := cacheItem.Update(); err != nil {
-				logger.Error.Println(userInfo+"failed to update cache:", err)
+		if config.Get().Cache.Enabled == true {
+			cacheItem, err = db.CacheSticker(*update.Message.Sticker, convertTask.OutputFilePath)
+			if err != nil {
+				logger.Error.Println(userInfo+"CacheSticker Error ", err)
+			} else {
+				cacheItem.ConvertedFileID = sentMsg.Document.FileID
+				if err := cacheItem.Update(); err != nil {
+					logger.Error.Println(userInfo+"failed to update cache:", err)
+				}
 			}
 		}
 		utils.RemoveFile(outPath)