feat(container): Added inCluster related API and xenon checker

cmd/staticcheck.conf

@@ -0,0 +1 @@
+checks=[ "all", "-ST1000", "-ST1003", "-ST1016", "-ST1020", "-ST1021", "-ST1022", "-ST1001"]

cmd/xenon/main.go

@@ -0,0 +1,88 @@
+package main
+
+import (
+	"log"
+	"os"
+
+	. "github.com/radondb/radondb-mysql-kubernetes/utils"
+)
+
+var (
+	ns      string
+	podName string
+)
+
+func init() {
+	ns = os.Getenv("NAMESPACE")
+	podName = os.Getenv("POD_NAME")
+}
+
+func main() {
+	switch os.Args[1] {
+	case "leaderStart":
+		if err := leaderStart(); err != nil {
+			log.Fatalf("leaderStart failed: %v", err)
+		}
+	case "leaderStop":
+		if err := leaderStop(); err != nil {
+			log.Fatalf("leaderStop failed: %v", err)
+		}
+	case "liveness":
+		if err := liveness(); err != nil {
+			log.Fatalf("liveness failed: %v", err)
+		}
+	case "readiness":
+		if err := readiness(); err != nil {
+			log.Fatalf("readiness failed: %v", err)
+		}
+	case "postStart":
+		if err := postStart(); err != nil {
+			log.Fatalf("postStart failed: %v", err)
+		}
+	case "preStop":
+		if err := preStop(); err != nil {
+			log.Fatalf("postStop failed: %v", err)
+		}
+	default:
+		log.Fatalf("Usage: %s leaderStart|leaderStop|liveness|readiness|postStart|preStop", os.Args[0])
+	}
+}
+
+// TODO
+func leaderStart() error {
+	return nil
+}
+
+func leaderStop() error {
+	return PatchRoleLabelTo(myself(string(Follower)))
+}
+
+func liveness() error {
+	return XenonPingMyself()
+}
+
+func readiness() error {
+	role := GetRole()
+	if role != string(Leader) {
+		return PatchRoleLabelTo(myself(role))
+	}
+	return nil
+}
+
+// TODO
+func postStart() error {
+	return nil
+}
+
+// TODO
+func preStop() error {
+	return nil
+}
+
+func myself(role string) MySQLNode {
+	return MySQLNode{
+		PodName:   podName,
+		Namespace: ns,
+		Role:      role,
+	}
+}

mysqlcluster/container/xenon.go

@@ -48,7 +48,26 @@ func (c *xenon) getCommand() []string {
 
 // getEnvVars get the container env.
 func (c *xenon) getEnvVars() []corev1.EnvVar {
-	return nil
+	return []corev1.EnvVar{
+		{
+			Name: "NameSpace",
+			ValueFrom: &corev1.EnvVarSource{
+				FieldRef: &corev1.ObjectFieldSelector{
+					APIVersion: "v1",
+					FieldPath:  "metadata.namespace",
+				},
+			},
+		},
+		{
+			Name: "POD_NAME",
+			ValueFrom: &corev1.EnvVarSource{
+				FieldRef: &corev1.ObjectFieldSelector{
+					APIVersion: "v1",
+					FieldPath:  "metadata.name",
+				},
+			},
+		},
+	}
 }
 
 // getLifecycle get the container lifecycle.

mysqlcluster/container/xenon_test.go

@@ -68,7 +68,26 @@ func TestGetXenonCommand(t *testing.T) {
 }
 
 func TestGetXenonEnvVar(t *testing.T) {
-	assert.Nil(t, xenonCase.Env)
+	assert.Equal(t, []corev1.EnvVar{
+		{
+			Name: "NameSpace",
+			ValueFrom: &corev1.EnvVarSource{
+				FieldRef: &corev1.ObjectFieldSelector{
+					APIVersion: "v1",
+					FieldPath:  "metadata.namespace",
+				},
+			},
+		},
+		{
+			Name: "POD_NAME",
+			ValueFrom: &corev1.EnvVarSource{
+				FieldRef: &corev1.ObjectFieldSelector{
+					APIVersion: "v1",
+					FieldPath:  "metadata.name",
+				},
+			},
+		},
+	}, xenonCase.Env)
 }
 
 func TestGetXenonLifecycle(t *testing.T) {

utils/incluster.go

@@ -0,0 +1,81 @@
+package utils
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"log"
+	"os/exec"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/rest"
+)
+
+type raftStatus struct {
+	Leader string   `json:"leader"`
+	State  string   `json:"state"`
+	Nodes  []string `json:"nodes"`
+}
+
+type MySQLNode struct {
+	PodName   string
+	Namespace string
+	Role      string
+}
+
+func GetClientSet() (*kubernetes.Clientset, error) {
+	// Creates the in-cluster config
+	config, err := rest.InClusterConfig()
+	if err != nil {
+		return nil, fmt.Errorf("failed to create in-cluster config: %v", err)
+	}
+	// Creates the clientset
+	clientset, err := kubernetes.NewForConfig(config)
+	if err != nil {
+		return nil, fmt.Errorf("failed to create clientset: %v", err)
+	}
+	return clientset, nil
+}
+
+func PatchRoleLabelTo(x MySQLNode) error {
+	// Creates the clientset
+	clientset, err := GetClientSet()
+	if err != nil {
+		return fmt.Errorf("failed to create clientset: %v", err)
+	}
+	patch := fmt.Sprintf(`{"metadata":{"labels":{"role":"%s"}}}`, x.Role)
+	_, err = clientset.CoreV1().Pods(x.Namespace).Patch(context.TODO(), x.PodName, types.MergePatchType, []byte(patch), metav1.PatchOptions{})
+	if err != nil {
+		return fmt.Errorf("failed to patch pod role label: %v", err)
+	}
+	return nil
+}
+
+func XenonPingMyself() error {
+	args := []string{"xenon", "ping"}
+	cmd := exec.Command("xenoncli", args...)
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("failed to exec xenoncli xenon ping: %v", err)
+	}
+	return nil
+}
+
+func GetRaftStatus() *raftStatus {
+	args := []string{"raft", "status"}
+	cmd := exec.Command("xenoncli", args...)
+	res, err := cmd.Output()
+	if err != nil {
+		log.Fatalf("failed to exec xenoncli raft status: %v", err)
+	}
+	raftStatus := raftStatus{}
+	if err := json.Unmarshal(res, &raftStatus); err != nil {
+		log.Fatalf("failed to unmarshal raft status: %v", err)
+	}
+	return &raftStatus
+}
+
+func GetRole() string {
+	return GetRaftStatus().State
+}