mirror of
https://github.com/opentofu/opentofu.git
synced 2025-01-24 15:36:26 -06:00
872371d21d
Previous deps merge had been captured by an older version of godep,
updating godep and rerunning...
godep save $(go list ./... | grep -v /vendor/)
...yielded this diff.
New `godep` also produced a warning to pin to the major Go version
instead of the minor one, which we do here as well.
59 lines
1.8 KiB
Go
59 lines
1.8 KiB
Go
package internal
|
|
|
|
import (
|
|
"github.com/golang/protobuf/proto"
|
|
netcontext "golang.org/x/net/context"
|
|
)
|
|
|
|
type callOverrideFunc func(ctx netcontext.Context, service, method string, in, out proto.Message) error
|
|
|
|
var callOverrideKey = "holds a callOverrideFunc"
|
|
|
|
func WithCallOverride(ctx netcontext.Context, f callOverrideFunc) netcontext.Context {
|
|
return netcontext.WithValue(ctx, &callOverrideKey, f)
|
|
}
|
|
|
|
type logOverrideFunc func(level int64, format string, args ...interface{})
|
|
|
|
var logOverrideKey = "holds a logOverrideFunc"
|
|
|
|
func WithLogOverride(ctx netcontext.Context, f logOverrideFunc) netcontext.Context {
|
|
return netcontext.WithValue(ctx, &logOverrideKey, f)
|
|
}
|
|
|
|
var appIDOverrideKey = "holds a string, being the full app ID"
|
|
|
|
func WithAppIDOverride(ctx netcontext.Context, appID string) netcontext.Context {
|
|
return netcontext.WithValue(ctx, &appIDOverrideKey, appID)
|
|
}
|
|
|
|
var namespaceKey = "holds the namespace string"
|
|
|
|
func WithNamespace(ctx netcontext.Context, ns string) netcontext.Context {
|
|
return netcontext.WithValue(ctx, &namespaceKey, ns)
|
|
}
|
|
|
|
func NamespaceFromContext(ctx netcontext.Context) string {
|
|
// If there's no namespace, return the empty string.
|
|
ns, _ := ctx.Value(&namespaceKey).(string)
|
|
return ns
|
|
}
|
|
|
|
// FullyQualifiedAppID returns the fully-qualified application ID.
|
|
// This may contain a partition prefix (e.g. "s~" for High Replication apps),
|
|
// or a domain prefix (e.g. "example.com:").
|
|
func FullyQualifiedAppID(ctx netcontext.Context) string {
|
|
if id, ok := ctx.Value(&appIDOverrideKey).(string); ok {
|
|
return id
|
|
}
|
|
return fullyQualifiedAppID(ctx)
|
|
}
|
|
|
|
func Logf(ctx netcontext.Context, level int64, format string, args ...interface{}) {
|
|
if f, ok := ctx.Value(&logOverrideKey).(logOverrideFunc); ok {
|
|
f(level, format, args...)
|
|
return
|
|
}
|
|
logf(fromContext(ctx), level, format, args...)
|
|
}
|