110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
package state
|
|
|
|
import (
|
|
"fmt"
|
|
"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{
|
|
readIndex: indexFromConfigEntryKindName,
|
|
writeIndex: indexFromConfigEntry,
|
|
prefixIndex: indexFromConfigEntryKindName,
|
|
},
|
|
},
|
|
indexKind: {
|
|
Name: indexKind,
|
|
AllowMissing: false,
|
|
Unique: false,
|
|
Indexer: indexerSingle{
|
|
readIndex: indexFromConfigEntryKindQuery,
|
|
writeIndex: indexKindFromConfigEntry,
|
|
},
|
|
},
|
|
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{},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func indexFromConfigEntry(raw interface{}) ([]byte, error) {
|
|
c, ok := raw.(structs.ConfigEntry)
|
|
if !ok {
|
|
return nil, fmt.Errorf("type must be structs.ConfigEntry: %T", raw)
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
func indexFromConfigEntryKindQuery(raw interface{}) ([]byte, error) {
|
|
q, ok := raw.(ConfigEntryKindQuery)
|
|
if !ok {
|
|
return nil, fmt.Errorf("type must be ConfigEntryKindQuery: %T", raw)
|
|
}
|
|
|
|
var b indexBuilder
|
|
b.String(strings.ToLower(q.Kind))
|
|
return b.Bytes(), nil
|
|
}
|
|
|
|
// indexKindFromConfigEntry indexes kinds without a namespace for any config
|
|
// entries that span all namespaces.
|
|
func indexKindFromConfigEntry(raw interface{}) ([]byte, error) {
|
|
c, ok := raw.(structs.ConfigEntry)
|
|
if !ok {
|
|
return nil, fmt.Errorf("type must be structs.ConfigEntry: %T", raw)
|
|
}
|
|
|
|
if c.GetKind() == "" {
|
|
return nil, errMissingValueForIndex
|
|
}
|
|
|
|
var b indexBuilder
|
|
b.String(strings.ToLower(c.GetKind()))
|
|
return b.Bytes(), nil
|
|
}
|