From aac21b5ea4168f4d995b6d0b20e0f9dcf0a986bb Mon Sep 17 00:00:00 2001
From: Max Xu <xuhuan@live.cn>
Date: Mon, 14 Feb 2022 21:37:42 +0800
Subject: [PATCH] feat: create zookeeper headless service and statefulset

---
 controllers/zookeepercluster_controller.go | 100 ++++++++++++++++++++-
 1 file changed, 97 insertions(+), 3 deletions(-)

diff --git a/controllers/zookeepercluster_controller.go b/controllers/zookeepercluster_controller.go
index a8b499b..ca66240 100644
--- a/controllers/zookeepercluster_controller.go
+++ b/controllers/zookeepercluster_controller.go
@@ -18,11 +18,16 @@ package controllers
 
 import (
 	"context"
+	"fmt"
 
+	"github.com/go-logr/logr"
+	appsv1 "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
-	"sigs.k8s.io/controller-runtime/pkg/log"
+	ctrl_log "sigs.k8s.io/controller-runtime/pkg/log"
 
 	zookeeperv1alpha1 "github.com/jsonbruce/zookeeper-operator/api/v1alpha1"
 )
@@ -31,6 +36,8 @@ import (
 type ZookeeperClusterReconciler struct {
 	client.Client
 	Scheme *runtime.Scheme
+
+	Logger logr.Logger
 }
 
 //+kubebuilder:rbac:groups=zookeeper.atmax.io,resources=zookeeperclusters,verbs=get;list;watch;create;update;patch;delete
@@ -47,13 +54,100 @@ type ZookeeperClusterReconciler struct {
 // For more details, check Reconcile and its Result here:
 // - https://pkg.go.dev/sigs.k8s.io/controller-runtime@v0.11.0/pkg/reconcile
 func (r *ZookeeperClusterReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
-	_ = log.FromContext(ctx)
+	r.Logger = ctrl_log.FromContext(ctx)
+	r.Logger.Info("starting reconcile")
+
+	zk := &zookeeperv1alpha1.ZookeeperCluster{}
+
+	if err := r.Get(ctx, req.NamespacedName, zk); err != nil {
+		r.Logger.Error(err, "unable to fetch ZookeeperCluster")
+		return ctrl.Result{}, client.IgnoreNotFound(err)
+	}
+
+	svcHeadless := r.createHeadlessService(zk)
+	if err := r.Create(ctx, svcHeadless); err != nil {
+		r.Logger.Error(err, "creating headless service")
+		return ctrl.Result{}, err
+	}
 
-	// TODO(user): your logic here
+	sts := r.createStatefulSet(zk)
+	if err := r.Create(ctx, sts); err != nil {
+		r.Logger.Error(err, "creating statefulset")
+		return ctrl.Result{}, err
+	}
 
 	return ctrl.Result{}, nil
 }
 
+func (r *ZookeeperClusterReconciler) createHeadlessService(zk *zookeeperv1alpha1.ZookeeperCluster) *corev1.Service {
+	return &corev1.Service{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      fmt.Sprintf("%s-headless", zk.Name),
+			Namespace: zk.Namespace,
+			Labels: map[string]string{
+				"app": zk.Name,
+			},
+		},
+		Spec: corev1.ServiceSpec{
+			Ports: []corev1.ServicePort{
+				{
+					Name: "server",
+					Port: 2888,
+				},
+				{
+					Name: "leader-election",
+					Port: 3888,
+				},
+			},
+			Selector: map[string]string{
+				"app": zk.Name,
+			},
+			ClusterIP: corev1.ClusterIPNone,
+		},
+	}
+}
+
+func (r *ZookeeperClusterReconciler) createStatefulSet(zk *zookeeperv1alpha1.ZookeeperCluster) *appsv1.StatefulSet {
+	return &appsv1.StatefulSet{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      zk.Name,
+			Namespace: zk.Namespace,
+			Labels:    zk.Labels,
+		},
+		Spec: appsv1.StatefulSetSpec{
+			Replicas: &zk.Spec.Replicas,
+			Selector: &metav1.LabelSelector{
+				MatchLabels: map[string]string{
+					"app": zk.Name,
+				},
+			},
+			ServiceName: fmt.Sprintf("%s-headless", zk.Name),
+			Template: corev1.PodTemplateSpec{
+				ObjectMeta: metav1.ObjectMeta{
+					Labels: map[string]string{
+						"app": zk.Name,
+					},
+				},
+				Spec: corev1.PodSpec{
+					Containers: []corev1.Container{
+						{
+							Name:  "zookeeper",
+							Image: "zookeeper:latest",
+							Ports: nil,
+							Env: []corev1.EnvVar{
+								{
+									Name:  "ZOO_ENV",
+									Value: "10",
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
 // SetupWithManager sets up the controller with the Manager.
 func (r *ZookeeperClusterReconciler) SetupWithManager(mgr ctrl.Manager) error {
 	return ctrl.NewControllerManagedBy(mgr).