Merge pull request imroc#43 from maddie/last-progress

Report progress on upload and download finish
yuanjunliang · Aug 26, 2019 · e3a4bfb · e3a4bfb
2 parents fc8e479 + ba3ab25
commit e3a4bfb
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/req.go b/req.go
@@ -514,6 +514,11 @@ func (m *multipartHelper) Upload(req *http.Request) {
 			var current int64
 			buf := make([]byte, 1024)
 			var lastTime time.Time
+
+			defer func() {
+				m.uploadProgress(current, total)
+			}()
+
 			upload = func(w io.Writer, r io.Reader) error {
 				for {
 					n, err := r.Read(buf)

diff --git a/resp.go b/resp.go
@@ -124,6 +124,11 @@ func (r *Resp) download(file *os.File) error {
 	total := r.resp.ContentLength
 	var current int64
 	var lastTime time.Time
+
+	defer func() {
+		r.downloadProgress(current, total)
+	}()
+
 	for {
 		l, err := b.Read(p)
 		if l > 0 {