mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 20:28:17 +08:00
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gclient
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
|
|
"github.com/gogf/gf/v2/internal/intlog"
|
|
"github.com/gogf/gf/v2/internal/utils"
|
|
)
|
|
|
|
// dumpTextFormat is the format of the dumped raw string
|
|
const dumpTextFormat = `+---------------------------------------------+
|
|
| %s |
|
|
+---------------------------------------------+
|
|
%s
|
|
%s
|
|
`
|
|
|
|
// getResponseBody returns the text of the response body.
|
|
func getResponseBody(res *http.Response) string {
|
|
if res.Body == nil {
|
|
return ""
|
|
}
|
|
bodyContent, _ := ioutil.ReadAll(res.Body)
|
|
res.Body = utils.NewReadCloser(bodyContent, true)
|
|
return string(bodyContent)
|
|
}
|
|
|
|
// RawRequest returns the raw content of the request.
|
|
func (r *Response) RawRequest() string {
|
|
// Response can be nil.
|
|
if r == nil {
|
|
return ""
|
|
}
|
|
if r.request == nil {
|
|
return ""
|
|
}
|
|
// DumpRequestOut writes more request headers than DumpRequest, such as User-Agent.
|
|
bs, err := httputil.DumpRequestOut(r.request, false)
|
|
if err != nil {
|
|
intlog.Errorf(r.request.Context(), `%+v`, err)
|
|
return ""
|
|
}
|
|
return fmt.Sprintf(
|
|
dumpTextFormat,
|
|
"REQUEST ",
|
|
string(bs),
|
|
r.requestBody,
|
|
)
|
|
}
|
|
|
|
// RawResponse returns the raw content of the response.
|
|
func (r *Response) RawResponse() string {
|
|
// Response might be nil.
|
|
if r == nil || r.Response == nil {
|
|
return ""
|
|
}
|
|
bs, err := httputil.DumpResponse(r.Response, false)
|
|
if err != nil {
|
|
intlog.Errorf(r.request.Context(), `%+v`, err)
|
|
return ""
|
|
}
|
|
|
|
return fmt.Sprintf(
|
|
dumpTextFormat,
|
|
"RESPONSE",
|
|
string(bs),
|
|
getResponseBody(r.Response),
|
|
)
|
|
}
|
|
|
|
// Raw returns the raw text of the request and the response.
|
|
func (r *Response) Raw() string {
|
|
return fmt.Sprintf("%s\n%s", r.RawRequest(), r.RawResponse())
|
|
}
|
|
|
|
// RawDump outputs the raw text of the request and the response to stdout.
|
|
func (r *Response) RawDump() {
|
|
fmt.Println(r.Raw())
|
|
}
|