support *os.File in request param

khorevaa · Sep 25, 2018 · bc0679a · bc0679a
1 parent 293bdc8
commit bc0679a
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/req.go b/req.go
@@ -429,9 +429,17 @@ func setContentType(req *http.Request, contentType string) {
 
 func setBodyReader(req *http.Request, resp *Resp, rd io.Reader) func() {
 	var rc io.ReadCloser
-	if trc, ok := rd.(io.ReadCloser); ok {
-		rc = trc
-	} else {
+	switch r := rd.(type) {
+	case *os.File:
+		stat, err := r.Stat()
+		if err == nil {
+			req.ContentLength = stat.Size()
+		}
+		rc = r
+
+	case io.ReadCloser:
+		rc = r
+	default:
 		rc = ioutil.NopCloser(rd)
 	}
 	bw := &bodyWrapper{