refactor: share restore rollback helpers via mysql_common.sh

2026-07-05 15:04:20 +02:00
parent 74ac47f75b
commit 9e54bb32da
3 changed files with 51 additions and 106 deletions
@@ -487,3 +487,54 @@ mysql_restore_stream() {
    fi
    mysql_exec "$database" < "$file"
 }
+
+RESTORE_ROLLBACK_FILE=""
+RESTORE_ROLLBACK_CREATED=0
+
+cleanup_restore_rollback() {
+    if [ -n "${RESTORE_ROLLBACK_FILE:-}" ] && [ -f "$RESTORE_ROLLBACK_FILE" ]; then
+        rm -f "$RESTORE_ROLLBACK_FILE"
+    fi
+    RESTORE_ROLLBACK_FILE=""
+    RESTORE_ROLLBACK_CREATED=0
+}
+
+backup_alt_database_for_rollback() {
+    local db="$1"
+    cleanup_restore_rollback
+    RESTORE_ROLLBACK_FILE="$(mktemp)"
+
+    if [ -z "$(mysql_exec mysql -Nse "SELECT 1 FROM information_schema.SCHEMATA WHERE SCHEMA_NAME='$(mysql_escape_literal "$db")' LIMIT 1" 2>>"${LOG_FILE:-/dev/null}" || true)" ]; then
+        cleanup_restore_rollback
+        return 0
+    fi
+
+    if mysqldump_exec \
+        --single-transaction \
+        --quick \
+        --skip-lock-tables \
+        --routines \
+        --triggers \
+        --events \
+        --default-character-set=utf8mb4 \
+        "$db" > "$RESTORE_ROLLBACK_FILE" 2>>"${LOG_FILE:-/dev/null}"; then
+        RESTORE_ROLLBACK_CREATED=1
+        return 0
+    fi
+
+    cleanup_restore_rollback
+    return 1
+}
+
+restore_alt_database_from_rollback() {
+    local db="$1"
+    if [ "$RESTORE_ROLLBACK_CREATED" -ne 1 ] || [ -z "${RESTORE_ROLLBACK_FILE:-}" ] || [ ! -r "$RESTORE_ROLLBACK_FILE" ]; then
+        return 0
+    fi
+
+    mysql_exec mysql -e "DROP DATABASE IF EXISTS $(mysql_quote_identifier "$db");" >> "${LOG_FILE:-/dev/null}" 2>&1 || true
+    mysql_exec mysql -e "CREATE DATABASE $(mysql_quote_identifier "$db") CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;" >> "${LOG_FILE:-/dev/null}" 2>&1 || true
+    mysql_exec "$db" < "$RESTORE_ROLLBACK_FILE" >> "${LOG_FILE:-/dev/null}" 2>&1 || {
+        return 1
+    }
+}