f8c51004d4
* Add providers subcommand to verify providers Allows users to check what providers are available * Fix version output to add new line This command was totally broken until we moved around the call to `initConfig()`, this just fixes the output now that it works. * Flip boolean of provider include tags All providers are still included by default and fix tags using the old format.
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package register
|
|
|
|
import (
|
|
"sort"
|
|
|
|
"github.com/cpuguy83/strongerrors"
|
|
"github.com/pkg/errors"
|
|
"github.com/virtual-kubelet/virtual-kubelet/manager"
|
|
"github.com/virtual-kubelet/virtual-kubelet/providers"
|
|
)
|
|
|
|
var providerInits = make(map[string]initFunc)
|
|
|
|
// InitConfig is the config passed to initialize a registered provider.
|
|
type InitConfig struct {
|
|
ConfigPath string
|
|
NodeName string
|
|
OperatingSystem string
|
|
InternalIP string
|
|
DaemonPort int32
|
|
ResourceManager *manager.ResourceManager
|
|
}
|
|
|
|
type initFunc func(InitConfig) (providers.Provider, error)
|
|
|
|
// GetProvider gets the provider specified by the given name
|
|
func GetProvider(name string, cfg InitConfig) (providers.Provider, error) {
|
|
f, ok := providerInits[name]
|
|
if !ok {
|
|
return nil, strongerrors.NotFound(errors.Errorf("provider not found: %s", name))
|
|
}
|
|
return f(cfg)
|
|
}
|
|
|
|
// Exists checks if a provider is regstered
|
|
func Exists(name string) bool {
|
|
_, ok := providerInits[name]
|
|
return ok
|
|
}
|
|
|
|
// List gets the list of all provider names
|
|
func List() []string {
|
|
ls := make([]string, 0, len(providerInits))
|
|
for name := range providerInits {
|
|
ls = append(ls, name)
|
|
}
|
|
sort.Strings(ls)
|
|
return ls
|
|
}
|
|
|
|
func register(name string, f initFunc) {
|
|
providerInits[name] = f
|
|
}
|