open-consul/agent/consul/context.go
Ronald dd0e8eec14
copyright headers for agent folder (#16704)
* copyright headers for agent folder

* Ignore test data files

* fix proto files and remove headers in agent/uiserver folder

* ignore deep-copy files
2023-03-28 14:39:22 -04:00

24 lines
471 B
Go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0
package consul
import (
"context"
"net"
)
type contextKeyRemoteAddr struct{}
func ContextWithRemoteAddr(ctx context.Context, addr net.Addr) context.Context {
return context.WithValue(ctx, contextKeyRemoteAddr{}, addr)
}
func RemoteAddrFromContext(ctx context.Context) (net.Addr, bool) {
v := ctx.Value(contextKeyRemoteAddr{})
if v == nil {
return nil, false
}
return v.(net.Addr), true
}