128 lines
3.5 KiB
Go
128 lines
3.5 KiB
Go
package kube
|
|
|
|
import (
|
|
"golang.org/x/sync/errgroup"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
)
|
|
|
|
// Objects encapsulates all the objects from a Kubernetes cluster.
|
|
type Objects struct {
|
|
Nodes *corev1.NodeList
|
|
PersistentVolumes *corev1.PersistentVolumeList
|
|
ComponentStatuses *corev1.ComponentStatusList
|
|
Pods *corev1.PodList
|
|
PodTemplates *corev1.PodTemplateList
|
|
PersistentVolumeClaims *corev1.PersistentVolumeClaimList
|
|
ConfigMaps *corev1.ConfigMapList
|
|
Services *corev1.ServiceList
|
|
Secrets *corev1.SecretList
|
|
ServiceAccounts *corev1.ServiceAccountList
|
|
ResourceQuotas *corev1.ResourceQuotaList
|
|
LimitRanges *corev1.LimitRangeList
|
|
}
|
|
|
|
// Client encapsulates a client for a Kubernetes cluster.
|
|
type Client struct {
|
|
kubeClient kubernetes.Interface
|
|
}
|
|
|
|
// FetchObjects returns the objects from a Kubernetes cluster.
|
|
func (c *Client) FetchObjects() (*Objects, error) {
|
|
client := c.kubeClient.CoreV1()
|
|
opts := metav1.ListOptions{}
|
|
objects := &Objects{}
|
|
|
|
var g errgroup.Group
|
|
|
|
g.Go(func() (err error) {
|
|
objects.Nodes, err = client.Nodes().List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.PersistentVolumes, err = client.PersistentVolumes().List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.ComponentStatuses, err = client.ComponentStatuses().List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.Pods, err = client.Pods(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.PodTemplates, err = client.PodTemplates(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.PersistentVolumeClaims, err = client.PersistentVolumeClaims(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.ConfigMaps, err = client.ConfigMaps(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.Secrets, err = client.Secrets(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.Services, err = client.Services(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.ServiceAccounts, err = client.ServiceAccounts(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.ResourceQuotas, err = client.ResourceQuotas(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
g.Go(func() (err error) {
|
|
objects.LimitRanges, err = client.LimitRanges(corev1.NamespaceAll).List(opts)
|
|
return
|
|
})
|
|
|
|
err := g.Wait()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return objects, nil
|
|
}
|
|
|
|
// NewClient builds a kubernetes client to interact with the live cluster.
|
|
// The kube config file path and the context must be specified for the client
|
|
// If not specified, defaults are assumed - configPath: ~/.kube/config, configContext: current context
|
|
func NewClient(configPath, configContext string) (*Client, error) {
|
|
var config *rest.Config
|
|
var err error
|
|
|
|
if configContext != "" {
|
|
config, err = clientcmd.NewNonInteractiveDeferredLoadingClientConfig(
|
|
&clientcmd.ClientConfigLoadingRules{ExplicitPath: configPath},
|
|
&clientcmd.ConfigOverrides{
|
|
CurrentContext: configContext,
|
|
}).ClientConfig()
|
|
} else {
|
|
config, err = clientcmd.BuildConfigFromFlags("", configPath)
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
client, err := kubernetes.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Client{
|
|
kubeClient: client,
|
|
}, nil
|
|
}
|