service-handler.go

package main

import (
	"context"
	apiv1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/util/intstr"
	v1 "k8s.io/client-go/kubernetes/typed/core/v1"
)

func ServiceReady(service *apiv1.Service, serviceName string, labelName string, gracePeriodSeconds *int64) string {

	*service = apiv1.Service{
		ObjectMeta: metav1.ObjectMeta{
			Name: serviceName,
			Labels: map[string]string{
				labelName: labelName,
			},
			Annotations:                nil,
			DeletionGracePeriodSeconds: gracePeriodSeconds,
		},
		Spec: apiv1.ServiceSpec{
			Ports: []apiv1.ServicePort{
				{
					Name:        "ssh",
					Protocol:    apiv1.ProtocolTCP,
					AppProtocol: nil,
					Port:        22,
					TargetPort:  intstr.IntOrString{IntVal: 22},
				},
			},
			Selector: map[string]string{
				"app": labelName,
			},
		},
	}

	return serviceName
}

func Create_service(svcClient v1.ServiceInterface, serviceName string, labelName string, gracePeriodSeconds *int64) string {
	var service apiv1.Service

	realSvcName := ServiceReady(&service, serviceName, labelName, gracePeriodSeconds)
	_, err := svcClient.Create(context.TODO(), &service, metav1.CreateOptions{})
	if err != nil {
		Error.Println("create the service err : ", err)
	}
	Trace.Printf("created %s\n", serviceName)
	return realSvcName
}

func List_service(svcClient v1.ServiceInterface, labelName string) {
	Trace.Println("list service...")
	list, err := svcClient.List(context.TODO(), metav1.ListOptions{LabelSelector: labelName})
	if err != nil {
		Error.Println("list svc err: ", err)
	}
	for _, s := range list.Items {
		Trace.Printf(" * [%s] svc in [%s] with [%v] label\n", s.Name, s.Namespace, s.Labels)
	}
}

func Delete_service(svcClient v1.ServiceInterface, serviceName string, gracePeriodSeconds *int64) {
	deletePolicy := metav1.DeletePropagationForeground
	if err := svcClient.Delete(context.TODO(), serviceName, metav1.DeleteOptions{
		GracePeriodSeconds: gracePeriodSeconds,
		PropagationPolicy:  &deletePolicy,
	}); err != nil {
		Error.Println("delete svc err: ", err)
	}
}