mirror of
https://github.com/eolinker/apinto
synced 2025-10-05 08:47:04 +08:00
122 lines
2.4 KiB
Go
122 lines
2.4 KiB
Go
package http_context_copy
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
http_service "github.com/eolinker/eosc/eocontext/http-context"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
var _ http_service.IResponse = (*Response)(nil)
|
|
|
|
type Response struct {
|
|
ResponseHeader
|
|
*fasthttp.Response
|
|
length int
|
|
responseTime time.Duration
|
|
proxyStatusCode int
|
|
responseError error
|
|
}
|
|
|
|
func (r *Response) ContentLength() int {
|
|
if r.length == 0 {
|
|
return r.Response.Header.ContentLength()
|
|
}
|
|
return r.length
|
|
}
|
|
|
|
func (r *Response) ContentType() string {
|
|
return string(r.Response.Header.ContentType())
|
|
}
|
|
|
|
func (r *Response) HeadersString() string {
|
|
return r.header.String()
|
|
}
|
|
|
|
func (r *Response) ResponseError() error {
|
|
return r.responseError
|
|
}
|
|
|
|
func (r *Response) ClearError() {
|
|
r.responseError = nil
|
|
}
|
|
func (r *Response) Finish() error {
|
|
r.Response = nil
|
|
r.ResponseHeader.reset(nil)
|
|
r.responseError = nil
|
|
r.proxyStatusCode = 0
|
|
return nil
|
|
}
|
|
func (r *Response) reset(resp *fasthttp.Response) {
|
|
r.Response = resp
|
|
r.ResponseHeader.reset(&resp.Header)
|
|
r.responseError = nil
|
|
r.proxyStatusCode = 0
|
|
|
|
}
|
|
|
|
func (r *Response) BodyLen() int {
|
|
return r.header.ContentLength()
|
|
}
|
|
|
|
func (r *Response) GetBody() []byte {
|
|
if strings.Contains(r.GetHeader("Content-Encoding"), "gzip") {
|
|
body, _ := r.BodyGunzip()
|
|
r.DelHeader("Content-Encoding")
|
|
r.SetHeader("Content-Length", strconv.Itoa(len(body)))
|
|
r.Response.SetBody(body)
|
|
}
|
|
|
|
return r.Response.Body()
|
|
}
|
|
|
|
func (r *Response) SetBody(bytes []byte) {
|
|
if strings.Contains(r.GetHeader("Content-Encoding"), "gzip") {
|
|
r.DelHeader("Content-Encoding")
|
|
}
|
|
r.Response.SetBody(bytes)
|
|
r.length = len(bytes)
|
|
r.SetHeader("Content-Length", strconv.Itoa(r.length))
|
|
r.responseError = nil
|
|
}
|
|
|
|
func (r *Response) StatusCode() int {
|
|
if r.responseError != nil {
|
|
return 504
|
|
}
|
|
return r.Response.StatusCode()
|
|
}
|
|
|
|
func (r *Response) Status() string {
|
|
return strconv.Itoa(r.StatusCode())
|
|
}
|
|
|
|
func (r *Response) SetStatus(code int, status string) {
|
|
r.Response.SetStatusCode(code)
|
|
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) SetResponseTime(t time.Duration) {
|
|
r.responseTime = t
|
|
}
|
|
|
|
func (r *Response) ResponseTime() time.Duration {
|
|
return r.responseTime
|
|
}
|