Feat: cancel aria2 download task

This commit is contained in:
HFO4
2020-02-06 10:07:46 +08:00
parent 1a93c9a2b9
commit c29695a40d
5 changed files with 62 additions and 15 deletions

View File

@@ -5,6 +5,7 @@ import (
model "github.com/HFO4/cloudreve/models"
"github.com/HFO4/cloudreve/pkg/util"
"github.com/zyxar/argo/rpc"
"os"
"path/filepath"
"strconv"
"strings"
@@ -44,7 +45,24 @@ func (client *RPCService) Status(task *model.Download) (rpc.StatusInfo, error) {
// Cancel 取消下载
func (client *RPCService) Cancel(task *model.Download) error {
// 取消下载任务
_, err := client.caller.Remove(task.GID)
if err != nil {
util.Log().Warning("无法取消离线下载任务[%s], %s", task.GID, err)
}
// 删除临时文件
util.Log().Debug("离线下载任务[%s]已取消1 分钟后删除临时文件", task.GID)
go func(task *model.Download) {
select {
case <-time.After(time.Duration(60) * time.Second):
err := os.RemoveAll(task.Parent)
if err != nil {
util.Log().Warning("无法删除离线下载临时目录[%s], %s", task.Parent, err)
}
}
}(task)
return err
}