cleaning up some code

kube_client/client.go

@@ -0,0 +1,82 @@
+package kube_client
+
+import (
+	"context"
+	v1a "k8s.io/api/apps/v1"
+	v1c "k8s.io/api/core/v1"
+	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/rest"
+	"time"
+)
+
+type kubeClient interface {
+	GetSecret(namespace string, secretName string) (*v1c.Secret, error)
+	CreateSecret(namespace string, secret *v1c.Secret) (*v1c.Secret, error)
+	UpdateSecret(namespace string, secret *v1c.Secret) (*v1c.Secret, error)
+	ListDeployments(namespace string) (*v1a.DeploymentList, error)
+	RestartDeployments(namespace string, deployments *v1a.DeploymentList) error
+}
+
+type KubeClientImpl struct {
+	kubeClient
+	clientSet *kubernetes.Clientset
+}
+
+func NewKubeClient() (*KubeClientImpl, error) {
+
+	config, err := rest.InClusterConfig()
+	if err != nil {
+		return nil, err
+	}
+
+	client, err := kubernetes.NewForConfig(config)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return &KubeClientImpl{
+		clientSet: client,
+	}, nil
+}
+
+func (k KubeClientImpl) GetSecret(namespace string, secretName string) (*v1c.Secret, error) {
+	secret, err := k.clientSet.CoreV1().Secrets(namespace).Get(context.TODO(), secretName, v1.GetOptions{})
+
+	if err != nil {
+		return nil, err
+	}
+
+	return secret, nil
+}
+
+func (k KubeClientImpl) CreateSecret(namespace string, secret *v1c.Secret) (*v1c.Secret, error) {
+	return k.clientSet.CoreV1().Secrets(namespace).Create(context.TODO(), secret, v1.CreateOptions{})
+}
+
+func (k KubeClientImpl) UpdateSecret(namespace string, secret *v1c.Secret) (*v1c.Secret, error) {
+	return k.clientSet.CoreV1().Secrets(namespace).Update(context.TODO(), secret, v1.UpdateOptions{})
+}
+
+func (k KubeClientImpl) ListDeployments(namespace string) (*v1a.DeploymentList, error) {
+	return k.clientSet.AppsV1().Deployments(namespace).List(context.TODO(), v1.ListOptions{
+		LabelSelector: "iam-role-type=aws-iam-anywhere",
+	})
+}
+
+func (k KubeClientImpl) RestartDeployments(namespace string, deployments *v1a.DeploymentList) error {
+	for _, deployment := range deployments.Items {
+		if deployment.Spec.Template.ObjectMeta.Annotations == nil {
+			deployment.Spec.Template.ObjectMeta.Annotations = make(map[string]string)
+		}
+
+		deployment.Spec.Template.ObjectMeta.Annotations["kubectl.kubernetes.io/restartedAt"] = time.Now().Format(time.RFC3339)
+		_, err := k.clientSet.AppsV1().Deployments(namespace).Update(context.TODO(), &deployment, v1.UpdateOptions{})
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}