feat(task): add speed monitor (#7655)

internal/fs/put.go

@@ -10,10 +10,11 @@ import (
 	"github.com/alist-org/alist/v3/internal/task"
 	"github.com/pkg/errors"
 	"github.com/xhofe/tache"
+	"time"
 )
 
 type UploadTask struct {
-	task.TaskWithCreator
+	task.TaskExtension
 	storage          driver.Driver
 	dstDirActualPath string
 	file             model.FileStreamer
@@ -28,13 +29,16 @@ func (t *UploadTask) GetStatus() string {
 }
 
 func (t *UploadTask) Run() error {
+	t.ClearEndTime()
+	t.SetStartTime(time.Now())
+	defer func() { t.SetEndTime(time.Now()) }()
 	return op.Put(t.Ctx(), t.storage, t.dstDirActualPath, t.file, t.SetProgress, true)
 }
 
 var UploadTaskManager *tache.Manager[*UploadTask]
 
 // putAsTask add as a put task and return immediately
-func putAsTask(ctx context.Context, dstDirPath string, file model.FileStreamer) (task.TaskInfoWithCreator, error) {
+func putAsTask(ctx context.Context, dstDirPath string, file model.FileStreamer) (task.TaskExtensionInfo, error) {
 	storage, dstDirActualPath, err := op.GetStorageAndActualPath(dstDirPath)
 	if err != nil {
 		return nil, errors.WithMessage(err, "failed get storage")
@@ -52,13 +56,14 @@ func putAsTask(ctx context.Context, dstDirPath string, file model.FileStreamer)
 	}
 	taskCreator, _ := ctx.Value("user").(*model.User) // taskCreator is nil when convert failed
 	t := &UploadTask{
-		TaskWithCreator: task.TaskWithCreator{
+		TaskExtension: task.TaskExtension{
 			Creator: taskCreator,
 		},
 		storage:          storage,
 		dstDirActualPath: dstDirActualPath,
 		file:             file,
 	}
+	t.SetTotalBytes(file.GetSize())
 	UploadTaskManager.Add(t)
 	return t, nil
 }