Initial commit

app/interface/bbq/common/http/header.go

@@ -0,0 +1,24 @@
+package http
+
+import (
+	"fmt"
+
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/trace"
+)
+
+// WrapHeader 为返回头添加自定义字段
+func WrapHeader(ctx *bm.Context) {
+	// Traceid
+	tracer, _ := trace.FromContext(ctx.Context)
+	traceid := fmt.Sprintf("%s", tracer)
+	ctx.Writer.Header().Set("traceid", traceid)
+
+	// Sessionid
+	sid := ctx.Request.Header.Get("SessionID")
+	if sid == "" {
+		sid = SessionID(ctx)
+	}
+	ctx.Set("SessionID", sid)
+	ctx.Writer.Header().Set("SessionID", sid)
+}