Discard broken S3 downloads

nienbo · Oct 27, 2023 · c8e7b29 · c8e7b29
1 parent 9df04eb
commit c8e7b29
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/lib/backends/s3.bash b/lib/backends/s3.bash
@@ -112,7 +112,15 @@ function restore() {
   fi
 
   if [[ ! "${BK_AWS_FOUND}" =~ (false) ]]; then
-    aws s3 cp ${BK_CUSTOM_AWS_ARGS} "s3://${BUCKET}/${TAR_FILE}" .
+    TMP_FILE="$(mktemp)"
+    aws s3 cp ${BK_CUSTOM_AWS_ARGS} "s3://${BUCKET}/${TAR_FILE}" "${TMP_FILE}" || s3_download_failed=true
+    if ${s3_download_failed:-false}; then
+      echo -e "AWS download failed, skipping cache restore..."
+      exit 0
+    else
+      mv "${TMP_FILE}" "${TAR_FILE}"
+    fi
+
     [ "${BK_CACHE_SAVE_CACHE}" == "true" ] && cp "${TAR_FILE}" "${BK_CACHE_LOCAL_PATH}/${TAR_FILE}"
     tar ${BK_TAR_EXTRACT_ARGS} "${TAR_FILE}" -C .
   else