1
0
Fork 0
mirror of https://github.com/kubernetes-sigs/node-feature-discovery.git synced 2024-12-14 11:57:51 +00:00
node-feature-discovery/source/source.go
Markus Lehtonen b7658c25fd generate: update mockery to latest version
In order to be able to run it on Go v1.18.
2022-08-10 09:47:23 +03:00

156 lines
3.9 KiB
Go

/*
Copyright 2017-2021 The Kubernetes Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package source
//go:generate mockery --name=LabelSource --inpackage
import (
"fmt"
"sigs.k8s.io/node-feature-discovery/pkg/api/feature"
)
// Source is the base interface for all other source interfaces
type Source interface {
// Name returns a friendly name for this source
Name() string
}
// FeatureSource is an interface for discovering node features
type FeatureSource interface {
Source
// Discover does feature discovery
Discover() error
// GetFeatures returns discovered features in raw form
GetFeatures() *feature.DomainFeatures
}
// LabelSource represents a source of node feature labels
type LabelSource interface {
Source
// GetLabels returns discovered feature labels
GetLabels() (FeatureLabels, error)
// Priority returns the priority of the source
Priority() int
}
// ConfigurableSource is an interface for a source that can be configured
type ConfigurableSource interface {
Source
// NewConfig returns a new default config of the source
NewConfig() Config
// GetConfig returns the effective configuration of the source
GetConfig() Config
// SetConfig changes the effective configuration of the source
SetConfig(Config)
}
// SupplementalSource represents a source that does not belong to the core set
// sources to be used in production, e.g. is deprecated, very experimental or
// purposed for testing only.
type SupplementalSource interface {
Source
// DisableByDefault returns true if the source should be disabled by
// default in production.
DisableByDefault() bool
}
// FeatureLabelValue represents the value of one feature label
type FeatureLabelValue interface{}
// FeatureLabels is a collection of feature labels
type FeatureLabels map[string]FeatureLabelValue
// Config is the generic interface for source configuration data
type Config interface {
}
// sources contain all registered sources
var sources = make(map[string]Source)
// Register registers a source.
func Register(s Source) {
if name, ok := sources[s.Name()]; ok {
panic(fmt.Sprintf("source %q already registered", name))
}
sources[s.Name()] = s
}
// GetFeatureSource returns a registered FeatureSource interface
func GetFeatureSource(name string) FeatureSource {
if s, ok := sources[name].(FeatureSource); ok {
return s
}
return nil
}
// GetAllFeatureSources returns all registered label sources
func GetAllFeatureSources() map[string]FeatureSource {
all := make(map[string]FeatureSource)
for k, v := range sources {
if s, ok := v.(FeatureSource); ok {
all[k] = s
}
}
return all
}
// GetLabelSource a registered label source
func GetLabelSource(name string) LabelSource {
if s, ok := sources[name].(LabelSource); ok {
return s
}
return nil
}
// GetAllLabelSources returns all registered label sources
func GetAllLabelSources() map[string]LabelSource {
all := make(map[string]LabelSource)
for k, v := range sources {
if s, ok := v.(LabelSource); ok {
all[k] = s
}
}
return all
}
// GetConfigurableSource a registered configurable source
func GetConfigurableSource(name string) ConfigurableSource {
if s, ok := sources[name].(ConfigurableSource); ok {
return s
}
return nil
}
// GetAllConfigurableSources returns all registered configurable sources
func GetAllConfigurableSources() map[string]ConfigurableSource {
all := make(map[string]ConfigurableSource)
for k, v := range sources {
if s, ok := v.(ConfigurableSource); ok {
all[k] = s
}
}
return all
}