componentdefinition.yaml

apiVersion: apps.kubeblocks.io/v1alpha1
kind: ComponentDefinition
metadata:
  name: etcd-{{ .Chart.Version }}
  labels:
    {{- include "etcd.labels" . | nindent 4}}
spec:
  serviceKind: etcd
  serviceVersion: {{ .Chart.AppVersion }}
  runtime:
    containers:
      - name: etcd
        image: {{ .Values.image.registry | default "docker.io" }}/{{ .Values.image.repository }}:{{ default .Chart.AppVersion .Values.image.tag }}
        imagePullPolicy: IfNotPresent
        ports:
          - containerPort: 2379
            name: client
          - containerPort: 2380
            name: peer
        volumeMounts:
          - name: data
            mountPath: /var/run/etcd
        command:
          - /bin/sh
          - -c
          - |
            echo "start etcd..."

            PEERS=""
            MY_PEER=""

            echo ${PEER_ENDPOINT}
            if [ -z "${PEER_ENDPOINT}" ]; then
                DOMAIN=$KB_NAMESPACE".svc{{ .Values.clusterDomain }}"
                SUBDOMAIN=${KB_CLUSTER_COMP_NAME}-headless
                replicas=$(eval echo ${KB_POD_LIST} | tr ',' '\n')
                for replica in ${replicas}; do
                    host=${replica}.${SUBDOMAIN}.${DOMAIN}
                    PEERS="${PEERS}${replica}=http://${host}:2380,"
                done
                PEERS=${PEERS%,}
                MY_PEER=$KB_POD_FQDN{{ .Values.clusterDomain }}
            else
                my_id=$(eval echo ${KB_POD_NAME} | grep -oE "[0-9]+\$")
                endpoints=$(eval echo ${PEER_ENDPOINT} | tr ',' '\n')
                for endpoint in ${endpoints}; do
                    host=$(eval echo ${endpoint} | cut -d ':' -f 1)
                    ip=$(eval echo ${endpoint} | cut -d ':' -f 2)
                    host_id=$(eval echo ${host} | grep -oE "[0-9]+\$")
                    hostname=${KB_CLUSTER_COMP_NAME}-${host_id}
                    PEERS="${PEERS}${hostname}=http://${ip}:2380,"
                    if [ "${my_id}" = "${host_id}" ]; then
                        MY_PEER=${ip}
                    fi
                done
                PEERS=${PEERS%,}
            fi
            echo "peers: ${PEERS}, my-peer: ${MY_PEER}"

            exec etcd --name ${HOSTNAME} \
            --experimental-initial-corrupt-check=true \
            --listen-peer-urls http://0.0.0.0:2380 \
            --listen-client-urls http://0.0.0.0:2379 \
            --advertise-client-urls http://${MY_PEER}:2379 \
            --initial-advertise-peer-urls http://${MY_PEER}:2380 \
            --initial-cluster ${PEERS} \
            --data-dir /var/run/etcd/default.etcd
  services:
    - name: peer
      serviceName: peer
      spec:
        ports:
          - name: peer
            port: 2380
            targetPort: peer
      podService: true
      disableAutoProvision: true
  vars:
    - name: PEER_ENDPOINT
      valueFrom:
        serviceVarRef:
          name: peer
          optional: true
          host: Required
          loadBalancer: Required
  updateStrategy: BestEffortParallel
  roles:
    - name: leader
      serviceable: true
      writable: true
      votable: true
    - name: follower
      serviceable: true
      writable: false
      votable: true
  lifecycleActions:
    roleProbe:
      customHandler:
        image: {{ .Values.image.registry | default "docker.io" }}/{{ .Values.image.repository }}:{{ default .Chart.AppVersion .Values.image.tag }}
        exec:
          command:
            - |
              Status=$(etcdctl --endpoints=127.0.0.1:2379 endpoint status -w simple --command-timeout=300ms --dial-timeout=100m) &&
              IsLeader=$(echo $Status | awk -F ', ' '{print $5}') &&
              IsLearner=$(echo $Status | awk -F ', ' '{print $6}') &&
              if [ "true" = "$IsLeader" ]; then echo -n "leader"; elif [ "true" = "$IsLearner" ]; then echo -n "learner"; else echo -n "follower"; fi