优化fasthttp的header 逻辑

drivers/router/http-router/http-complete/complete.go

@@ -83,6 +83,7 @@ func (h *HttpComplete) Complete(org eocontext.EoContext) error {
 		}
 		lastErr = ctx.SendTo(scheme, node, timeOut)
 		if lastErr == nil {
+
 			return nil
 		}
 		log.Error("http upstream send error: ", lastErr)

node/http-context/context.go

@@ -31,7 +31,7 @@ type HttpContext struct {
 	ctx                context.Context
 	completeHandler    eoscContext.CompleteHandler
 	finishHandler      eoscContext.FinishHandler
-	app                 eoscContext.EoApp
+
 	balance             eoscContext.BalanceHandler
 	upstreamHostHandler eoscContext.UpstreamHostHandler
 	labels              map[string]string
@@ -138,6 +138,7 @@ func (ctx *HttpContext) SendTo(scheme string, node eoscContext.INode, timeout ti
 	if ctx.response.responseError != nil {
 		agent.setStatusCode(504)
 	} else {
+		ctx.response.ResponseHeader.refresh()
 		agent.setStatusCode(ctx.fastHttpRequestCtx.Response.StatusCode())
 	}
 
@@ -257,7 +258,6 @@ func (ctx *HttpContext) FastFinish() {
 
 	ctx.port = 0
 	ctx.ctx = nil
-	ctx.app = nil
 	ctx.balance = nil
 	ctx.upstreamHostHandler = nil
 	ctx.finishHandler = nil

node/http-context/header.go

@@ -2,11 +2,10 @@ package http_context
 
 import (
 	"bytes"
+	http_service "github.com/eolinker/eosc/eocontext/http-context"
 	"net/http"
 	"strings"
 
-	http_service "github.com/eolinker/eosc/eocontext/http-context"
-
 	"github.com/valyala/fasthttp"
 )
 
@@ -77,39 +76,47 @@ func (h *RequestHeader) SetHost(host string) {
 	h.header.SetHost(host)
 }
 
+type headerActionHandleFunc func(target *ResponseHeader, key string, value ...string)
+
+var (
+	headerActionAdd = func(target *ResponseHeader, key string, value ...string) {
+		target.cache.Add(key, value[0])
+		target.header.Add(key, value[0])
+
+	}
+	headerActionSet = func(target *ResponseHeader, key string, value ...string) {
+		target.cache.Set(key, value[0])
+		target.header.Set(key, value[0])
+	}
+	headerActionDel = func(target *ResponseHeader, key string, value ...string) {
+		target.cache.Del(key)
+		target.header.Del(key)
+	}
+)
+
+type headerAction struct {
+	Action headerActionHandleFunc
+	Key    string
+	Value  string
+}
 type ResponseHeader struct {
 	header *fasthttp.ResponseHeader
-	tmp    http.Header
+
+	cache      http.Header
+	actions    []*headerAction
+	afterProxy bool
 }
 
 func (r *ResponseHeader) reset(header *fasthttp.ResponseHeader) {
+
 	r.header = header
+	r.cache = http.Header{}
+	r.actions = nil
+	r.refresh()
+}
+func (r *ResponseHeader) refresh() {
 
-	r.tmp = nil
+	tmp := make(http.Header)
-}
-func (r *ResponseHeader) Finish() {
-	if r.tmp != nil {
-		for k, vs := range r.tmp {
-			r.header.Del(k)
-			for _, v := range vs {
-				r.header.Add(k, v)
-			}
-		}
-	}
-	r.reset(nil)
-}
-func NewResponseHeader(header *fasthttp.ResponseHeader) *ResponseHeader {
-	return &ResponseHeader{header: header}
-}
-
-func (r *ResponseHeader) GetHeader(name string) string {
-	return r.Headers().Get(name)
-}
-
-func (r *ResponseHeader) Headers() http.Header {
-
-	if r.tmp == nil {
-		r.tmp = make(http.Header)
 	hs := strings.Split(r.header.String(), "\r\n")
 	for _, t := range hs {
 		vs := strings.Split(t, ":")
@@ -117,34 +124,76 @@ func (r *ResponseHeader) Headers() http.Header {
 			if vs[0] == "" {
 				continue
 			}
-				r.tmp[vs[0]] = []string{""}
+			tmp[vs[0]] = []string{""}
 			continue
 		}
-			r.tmp[vs[0]] = []string{strings.TrimSpace(vs[1])}
+		tmp[vs[0]] = []string{strings.TrimSpace(vs[1])}
 	}
+	r.cache = tmp
+	for _, ac := range r.actions {
+		ac.Action(r, ac.Key, ac.Value)
 	}
-	return r.tmp
+	r.afterProxy = true
+	r.actions = nil
+
+}
+func (r *ResponseHeader) Finish() {
+
+	r.reset(nil)
+}
+
+func (r *ResponseHeader) GetHeader(name string) string {
+	return r.Headers().Get(name)
+}
+
+func (r *ResponseHeader) Headers() http.Header {
+	return r.cache
+
 }
 
 func (r *ResponseHeader) SetHeader(key, value string) {
-	if r.tmp != nil {
+
-		r.tmp.Set(key, value)
+	r.cache.Set(key, value)
-	}
+
+	if r.afterProxy {
 		r.header.Set(key, value)
+	} else {
+		r.actions = append(r.actions, &headerAction{
+			Key:    key,
+			Value:  value,
+			Action: headerActionSet,
+		})
+	}
 }
 
 func (r *ResponseHeader) AddHeader(key, value string) {
-	if r.tmp != nil {
+
-		r.tmp.Add(key, value)
+	r.cache.Add(key, value)
-	}
+
+	if r.afterProxy {
 		r.header.Add(key, value)
+	} else {
+		r.actions = append(r.actions, &headerAction{
+			Key:    key,
+			Value:  value,
+			Action: headerActionAdd,
+		})
+	}
 }
 
 func (r *ResponseHeader) DelHeader(key string) {
-	if r.tmp != nil {
+
-		r.tmp.Del(key)
+	r.cache.Del(key)
-	}
+	if r.afterProxy {
 		r.header.Del(key)
+
+	} else {
+		r.actions = append(r.actions, &headerAction{
+			Key:    key,
+			Action: headerActionDel,
+		})
+	}
+
 }
 
 func (h *RequestHeader) GetCookie(key string) string {