k3k/pkg/controller/cluster/server/service.go

package server

import (
	"github.com/rancher/k3k/pkg/apis/k3k.io/v1alpha1"
	"github.com/rancher/k3k/pkg/controller"
	v1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/util/intstr"
)

func Service(cluster *v1alpha1.Cluster) *v1.Service {
	service := &v1.Service{
		TypeMeta: metav1.TypeMeta{
			Kind:       "Service",
			APIVersion: "v1",
		},
		ObjectMeta: metav1.ObjectMeta{
			Name:      ServiceName(cluster.Name),
			Namespace: cluster.Namespace,
		},
		Spec: v1.ServiceSpec{
			Type: v1.ServiceTypeClusterIP,
			Selector: map[string]string{
				"cluster": cluster.Name,
				"role":    "server",
			},
		},
	}

	k3sServerPort := v1.ServicePort{
		Name:     "k3s-server-port",
		Protocol: v1.ProtocolTCP,
		Port:     serverPort,
	}

	k3sServicePort := v1.ServicePort{
		Name:       "k3s-service-port",
		Protocol:   v1.ProtocolTCP,
		Port:       servicePort,
		TargetPort: intstr.FromInt(serverPort),
	}

	etcdPort := v1.ServicePort{
		Name:     "k3s-etcd-port",
		Protocol: v1.ProtocolTCP,
		Port:     etcdPort,
	}

	if cluster.Spec.Expose != nil {
		nodePortConfig := cluster.Spec.Expose.NodePort
		if nodePortConfig != nil {
			service.Spec.Type = v1.ServiceTypeNodePort

			if nodePortConfig.ServerPort != nil {
				k3sServerPort.NodePort = *nodePortConfig.ServerPort
			}
			if nodePortConfig.ServicePort != nil {
				k3sServicePort.NodePort = *nodePortConfig.ServicePort
			}
			if nodePortConfig.ETCDPort != nil {
				etcdPort.NodePort = *nodePortConfig.ETCDPort
			}
		}
	}

	service.Spec.Ports = append(
		service.Spec.Ports,
		k3sServicePort,
		etcdPort,
		k3sServerPort,
	)

	return service
}

func (s *Server) StatefulServerService() *v1.Service {
	return &v1.Service{
		TypeMeta: metav1.TypeMeta{
			Kind:       "Service",
			APIVersion: "v1",
		},
		ObjectMeta: metav1.ObjectMeta{
			Name:      headlessServiceName(s.cluster.Name),
			Namespace: s.cluster.Namespace,
		},
		Spec: v1.ServiceSpec{
			Type:      v1.ServiceTypeClusterIP,
			ClusterIP: v1.ClusterIPNone,
			Selector: map[string]string{
				"cluster": s.cluster.Name,
				"role":    "server",
			},
			Ports: []v1.ServicePort{
				{
					Name:     "k3s-server-port",
					Protocol: v1.ProtocolTCP,
					Port:     serverPort,
				},
				{
					Name:       "k3s-service-port",
					Protocol:   v1.ProtocolTCP,
					Port:       servicePort,
					TargetPort: intstr.FromInt(serverPort),
				},
				{
					Name:     "k3s-etcd-port",
					Protocol: v1.ProtocolTCP,
					Port:     etcdPort,
				},
			},
		},
	}
}

func ServiceName(clusterName string) string {
	return controller.SafeConcatNameWithPrefix(clusterName, "service")
}

func headlessServiceName(clusterName string) string {
	return controller.SafeConcatNameWithPrefix(clusterName, "service", "headless")
}