mirror of https://github.com/hashicorp/consul
94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package resource
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/hashicorp/consul/proto-public/pbresource"
|
|
)
|
|
|
|
// ReferenceKey is the pointer-free representation of a ReferenceOrID
|
|
// suitable for a go map key.
|
|
type ReferenceKey struct {
|
|
GVK string
|
|
Partition string // Tenancy.*
|
|
Namespace string // Tenancy.*
|
|
PeerName string // Tenancy.*
|
|
Name string
|
|
}
|
|
|
|
// String returns a string representation of the ReferenceKey. This should not
|
|
// be relied upon nor parsed and is provided just for debugging and logging
|
|
// reasons.
|
|
//
|
|
// This format should be aligned with IDToString and ReferenceToString.
|
|
func (r ReferenceKey) String() string {
|
|
return fmt.Sprintf("%s/%s.%s.%s/%s",
|
|
r.GVK,
|
|
orDefault(r.Partition, "default"),
|
|
orDefault(r.PeerName, "local"),
|
|
orDefault(r.Namespace, "default"),
|
|
r.Name,
|
|
)
|
|
}
|
|
|
|
func (r ReferenceKey) GetTenancy() *pbresource.Tenancy {
|
|
return &pbresource.Tenancy{
|
|
Partition: r.Partition,
|
|
PeerName: r.PeerName,
|
|
Namespace: r.Namespace,
|
|
}
|
|
}
|
|
|
|
// ToReference converts this back into a pbresource.ID.
|
|
func (r ReferenceKey) ToID() *pbresource.ID {
|
|
return &pbresource.ID{
|
|
Type: GVKToType(r.GVK),
|
|
Tenancy: r.GetTenancy(),
|
|
Name: r.Name,
|
|
}
|
|
}
|
|
|
|
// ToReference converts this back into a pbresource.Reference.
|
|
func (r ReferenceKey) ToReference() *pbresource.Reference {
|
|
return &pbresource.Reference{
|
|
Type: GVKToType(r.GVK),
|
|
Tenancy: r.GetTenancy(),
|
|
Name: r.Name,
|
|
}
|
|
}
|
|
|
|
func (r ReferenceKey) GoString() string { return r.String() }
|
|
|
|
func NewReferenceKey(refOrID ReferenceOrID) ReferenceKey {
|
|
return ReferenceKey{
|
|
GVK: ToGVK(refOrID.GetType()),
|
|
Partition: orDefault(refOrID.GetTenancy().GetPartition(), "default"),
|
|
Namespace: orDefault(refOrID.GetTenancy().GetNamespace(), "default"),
|
|
PeerName: orDefault(refOrID.GetTenancy().GetPeerName(), "local"),
|
|
Name: refOrID.GetName(),
|
|
}
|
|
}
|
|
|
|
func orDefault(v, def string) string {
|
|
if v == "" {
|
|
return def
|
|
}
|
|
return v
|
|
}
|
|
|
|
func GVKToType(gvk string) *pbresource.Type {
|
|
parts := strings.Split(gvk, ".")
|
|
if len(parts) != 3 {
|
|
panic("bad gvk")
|
|
}
|
|
return &pbresource.Type{
|
|
Group: parts[0],
|
|
GroupVersion: parts[1],
|
|
Kind: parts[2],
|
|
}
|
|
}
|