support context.Context

bilalyasar · Nov 25, 2018 · b355b6f · b355b6f
1 parent 38b12ac
commit b355b6f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/req.go b/req.go
@@ -3,6 +3,7 @@ package req
 import (
 	"bytes"
 	"compress/gzip"
+	"context"
 	"encoding/json"
 	"encoding/xml"
 	"errors"
@@ -254,6 +255,8 @@ func (r *Req) Do(method, rawurl string, vs ...interface{}) (resp *Resp, err erro
 			resp.downloadProgress = vv
 		case func(int64, int64):
 			progress = vv
+		case context.Context:
+			req = req.WithContext(vv)
 		case error:
 			return nil, vv
 		}