httpCtx 副本包

node/http-context/http-context-copy/response.go

@@ -0,0 +1,121 @@
+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
+}