083f6dee05
* Refactor provider init This moves provider init out of vkubelet setup, instead preferring to initialize vkubelet with a provider. * Split API server configuration from setup. This makes sure that configuration (which is done primarily through env vars) is separate from actually standing up the servers. This also makes sure to abort daemon initialization if the API servers are not able to start.
36 lines
918 B
Go
36 lines
918 B
Go
package register
|
|
|
|
import (
|
|
"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)
|
|
}
|
|
|
|
func register(name string, f initFunc) {
|
|
providerInits[name] = f
|
|
}
|