feat: task progress

This commit is contained in:
krau
2024-10-12 13:03:11 +08:00
parent c8376f78a2
commit f58e28412f
+5
View File
@@ -21,6 +21,7 @@ func processPendingTask(task types.Task) error {
return err return err
} }
logger.L.Debugf("Start downloading file: %s", task.FileName) logger.L.Debugf("Start downloading file: %s", task.FileName)
bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "正在下载文件...")
dest, err := message.Download(&telegram.DownloadOptions{ dest, err := message.Download(&telegram.DownloadOptions{
FileName: common.GetCacheFilePath(task.FileName), FileName: common.GetCacheFilePath(task.FileName),
// ProgressCallback: func(totalBytes, downloadedBytes int64) {}, // ProgressCallback: func(totalBytes, downloadedBytes int64) {},
@@ -42,9 +43,11 @@ func processPendingTask(task types.Task) error {
task.StoragePath = task.FileName task.StoragePath = task.FileName
} }
bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "下载完成, 正在转存文件...")
if err := storage.Save(task.Storage, task.Ctx, dest, task.StoragePath); err != nil { if err := storage.Save(task.Storage, task.Ctx, dest, task.StoragePath); err != nil {
return err return err
} }
return nil return nil
} }
@@ -71,8 +74,10 @@ func worker(queue *queue.TaskQueue, semaphore chan struct{}) {
queue.AddTask(task) queue.AddTask(task)
case types.Succeeded: case types.Succeeded:
logger.L.Infof("Task succeeded: %s", task.String()) logger.L.Infof("Task succeeded: %s", task.String())
bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "文件保存成功")
case types.Failed: case types.Failed:
logger.L.Errorf("Task failed: %s", task.String()) logger.L.Errorf("Task failed: %s", task.String())
bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "文件保存失败")
case types.Canceled: case types.Canceled:
logger.L.Infof("Task canceled: %s", task.String()) logger.L.Infof("Task canceled: %s", task.String())
default: default: