File size: 1,770 Bytes
619f93d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 | package task
import (
"context"
"time"
"github.com/OpenListTeam/OpenList/v4/internal/conf"
"github.com/OpenListTeam/OpenList/v4/internal/model"
"github.com/OpenListTeam/tache"
)
type TaskExtension struct {
tache.Base
Creator *model.User
startTime *time.Time
endTime *time.Time
TotalBytes int64
ApiUrl string
}
func (t *TaskExtension) SetCtx(ctx context.Context) {
if t.Creator != nil {
ctx = context.WithValue(ctx, conf.UserKey, t.Creator)
}
if len(t.ApiUrl) > 0 {
ctx = context.WithValue(ctx, conf.ApiUrlKey, t.ApiUrl)
}
t.Base.SetCtx(ctx)
}
func (t *TaskExtension) SetCreator(creator *model.User) {
t.Creator = creator
t.Persist()
}
func (t *TaskExtension) GetCreator() *model.User {
return t.Creator
}
func (t *TaskExtension) SetStartTime(startTime time.Time) {
t.startTime = &startTime
}
func (t *TaskExtension) GetStartTime() *time.Time {
return t.startTime
}
func (t *TaskExtension) SetEndTime(endTime time.Time) {
t.endTime = &endTime
}
func (t *TaskExtension) GetEndTime() *time.Time {
return t.endTime
}
func (t *TaskExtension) ClearEndTime() {
t.endTime = nil
}
func (t *TaskExtension) SetTotalBytes(totalBytes int64) {
t.TotalBytes = totalBytes
}
func (t *TaskExtension) GetTotalBytes() int64 {
return t.TotalBytes
}
func (t *TaskExtension) SetRetry(retry int, maxRetry int) {
t.Base.SetRetry(retry, maxRetry)
if retry > 0 || !conf.Conf.Tasks.AllowRetryCanceled || t.Ctx() == nil {
return
}
select {
case <-t.Ctx().Done():
ctx, cancel := context.WithCancel(context.Background())
t.SetCtx(ctx)
t.SetCancelFunc(cancel)
default:
}
}
type TaskExtensionInfo interface {
tache.TaskWithInfo
GetCreator() *model.User
GetStartTime() *time.Time
GetEndTime() *time.Time
GetTotalBytes() int64
}
|