增加返回cookie的信息

Author: 王聪

request.go

@@ -54,6 +54,7 @@ type Result struct {
 	err         error
 	statusCode  int
 	headers     map[string][]string
+	cookies     []*http.Cookie
 
 	decoder Decoder
 }
@@ -108,6 +109,10 @@ func (r Result) Headers() map[string][]string {
 	return r.headers
 }
 
+func (r Result) Cookies() []*http.Cookie {
+	return r.cookies
+}
+
 // StatusCode returns the HTTP status code of the request. (Only valid if no
 // error was returned.)
 func (r Result) StatusCode(statusCode *int) Result {
@@ -453,6 +458,7 @@ func (r *Request) transformResponse(resp *http.Response, req *http.Request) Resu
 			decoder:     decoder,
 			err:         r.transformUnstructuredResponseError(resp, req, body),
 			headers:     resp.Header,
+			cookies:     resp.Cookies(),
 		}
 	}
 
@@ -462,6 +468,7 @@ func (r *Request) transformResponse(resp *http.Response, req *http.Request) Resu
 		statusCode:  resp.StatusCode,
 		decoder:     decoder,
 		headers:     resp.Header,
+		cookies:     resp.Cookies(),
 	}
 }
 

request_test.go

@@ -31,6 +31,10 @@ func TestRequest_Do(t *testing.T) {
 	resp := NewRequest("https://www.iphpt.com/", "POST").
 		Body([]byte(body)).Do()
 	t.Log(resp.headers)
+	cookies := resp.cookies
+	for _, v := range cookies {
+		t.Log(v.String())
+	}
 
 }