telegraf/plugins/inputs/kube_inventory/client.go

105 lines
3.2 KiB
Go

package kube_inventory
import (
"context"
"time"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
netv1 "k8s.io/api/networking/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/rest"
"github.com/influxdata/telegraf/plugins/common/tls"
)
type client struct {
namespace string
timeout time.Duration
*kubernetes.Clientset
}
func newClient(baseURL, namespace, bearerToken string, timeout time.Duration, tlsConfig tls.ClientConfig) (*client, error) {
c, err := kubernetes.NewForConfig(&rest.Config{
TLSClientConfig: rest.TLSClientConfig{
ServerName: baseURL,
Insecure: tlsConfig.InsecureSkipVerify,
CAFile: tlsConfig.TLSCA,
CertFile: tlsConfig.TLSCert,
KeyFile: tlsConfig.TLSKey,
},
BearerToken: bearerToken,
ContentConfig: rest.ContentConfig{},
})
if err != nil {
return nil, err
}
return &client{
Clientset: c,
timeout: timeout,
namespace: namespace,
}, nil
}
func (c *client) getDaemonSets(ctx context.Context) (*appsv1.DaemonSetList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.AppsV1().DaemonSets(c.namespace).List(ctx, metav1.ListOptions{})
}
func (c *client) getDeployments(ctx context.Context) (*appsv1.DeploymentList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.AppsV1().Deployments(c.namespace).List(ctx, metav1.ListOptions{})
}
func (c *client) getEndpoints(ctx context.Context) (*corev1.EndpointsList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.CoreV1().Endpoints(c.namespace).List(ctx, metav1.ListOptions{})
}
func (c *client) getIngress(ctx context.Context) (*netv1.IngressList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.NetworkingV1().Ingresses(c.namespace).List(ctx, metav1.ListOptions{})
}
func (c *client) getNodes(ctx context.Context) (*corev1.NodeList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.CoreV1().Nodes().List(ctx, metav1.ListOptions{})
}
func (c *client) getPersistentVolumes(ctx context.Context) (*corev1.PersistentVolumeList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.CoreV1().PersistentVolumes().List(ctx, metav1.ListOptions{})
}
func (c *client) getPersistentVolumeClaims(ctx context.Context) (*corev1.PersistentVolumeClaimList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.CoreV1().PersistentVolumeClaims(c.namespace).List(ctx, metav1.ListOptions{})
}
func (c *client) getPods(ctx context.Context) (*corev1.PodList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.CoreV1().Pods(c.namespace).List(ctx, metav1.ListOptions{})
}
func (c *client) getServices(ctx context.Context) (*corev1.ServiceList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.CoreV1().Services(c.namespace).List(ctx, metav1.ListOptions{})
}
func (c *client) getStatefulSets(ctx context.Context) (*appsv1.StatefulSetList, error) {
ctx, cancel := context.WithTimeout(ctx, c.timeout)
defer cancel()
return c.AppsV1().StatefulSets(c.namespace).List(ctx, metav1.ListOptions{})
}