上传进度条功能

README.md

@@ -49,3 +49,21 @@ func main() {
  gout.POST(":6666/compress").RequestUse(request.GzipDecompress()).SetBody(buf).Do()
 }
 ```
+### upload 进度条
+```go
+package main
+
+import (
+ "bytes"
+ "github.com/antlabs/gout-middleware/request"
+ "github.com/guonaihong/gout"
+)
+
+func main() {
+ gout.POST(":8080").RequestUse(request.ProgressBar(func(currBytes, totalBytes int) {
+
+ fmt.Printf("%d:%d-->%f%%\n", currBytes, totalBytes, float64(currBytes)/float64(totalBytes))
+ })).SetBody(strings.Repeat("1", 100000) /*构造大点的测试数据，这里换成真实业务数据*/).Do()
+}
+
+```

request/progress_bar.go

@@ -0,0 +1,38 @@
+package request
+
+import (
+ "io"
+ "io/ioutil"
+ "net/http"
+
+ api "github.com/guonaihong/gout/interface"
+)
+
+//上传进度条
+type progressBar struct {
+ callback func(currBytes, totalBytes int)
+ r io.Reader
+ currBytes int
+ totalBytes int
+}
+
+func (g *progressBar) Read(p []byte) (n int, err error) {
+ n, err = g.r.Read(p)
+ g.currBytes += n
+ if n > 0 && g.callback != nil {
+ g.callback(g.currBytes, g.totalBytes)
+ }
+
+ return
+}
+
+func (g *progressBar) ModifyRequest(req *http.Request) error {
+ g.r = req.Body
+ req.Body = ioutil.NopCloser(g)
+ g.totalBytes = int(req.ContentLength)
+ return nil
+}
+
+func ProgressBar(callback func(currBytes, totalBytes int)) api.RequestMiddler {
+ return &progressBar{callback: callback}
+}

request/progress_bar_test.go

@@ -0,0 +1,46 @@
+package request
+
+import (
+ "io"
+ "io/ioutil"
+ "net/http"
+ "net/http/httptest"
+ "strings"
+ "testing"
+
+ "github.com/gin-gonic/gin"
+ "github.com/guonaihong/gout"
+ "github.com/stretchr/testify/assert"
+)
+
+type readWait struct {
+ r io.Reader
+}
+
+func (r *readWait) Read(p []byte) (n int, err error) {
+ n, err = r.r.Read(p[:512])
+ return
+}
+
+func createProgressBarServer() *httptest.Server {
+ r := gin.New()
+ r.POST("/", func(c *gin.Context) {
+ r := &readWait{r: c.Request.Body}
+ _, _ = ioutil.ReadAll(r)
+
+ c.String(200, "hello")
+ })
+
+ return httptest.NewServer(http.HandlerFunc(r.ServeHTTP))
+}
+
+func TestProgressBar(t *testing.T) {
+ ts := createProgressBarServer()
+ call := false
+ gout.POST(ts.URL).RequestUse(ProgressBar(func(currBytes, totalBytes int) {
+ call = true
+ //fmt.Printf("%d:%d\n", currBytes, totalBytes)
+ })).SetBody(strings.Repeat("1", 100000)).Do()
+
+ assert.True(t, call)
+}