context新增原始响应状态码及其接口,并且转发至上游时记录响应时间

This commit is contained in:
chenjiekun
2021-12-20 16:32:34 +08:00
parent e52c3b4745
commit 30156b9ba8
2 changed files with 26 additions and 1 deletions

View File

@@ -14,7 +14,8 @@ var _ http_service.IResponse = (*Response)(nil)
type Response struct {
*ResponseHeader
*fasthttp.Response
responseError error
proxyStatusCode int
responseError error
}
func (r *Response) HeadersString() string {
@@ -69,6 +70,19 @@ func (r *Response) SetStatus(code int, status string) {
r.responseError = nil
}
//原始的响应状态码
func (r *Response) ProxyStatusCode() int {
return r.proxyStatusCode
}
func (r *Response) ProxyStatus() string {
return strconv.Itoa(r.proxyStatusCode)
}
func (r *Response) SetProxyStatus(code int, status string) {
r.proxyStatusCode = code
}
func (r *Response) SetBody(bytes []byte) {
r.Response.SetBody(bytes)
r.responseError = nil