aaf3c051f2
We have many indexer functions in Consul which take interface{} and type assert before building the index. We can use generics to get rid of the initial plumbing and pass around functions with better defined signatures. This has two benefits: 1) Less verbosity; 2) Developers can parse the argument types to memdb schemas without having to introspect the function for the type assertion.
102 lines
3 KiB
Go
102 lines
3 KiB
Go
package state
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/hashicorp/go-memdb"
|
|
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
)
|
|
|
|
const (
|
|
tableConfigEntries = "config-entries"
|
|
|
|
indexLink = "link"
|
|
indexIntentionLegacyID = "intention-legacy-id"
|
|
indexSource = "intention-source"
|
|
)
|
|
|
|
// configTableSchema returns a new table schema used to store global
|
|
// config entries.
|
|
func configTableSchema() *memdb.TableSchema {
|
|
return &memdb.TableSchema{
|
|
Name: tableConfigEntries,
|
|
Indexes: map[string]*memdb.IndexSchema{
|
|
indexID: {
|
|
Name: indexID,
|
|
AllowMissing: false,
|
|
Unique: true,
|
|
Indexer: indexerSingleWithPrefix[any, structs.ConfigEntry, any]{
|
|
readIndex: indexFromConfigEntryKindName,
|
|
writeIndex: indexFromConfigEntry,
|
|
prefixIndex: indexFromConfigEntryKindName,
|
|
},
|
|
},
|
|
indexLink: {
|
|
Name: indexLink,
|
|
AllowMissing: true,
|
|
Unique: false,
|
|
Indexer: &ConfigEntryLinkIndex{},
|
|
},
|
|
indexIntentionLegacyID: {
|
|
Name: indexIntentionLegacyID,
|
|
AllowMissing: true,
|
|
Unique: true,
|
|
Indexer: &ServiceIntentionLegacyIDIndex{},
|
|
},
|
|
indexSource: {
|
|
Name: indexSource,
|
|
AllowMissing: true,
|
|
Unique: false,
|
|
Indexer: &ServiceIntentionSourceIndex{},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// configEntryIndexable is required because while structs.ConfigEntry
|
|
// has a GetEnterpriseMeta method, it does not directly expose the
|
|
// required NamespaceOrDefault and PartitionOrDefault methods of
|
|
// enterpriseIndexable.
|
|
//
|
|
// Config entries that embed *acl.EnterpriseMeta will automatically
|
|
// implement this interface.
|
|
type configEntryIndexable interface {
|
|
structs.ConfigEntry
|
|
enterpriseIndexable
|
|
}
|
|
|
|
var _ configEntryIndexable = (*structs.ExportedServicesConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.IngressGatewayConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.MeshConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.ProxyConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.ServiceConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.ServiceIntentionsConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.ServiceResolverConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.ServiceRouterConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.ServiceSplitterConfigEntry)(nil)
|
|
var _ configEntryIndexable = (*structs.TerminatingGatewayConfigEntry)(nil)
|
|
|
|
func indexFromConfigEntry(c structs.ConfigEntry) ([]byte, error) {
|
|
if c.GetName() == "" || c.GetKind() == "" {
|
|
return nil, errMissingValueForIndex
|
|
}
|
|
|
|
var b indexBuilder
|
|
b.String(strings.ToLower(c.GetKind()))
|
|
b.String(strings.ToLower(c.GetName()))
|
|
return b.Bytes(), nil
|
|
}
|
|
|
|
// indexKindFromConfigEntry indexes kinds without a namespace for any config
|
|
// entries that span all namespaces.
|
|
func indexKindFromConfigEntry(c configEntryIndexable) ([]byte, error) {
|
|
if c.GetKind() == "" {
|
|
return nil, errMissingValueForIndex
|
|
}
|
|
|
|
var b indexBuilder
|
|
b.String(strings.ToLower(c.GetKind()))
|
|
return b.Bytes(), nil
|
|
}
|