ginuerzh
93 lines
1.7 KiB
Go
93 lines
1.7 KiB
Go
package ctx
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
)
|
|
|
|
type Context interface {
|
|
Context() context.Context
|
|
}
|
|
|
|
type WithContext interface {
|
|
WithContext(ctx context.Context)
|
|
}
|
|
|
|
type srcAddrKey struct{}
|
|
|
|
func ContextWithSrcAddr(ctx context.Context, addr net.Addr) context.Context {
|
|
return context.WithValue(ctx, srcAddrKey{}, addr)
|
|
}
|
|
|
|
func SrcAddrFromContext(ctx context.Context) net.Addr {
|
|
v, _ := ctx.Value(srcAddrKey{}).(net.Addr)
|
|
return v
|
|
}
|
|
|
|
type dstAddrKey struct{}
|
|
|
|
func ContextWithDstAddr(ctx context.Context, addr net.Addr) context.Context {
|
|
return context.WithValue(ctx, dstAddrKey{}, addr)
|
|
}
|
|
|
|
func DstAddrFromContext(ctx context.Context) net.Addr {
|
|
v, _ := ctx.Value(dstAddrKey{}).(net.Addr)
|
|
return v
|
|
}
|
|
|
|
type (
|
|
Sid string
|
|
// sidKey saves the session ID.
|
|
sidKey struct{}
|
|
)
|
|
|
|
func (s Sid) String() string {
|
|
return string(s)
|
|
}
|
|
|
|
func ContextWithSid(ctx context.Context, sid Sid) context.Context {
|
|
return context.WithValue(ctx, sidKey{}, sid)
|
|
}
|
|
|
|
func SidFromContext(ctx context.Context) Sid {
|
|
v, _ := ctx.Value(sidKey{}).(Sid)
|
|
return v
|
|
}
|
|
|
|
type (
|
|
// hashKey saves the hash source for Selector.
|
|
hashKey struct{}
|
|
Hash struct {
|
|
Source string
|
|
}
|
|
)
|
|
|
|
func ContextWithHash(ctx context.Context, hash *Hash) context.Context {
|
|
return context.WithValue(ctx, hashKey{}, hash)
|
|
}
|
|
|
|
func HashFromContext(ctx context.Context) *Hash {
|
|
if v, _ := ctx.Value(hashKey{}).(*Hash); v != nil {
|
|
return v
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type (
|
|
ClientID string
|
|
clientIDKey struct{}
|
|
)
|
|
|
|
func (s ClientID) String() string {
|
|
return string(s)
|
|
}
|
|
|
|
func ContextWithClientID(ctx context.Context, clientID ClientID) context.Context {
|
|
return context.WithValue(ctx, clientIDKey{}, clientID)
|
|
}
|
|
|
|
func ClientIDFromContext(ctx context.Context) ClientID {
|
|
v, _ := ctx.Value(clientIDKey{}).(ClientID)
|
|
return v
|
|
}
|