1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| package protocol
|
| import (
| "context"
| )
|
| type key int
|
| const (
| requestKey key = iota
| )
|
| func ContextWithRequestHeader(ctx context.Context, request *RequestHeader) context.Context {
| return context.WithValue(ctx, requestKey, request)
| }
|
| func RequestHeaderFromContext(ctx context.Context) *RequestHeader {
| request := ctx.Value(requestKey)
| if request == nil {
| return nil
| }
| return request.(*RequestHeader)
| }
|
|
