You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

24 lines
506 B

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)
}