src/foundation/scripts/setup-cni.sh

   1 #!/bin/bash
   2 set -o xtrace
   3 set -e
   4
   5 if [ -f "$HOME/.bashrc" ]; then
   6   # shellcheck source=/dev/null
   7   source "$HOME/.bashrc"
   8 fi
   9
  10
  11 CLUSTER_IP=${1:-172.16.1.136} # Align with the value in our K8s setup script
  12 POD_NETWORK_CIDR=${2:-192.168.0.0/16}
  13 CNI_TYPE=${3:-calico}
  14
  15 SCRIPTS_DIR=$(dirname "${BASH_SOURCE[0]}")
  16
  17 install_calico(){
  18   # Install the Etcd Database
  19   if [ "$(uname -m)" == 'aarch64' ]; then
  20     ETCD_YAML=etcd-arm64.yaml
  21   else
  22     ETCD_YAML=etcd-amd64.yaml
  23   fi
  24
  25   sed -i "s/10.96.232.136/${CLUSTER_IP}/" "${SCRIPTS_DIR}/cni/calico/${ETCD_YAML}"
  26   kubectl apply -f "${SCRIPTS_DIR}/cni/calico/${ETCD_YAML}"
  27
  28   # Install the RBAC Roles required for Calico
  29   kubectl apply -f "${SCRIPTS_DIR}/cni/calico/rbac.yaml"
  30
  31   # Install Calico to system
  32   sed -i "s@10.96.232.136@${CLUSTER_IP}@; s@192.168.0.0/16@${POD_NETWORK_CIDR}@" \
  33     "${SCRIPTS_DIR}/cni/calico/calico.yaml"
  34   kubectl apply -f "${SCRIPTS_DIR}/cni/calico/calico.yaml"
  35 }
  36
  37 install_flannel(){
  38   # Install the flannel CNI
  39   sed -i "s@10.244.0.0/16@${POD_NETWORK_CIDR}@" "${SCRIPTS_DIR}/cni/flannel/kube-flannel.yml"
  40   kubectl apply -f "${SCRIPTS_DIR}/cni/flannel/kube-flannel.yml"
  41 }
  42
  43
  44 case ${CNI_TYPE} in
  45  'calico')
  46         echo "Install calico ..."
  47         install_calico
  48         ;;
  49  'flannel')
  50         echo "Install flannel ..."
  51         install_flannel
  52         ;;
  53  *)
  54         echo "${CNI_TYPE} does not supportted"
  55         exit 1
  56         ;;
  57 esac
  58
  59 # Remove the taints on master node
  60 kubectl taint nodes --all node-role.kubernetes.io/master- || true