优化文件上传

Jrohy · Dec 24, 2021 · 8d72f2f · 8d72f2f
1 parent 019418f
commit 8d72f2f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/controller/file.go b/controller/file.go
@@ -187,7 +187,7 @@ func UploadProgressWs(c *gin.Context) *ResponseBody {
 		}
 		for _, v := range core.WcList {
 			if v.Id == id {
-				wsConn.WriteMessage(1, []byte(strconv.FormatInt(v.Total, 10)))
+				wsConn.WriteMessage(1, []byte(strconv.Itoa(v.Total)))
 				find = true
 				if !ready {
 					ready = true

diff --git a/core/models.go b/core/models.go
@@ -13,14 +13,14 @@ var WcList []*WriteCounter
 
 // WriteCounter 结构体
 type WriteCounter struct {
-	Total int64
+	Total int
 	Id    string
 }
 
 // Write: implement Write interface to write bytes from ssh server into bytes.Buffer.
 func (wc *WriteCounter) Write(p []byte) (int, error) {
 	n := len(p)
-	wc.Total += int64(n)
+	wc.Total += n
 	return n, nil
 }