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   ETCD_YAML=etcd.yaml
  20
  21   sed -i "s/10.96.232.136/${CLUSTER_IP}/" "${SCRIPTS_DIR}/cni/calico/${ETCD_YAML}"
  22   kubectl apply -f "${SCRIPTS_DIR}/cni/calico/${ETCD_YAML}"
  23
  24   # Install the RBAC Roles required for Calico
  25   kubectl apply -f "${SCRIPTS_DIR}/cni/calico/rbac.yaml"
  26
  27   # Install Calico to system
  28   sed -i "s@10.96.232.136@${CLUSTER_IP}@; s@192.168.0.0/16@${POD_NETWORK_CIDR}@" \
  29     "${SCRIPTS_DIR}/cni/calico/calico.yaml"
  30   kubectl apply -f "${SCRIPTS_DIR}/cni/calico/calico.yaml"
  31 }
  32
  33 install_flannel(){
  34   # Install the flannel CNI
  35   sed -i "s@10.244.0.0/16@${POD_NETWORK_CIDR}@" "${SCRIPTS_DIR}/cni/flannel/kube-flannel.yml"
  36   kubectl apply -f "${SCRIPTS_DIR}/cni/flannel/kube-flannel.yml"
  37 }
  38
  39
  40 case ${CNI_TYPE} in
  41  'calico')
  42         echo "Install calico ..."
  43         install_calico
  44         ;;
  45  'flannel')
  46         echo "Install flannel ..."
  47         install_flannel
  48         ;;
  49  *)
  50         echo "${CNI_TYPE} does not supportted"
  51         exit 1
  52         ;;
  53 esac
  54
  55 # Remove the taints on master node
  56 kubectl taint nodes --all node-role.kubernetes.io/master- || true