Make restore pod yaml specific to restoring node.

Author: retroflexer

bindata/etcd/cluster-restore.sh

@@ -53,6 +53,32 @@ function restore_static_pods() {
     tar -xvf "${BACKUP_FILE}" --strip-components=2 -C "${MANIFEST_DIR}"/ "${BACKUP_POD_PATH}"
   done
 }
+function replace_node_specific_parameters() {
+  pod_file="$1"
+# shellcheck disable=SC2016
+  declare -A SUBS_MAP
+  SUBS_MAP['export ETCD_NAME=.*$']='export ETCD_NAME=${NODE_NODE_ENVVAR_NAME_ETCD_NAME}'
+  SUBS_MAP['export ETCD_INITIAL_CLUSTER=.*$']='export ETCD_INITIAL_CLUSTER="${ETCD_NAME}=https://${NODE_NODE_ENVVAR_NAME_ETCD_URL_HOST}:2380"'
+  SUBS_MAP['export ETCD_NODE_PEER_URL=.*$']='export ETCD_NODE_PEER_URL=https://${NODE_NODE_ENVVAR_NAME_ETCD_URL_HOST}:2380'
+  SUBS_MAP['--initial-advertise-peer-urls .*$']='--initial-advertise-peer-urls $ETCD_NODE_PEER_URL \\'
+  SUBS_MAP['--initial-advertise-peer-urls=.*$']='--initial-advertise-peer-urls=https://${NODE_NODE_ENVVAR_NAME_IP}:2380 \\'
+  SUBS_MAP['--cert-file=.*$']='--cert-file=/etc/kubernetes/static-pod-certs/secrets/etcd-all-serving/etcd-serving-NODE_NAME.crt \\'
+  SUBS_MAP['--key-file=.*$']='--key-file=/etc/kubernetes/static-pod-certs/secrets/etcd-all-serving/etcd-serving-NODE_NAME.key \\'
+  SUBS_MAP['--peer-cert-file=.*$']='--peer-cert-file=/etc/kubernetes/static-pod-certs/secrets/etcd-all-peer/etcd-peer-NODE_NAME.crt \\'
+  SUBS_MAP['--peer-key-file=.*$']='--peer-key-file=/etc/kubernetes/static-pod-certs/secrets/etcd-all-peer/etcd-peer-NODE_NAME.key \\'
+  SUBS_MAP['--advertise-client-urls=.*$']='--advertise-client-urls=https://${NODE_NODE_ENVVAR_NAME_IP}:2379 \\'
+
+  for pattern in "${!SUBS_MAP[@]}"; do
+    # check if the pattern exists, and return with error if it doesn't
+    if ! grep -- "$pattern" "$pod_file" > /dev/null 2>&1; then
+      echo "Warning: restore pod extraction failed: expected pattern \"$pattern\" not found."
+      return 1
+    fi
+    # replace the pattern
+    sed -i "s,$pattern,${SUBS_MAP[$pattern]},g"  "$pod_file"
+  done
+}
+
 
 function extract_and_start_restore_etcd_pod() {
   POD_FILE_NAME="restore-etcd-pod/pod.yaml"
@@ -64,6 +90,13 @@ function extract_and_start_restore_etcd_pod() {
 
    echo "starting restored etcd-pod.yaml"
    tar -O -xvf "${BACKUP_FILE}" -C "${MANIFEST_DIR}"/ "${BACKUP_POD_PATH}" > "${MANIFEST_DIR}"/etcd-pod.yaml
+
+   # Since the backup could have been taken on any node, we need to replace the node specific values in the restore pod.
+   echo "replacing node specific parameters in restore etcd-pod.yaml"
+   if ! replace_node_specific_parameters "${MANIFEST_DIR}"/etcd-pod.yaml; then
+     return 1
+   fi
+
    return 0
 }
 
@@ -132,6 +165,7 @@ cp -p "${SNAPSHOT_FILE}" "${ETCD_DATA_DIR_BACKUP}"/snapshot.db
 # Extract and start restore-etcd pod.yaml
 if ! extract_and_start_restore_etcd_pod; then
   # If backup doesn't have restore-pod.yaml, copy it from local disk
+  echo "Warning: could not extract restore pod. Using the restore pod manifest on the disk."
   cp -p ${RESTORE_ETCD_POD_YAML} ${MANIFEST_DIR}/etcd-pod.yaml
 fi