mirror of
https://github.com/krau/SaveAny-Bot.git
synced 2026-06-08 17:10:47 +08:00
fix: add unique id to task struct to avoid duplicate file name overwrite, close #59
This commit is contained in:
@@ -27,6 +27,13 @@ import (
|
||||
|
||||
func processPendingTask(task *types.Task) error {
|
||||
common.Log.Debugf("Start processing task: %s", task.String())
|
||||
|
||||
if task.FileName() != "" && !task.IsTelegraph && task.File.FileSize != 0 && task.FileDBID != 0 {
|
||||
ext := path.Ext(task.FileName())
|
||||
name := task.FileName()[:len(task.FileName())-len(ext)]
|
||||
task.File.FileName = fmt.Sprintf("%s_%d%s", name, task.FileDBID, ext)
|
||||
}
|
||||
|
||||
if task.FileName() == "" {
|
||||
task.File.FileName = fmt.Sprintf("%d_%d_%s", task.FileChatID, task.FileMessageID, task.File.Hash())
|
||||
}
|
||||
@@ -60,6 +67,7 @@ func processPendingTask(task *types.Task) error {
|
||||
|
||||
notsupportStreamStorage, notsupportStream := taskStorage.(storage.StorageNotSupportStream)
|
||||
cancelMarkUp := getCancelTaskMarkup(task)
|
||||
|
||||
if config.Cfg.Stream {
|
||||
if !notsupportStream {
|
||||
text, entities := buildProgressMessageEntity(task, 0, task.StartTime, 0)
|
||||
|
||||
Reference in New Issue
Block a user