Check exit codes of mysql and mysqldump

[psmysqlbackup.git] / psmysqlbackup.ps1

diff --git a/psmysqlbackup.ps1 b/psmysqlbackup.ps1
index 04ca03aaf77bf9094c19dc7b5b508f2db89a85fa..a33b144607c5e5ed8fbb8abf086934a815d19791 100644 (file)
--- a/psmysqlbackup.ps1
+++ b/psmysqlbackup.ps1
@@ -1,20 +1,28 @@
-[String]$configMysqlHost = "localhost"
-[Int32]$configMysqlPort = 3306
-[String]$configMysqlUser = "backup"
-[String]$configMysqlPassword = "backup"
+# Config
 
-[String]$configMysqlCli = "C:\Program Files\MariaDB 10.5\bin\mysql.exe"
-[String]$configMysqldumpCli = "C:\Program Files\MariaDB 10.5\bin\mysqldump.exe"
+$configMysqlHost = "localhost"
+$configMysqlPort = 3306
+$configMysqlUser = "backup"
+$configMysqlPassword = "backup"
 
-[String]$configBackupDir = "backup"
-[Int32]$configRotate = 7
+$configMysqlCli = "C:\Program Files\MariaDB 10.5\bin\mysql.exe"
+$configMysqldumpCli = "C:\Program Files\MariaDB 10.5\bin\mysqldump.exe"
 
-[String]$configDbExclusions = @("test")
+$configBackupDir = "backup"
+$configRotate = 7
 
-$defaultExclusions = @("information_schema", "performance_schema")
+$configDbBackup = @()
+$configDbExclusions = @("test")
+
+# End of config
 
 function Get-Databases() {
     $databaseString = (& $configMysqlCli --host=$configMysqlHost --port=$configMysqlPort --user=$configMysqlUser --password=$configMysqlPassword --batch --skip-column-names -e "SHOW DATABASES;")
+    
+    if($LastExitCode -ne 0) {
+        throw "MySQL CLI exited with Exit code $LastExitCode"
+    }
+    
     $databases = $databaseString.split([Environment]::NewLine)
 
     return $databases
@@ -22,6 +30,10 @@ function Get-Databases() {
 
 function Create-Backup([String]$database, [String]$target) {
     & $configMysqldumpCli --host=$configMysqlHost --port=$configMysqlPort --user=$configMysqlUser --password=$configMysqlPassword --single-transaction --result-file=$target $database
+
+    if($LastExitCode -ne 0) {
+        throw "mysqldump exited with Exit code $LastExitCode"
+    }
 }
 
 function Rotate-Backups($backupDir) {
@@ -44,11 +56,29 @@ function Rotate-Backups($backupDir) {
        }
 }
 
+$defaultExclusions = @("information_schema", "performance_schema")
+
 $currDaytime = Get-Date -format "yyyyMMdd-HHmmss"
 
-$databases = Get-Databases | Where-Object {!($_ -in $defaultExclusions -or $_ -in $configDbExclusions)}
+try {
+    $databases = Get-Databases | Where-Object {!($_ -in $defaultExclusions -or $_ -in $configDbExclusions)}
+}
+catch {
+    Write-Output "Failed to get list of databases"
+    Write-Output $_
+    exit 1
+}
+
+$databasesToBackup = @()
 
-foreach($d in $databases) {
+if($configDbBackup -and $configDbBackup.count -gt 0) {
+    $databasesToBackup = $configDbBackup
+}
+else {
+    $databasesToBackup = $databases
+}
+
+foreach($d in $databasesToBackup) {
     $databaseBackupDir = Join-Path -Path $configBackupDir -ChildPath $d
 
     if(!(Test-Path $databaseBackupDir)) {
@@ -57,6 +87,14 @@ foreach($d in $databases) {
 
     $databaseBackupFile = Join-Path -Path $databaseBackupDir -ChildPath "backup-$d-$currDaytime.sql"
     Write-Output "Backing up $d to $databaseBackupFile..."
-    Create-Backup $d $databaseBackupFile
+    
+    try {
+        Create-Backup $d $databaseBackupFile
+    }
+    catch {
+        Write-Output "Could not backup database $d to $databaseBackupFile"
+        Write-Output $_
+    }
+    
     Rotate-Backups $databaseBackupDir
 }
\ No newline at end of file