traefik/pkg/server/provider/provider.go

46 lines
1 KiB
Go
Raw Normal View History

package provider
import (
"context"
"strings"
2022-11-21 17:36:05 +00:00
"github.com/rs/zerolog/log"
)
type contextKey int
const (
key contextKey = iota
)
2020-05-11 10:06:07 +00:00
// AddInContext Adds the provider name in the context.
func AddInContext(ctx context.Context, elementName string) context.Context {
parts := strings.Split(elementName, "@")
if len(parts) == 1 {
2022-11-21 17:36:05 +00:00
log.Ctx(ctx).Debug().Msgf("Could not find a provider for %s", elementName)
return ctx
}
if name, ok := ctx.Value(key).(string); ok && name == parts[1] {
return ctx
}
return context.WithValue(ctx, key, parts[1])
}
// GetQualifiedName Gets the fully qualified name.
func GetQualifiedName(ctx context.Context, elementName string) string {
parts := strings.Split(elementName, "@")
if len(parts) == 1 {
if providerName, ok := ctx.Value(key).(string); ok {
return MakeQualifiedName(providerName, parts[0])
}
}
return elementName
}
2020-05-11 10:06:07 +00:00
// MakeQualifiedName Creates a qualified name for an element.
2020-07-07 12:42:03 +00:00
func MakeQualifiedName(providerName, elementName string) string {
return elementName + "@" + providerName
}