52 lines
1.7 KiB
Go
52 lines
1.7 KiB
Go
package manager
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"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 {
|
|
sync.RWMutex
|
|
|
|
podLister corev1listers.PodLister
|
|
secretLister corev1listers.SecretLister
|
|
configMapLister corev1listers.ConfigMapLister
|
|
}
|
|
|
|
// NewResourceManager returns a ResourceManager with the internal maps initialized.
|
|
func NewResourceManager(podLister corev1listers.PodLister, secretLister corev1listers.SecretLister, configMapLister corev1listers.ConfigMapLister) (*ResourceManager, error) {
|
|
rm := ResourceManager{
|
|
podLister: podLister,
|
|
secretLister: secretLister,
|
|
configMapLister: configMapLister,
|
|
}
|
|
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.Pods(v1.NamespaceAll).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)
|
|
}
|