4ae2c3a101
This PS makes sure we have linter and unit tests processed. The code has been reformatted to adhere to Go's code formatting conventions. Change-Id: I31f15d6d6c4b9bda7e3837941b6c9c3c3735aea7
107 lines
2.7 KiB
Go
107 lines
2.7 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/client-go/dynamic"
|
|
"k8s.io/client-go/kubernetes"
|
|
appsv1 "k8s.io/client-go/kubernetes/typed/apps/v1"
|
|
v1batch "k8s.io/client-go/kubernetes/typed/batch/v1"
|
|
v1core "k8s.io/client-go/kubernetes/typed/core/v1"
|
|
"k8s.io/client-go/rest"
|
|
)
|
|
|
|
type ClientInterface interface {
|
|
Pods(string) v1core.PodInterface
|
|
Jobs(string) v1batch.JobInterface
|
|
Endpoints(string) v1core.EndpointsInterface
|
|
DaemonSets(string) appsv1.DaemonSetInterface
|
|
Services(string) v1core.ServiceInterface
|
|
CustomResource(ctx context.Context, apiVersion, namespace, resource, name string) (*unstructured.Unstructured, error)
|
|
}
|
|
type Client struct {
|
|
client kubernetes.Interface
|
|
dynamicClient dynamic.Interface
|
|
}
|
|
|
|
func (c Client) Pods(namespace string) v1core.PodInterface {
|
|
return c.client.CoreV1().Pods(namespace)
|
|
}
|
|
|
|
func (c Client) Jobs(namespace string) v1batch.JobInterface {
|
|
return c.client.BatchV1().Jobs(namespace)
|
|
}
|
|
|
|
func (c Client) Endpoints(namespace string) v1core.EndpointsInterface {
|
|
return c.client.CoreV1().Endpoints(namespace)
|
|
}
|
|
|
|
func (c Client) DaemonSets(namespace string) appsv1.DaemonSetInterface {
|
|
return c.client.AppsV1().DaemonSets(namespace)
|
|
}
|
|
|
|
func (c Client) Services(namespace string) v1core.ServiceInterface {
|
|
return c.client.CoreV1().Services(namespace)
|
|
}
|
|
|
|
func (c Client) CustomResource(
|
|
ctx context.Context,
|
|
apiVersion, kind, namespace, name string,
|
|
) (*unstructured.Unstructured, error) {
|
|
apiResourceList, err := c.client.Discovery().ServerResourcesForGroupVersion(apiVersion)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
parts := strings.Split(apiVersion, "/")
|
|
if len(parts) != 2 {
|
|
return nil, fmt.Errorf(`apiVersion [%s] must be "group/version"`,
|
|
apiVersion)
|
|
}
|
|
group, version := parts[0], parts[1]
|
|
|
|
for _, apiResource := range apiResourceList.APIResources {
|
|
if apiResource.Kind == kind {
|
|
gvr := schema.GroupVersionResource{
|
|
Group: group,
|
|
Version: version,
|
|
Resource: apiResource.Name,
|
|
}
|
|
|
|
return c.dynamicClient.Resource(gvr).
|
|
Namespace(namespace).
|
|
Get(ctx, name, metav1.GetOptions{})
|
|
}
|
|
}
|
|
return nil, fmt.Errorf("could not find resource with with version %v, "+
|
|
"kind %v, and name %v in namespace %v",
|
|
apiVersion, kind, name, namespace)
|
|
}
|
|
|
|
func New(config *rest.Config) (ClientInterface, error) {
|
|
if config == nil {
|
|
var err error
|
|
config, err = rest.InClusterConfig()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
clientset, err := kubernetes.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dynamicClient, err := dynamic.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return Client{clientset, dynamicClient}, nil
|
|
}
|