swarm/sctx/sctx.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

package sctx

import "context"

type (
    HTTPRequestIDKey struct{}
    requestHostKey   struct{}
    tagKey           struct{}
)

// SetHost sets the http request host in the context
func SetHost(ctx context.Context, domain string) context.Context {
    return context.WithValue(ctx, requestHostKey{}, domain)
}

// GetHost gets the request host from the context
func GetHost(ctx context.Context) string {
    v, ok := ctx.Value(requestHostKey{}).(string)
    if ok {
        return v
    }
    return ""
}

// SetTag sets the tag unique identifier in the context
func SetTag(ctx context.Context, tagId uint32) context.Context {
    return context.WithValue(ctx, tagKey{}, tagId)
}

// GetTag gets the tag unique identifier from the context
func GetTag(ctx context.Context) uint32 {
    v, ok := ctx.Value(tagKey{}).(uint32)
    if ok {
        return v
    }
    return 0
}