750de3195d
* Remove unused lock from the resource manager. * Add service lister to the resource manager. This change adds a service lister in the resource manager. This will be used to set the service env vars. Also added a List method to the resource manager and a simple test to confirm it's a pass through.
55 lines
1.9 KiB
Go
55 lines
1.9 KiB
Go
package manager
|
|
|
|
import (
|
|
"k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
corev1listers "k8s.io/client-go/listers/core/v1"
|
|
|
|
"github.com/virtual-kubelet/virtual-kubelet/log"
|
|
)
|
|
|
|
// ResourceManager acts as a passthrough to a cache (lister) for pods assigned to the current node.
|
|
// It is also a passthrough to a cache (lister) for Kubernetes secrets and config maps.
|
|
type ResourceManager struct {
|
|
podLister corev1listers.PodLister
|
|
secretLister corev1listers.SecretLister
|
|
configMapLister corev1listers.ConfigMapLister
|
|
serviceLister corev1listers.ServiceLister
|
|
}
|
|
|
|
// NewResourceManager returns a ResourceManager with the internal maps initialized.
|
|
func NewResourceManager(podLister corev1listers.PodLister, secretLister corev1listers.SecretLister, configMapLister corev1listers.ConfigMapLister, serviceLister corev1listers.ServiceLister) (*ResourceManager, error) {
|
|
rm := ResourceManager{
|
|
podLister: podLister,
|
|
secretLister: secretLister,
|
|
configMapLister: configMapLister,
|
|
serviceLister: serviceLister,
|
|
}
|
|
return &rm, nil
|
|
}
|
|
|
|
// GetPods returns a list of all known pods assigned to this virtual node.
|
|
func (rm *ResourceManager) GetPods() []*v1.Pod {
|
|
l, err := rm.podLister.List(labels.Everything())
|
|
if err == nil {
|
|
return l
|
|
}
|
|
log.L.Errorf("failed to fetch pods from lister: %v", err)
|
|
return make([]*v1.Pod, 0)
|
|
}
|
|
|
|
// GetConfigMap retrieves the specified config map from the cache.
|
|
func (rm *ResourceManager) GetConfigMap(name, namespace string) (*v1.ConfigMap, error) {
|
|
return rm.configMapLister.ConfigMaps(namespace).Get(name)
|
|
}
|
|
|
|
// GetSecret retrieves the specified secret from Kubernetes.
|
|
func (rm *ResourceManager) GetSecret(name, namespace string) (*v1.Secret, error) {
|
|
return rm.secretLister.Secrets(namespace).Get(name)
|
|
}
|
|
|
|
// ListServices retrieves the list of services from Kubernetes.
|
|
func (rm *ResourceManager) ListServices() ([]*v1.Service, error) {
|
|
return rm.serviceLister.List(labels.Everything())
|
|
}
|