forked from liuzhuoling2011/BaiduPCS-Go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
uploader.go
116 lines (93 loc) · 2.37 KB
/
uploader.go
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// Package uploader 上传包
package uploader
import (
"github.com/iikira/BaiduPCS-Go/requester"
"github.com/iikira/BaiduPCS-Go/requester/rio"
"net/http"
"time"
)
//CheckFunc 上传完成的检测函数
type CheckFunc func(resp *http.Response, uploadErr error)
// Uploader 上传
type Uploader struct {
url string // 上传地址
readed64 Readed64 // 要上传的对象
contentType string
client *requester.HTTPClient
executeTime time.Time
executed bool
finished chan struct{}
checkFunc CheckFunc
onExecute func()
onFinish func()
}
// NewUploader 返回 uploader 对象, url: 上传地址, readerlen64: 实现 rio.ReaderLen64 接口的对象, 例如文件
func NewUploader(url string, readedlen64 rio.ReaderLen64) (uploader *Uploader) {
uploader = &Uploader{
url: url,
readed64: NewReaded64(readedlen64),
}
return
}
func (u *Uploader) lazyInit() {
if u.finished == nil {
u.finished = make(chan struct{})
}
if u.client == nil {
u.client = requester.NewHTTPClient()
}
u.client.SetTimeout(0)
u.client.SetResponseHeaderTimeout(0)
}
// SetClient 设置http客户端
func (u *Uploader) SetClient(c *requester.HTTPClient) {
u.client = c
}
//SetContentType 设置Content-Type
func (u *Uploader) SetContentType(contentType string) {
u.contentType = contentType
}
//SetCheckFunc 设置上传完成的检测函数
func (u *Uploader) SetCheckFunc(checkFunc CheckFunc) {
u.checkFunc = checkFunc
}
// Execute 执行上传, 收到返回值信号则为上传结束
func (u *Uploader) Execute() {
trigger(u.onExecute)
// 开始上传
u.executeTime = time.Now()
u.executed = true
resp, _, err := u.execute()
// 上传结束
close(u.finished)
if u.checkFunc != nil {
u.checkFunc(resp, err)
}
trigger(u.onFinish) // 触发上传结束的事件
}
func (u *Uploader) execute() (resp *http.Response, code int, err error) {
u.lazyInit()
header := map[string]string{}
if u.contentType != "" {
header["Content-Type"] = u.contentType
}
resp, err = u.client.Req("POST", u.url, u.readed64, header)
if err != nil {
return nil, 2, err
}
return resp, 0, nil
}
// trigger 用于触发事件
func trigger(fn func()) {
if fn != nil {
go fn()
}
}
// OnExecute 任务开始时触发的事件
func (u *Uploader) OnExecute(fn func()) {
u.onExecute = fn
}
// OnFinish 任务完成时触发的事件
func (u *Uploader) OnFinish(fn func()) {
u.onFinish = fn
}