120 lines
3.5 KiB
Go
120 lines
3.5 KiB
Go
package kube_inventory
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/ericchiang/k8s"
|
|
"github.com/ericchiang/k8s/apis/apps/v1beta1"
|
|
"github.com/ericchiang/k8s/apis/apps/v1beta2"
|
|
"github.com/ericchiang/k8s/apis/core/v1"
|
|
v1beta1EXT "github.com/ericchiang/k8s/apis/extensions/v1beta1"
|
|
|
|
"github.com/influxdata/telegraf/internal/tls"
|
|
)
|
|
|
|
type client struct {
|
|
namespace string
|
|
timeout time.Duration
|
|
*k8s.Client
|
|
}
|
|
|
|
func newClient(baseURL, namespace, bearerToken string, timeout time.Duration, tlsConfig tls.ClientConfig) (*client, error) {
|
|
c, err := k8s.NewClient(&k8s.Config{
|
|
Clusters: []k8s.NamedCluster{{Name: "cluster", Cluster: k8s.Cluster{
|
|
Server: baseURL,
|
|
InsecureSkipTLSVerify: tlsConfig.InsecureSkipVerify,
|
|
CertificateAuthority: tlsConfig.TLSCA,
|
|
}}},
|
|
Contexts: []k8s.NamedContext{{Name: "context", Context: k8s.Context{
|
|
Cluster: "cluster",
|
|
AuthInfo: "auth",
|
|
Namespace: namespace,
|
|
}}},
|
|
AuthInfos: []k8s.NamedAuthInfo{{Name: "auth", AuthInfo: k8s.AuthInfo{
|
|
Token: bearerToken,
|
|
ClientCertificate: tlsConfig.TLSCert,
|
|
ClientKey: tlsConfig.TLSKey,
|
|
}}},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &client{
|
|
Client: c,
|
|
timeout: timeout,
|
|
namespace: namespace,
|
|
}, nil
|
|
}
|
|
|
|
func (c *client) getDaemonSets(ctx context.Context) (*v1beta2.DaemonSetList, error) {
|
|
list := new(v1beta2.DaemonSetList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, c.namespace, list)
|
|
}
|
|
|
|
func (c *client) getDeployments(ctx context.Context) (*v1beta1.DeploymentList, error) {
|
|
list := &v1beta1.DeploymentList{}
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, c.namespace, list)
|
|
}
|
|
|
|
func (c *client) getEndpoints(ctx context.Context) (*v1.EndpointsList, error) {
|
|
list := new(v1.EndpointsList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, c.namespace, list)
|
|
}
|
|
|
|
func (c *client) getIngress(ctx context.Context) (*v1beta1EXT.IngressList, error) {
|
|
list := new(v1beta1EXT.IngressList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, c.namespace, list)
|
|
}
|
|
|
|
func (c *client) getNodes(ctx context.Context) (*v1.NodeList, error) {
|
|
list := new(v1.NodeList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, "", list)
|
|
}
|
|
|
|
func (c *client) getPersistentVolumes(ctx context.Context) (*v1.PersistentVolumeList, error) {
|
|
list := new(v1.PersistentVolumeList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, "", list)
|
|
}
|
|
|
|
func (c *client) getPersistentVolumeClaims(ctx context.Context) (*v1.PersistentVolumeClaimList, error) {
|
|
list := new(v1.PersistentVolumeClaimList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, c.namespace, list)
|
|
}
|
|
|
|
func (c *client) getPods(ctx context.Context) (*v1.PodList, error) {
|
|
list := new(v1.PodList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, c.namespace, list)
|
|
}
|
|
|
|
func (c *client) getServices(ctx context.Context) (*v1.ServiceList, error) {
|
|
list := new(v1.ServiceList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, c.namespace, list)
|
|
}
|
|
|
|
func (c *client) getStatefulSets(ctx context.Context) (*v1beta1.StatefulSetList, error) {
|
|
list := new(v1beta1.StatefulSetList)
|
|
ctx, cancel := context.WithTimeout(ctx, c.timeout)
|
|
defer cancel()
|
|
return list, c.List(ctx, c.namespace, list)
|
|
}
|