Check exit codes of mysql and mysqldump

[psmysqlbackup.git] / psmysqlbackup.ps1

diff --git a/psmysqlbackup.ps1 b/psmysqlbackup.ps1
index dca87c2b8a8c01c95bd31c8c0fa1a46b7c5f36d4..a33b144607c5e5ed8fbb8abf086934a815d19791 100644 (file)
--- a/psmysqlbackup.ps1
+++ b/psmysqlbackup.ps1
@@ -1,28 +1,100 @@
-[String]$config_mysql_host = "localhost"
-[String]$config_mysql_user = "backup"
-[String]$config_mysql_password = "backup"
+# Config
 
-[String]$config_mysql_cli = "C:\Program Files\MariaDB 10.5\bin\mysql.exe"
-[String]$config_mysqldump_cli = "C:\Program Files\MariaDB 10.5\bin\mysqldump.exe"
+$configMysqlHost = "localhost"
+$configMysqlPort = 3306
+$configMysqlUser = "backup"
+$configMysqlPassword = "backup"
 
-[String]$config_backup_dir = "backup"
-[String]$config_rotate = 7
+$configMysqlCli = "C:\Program Files\MariaDB 10.5\bin\mysql.exe"
+$configMysqldumpCli = "C:\Program Files\MariaDB 10.5\bin\mysqldump.exe"
+
+$configBackupDir = "backup"
+$configRotate = 7
+
+$configDbBackup = @()
+$configDbExclusions = @("test")
+
+# End of config
 
 function Get-Databases() {
-    $databaseString = (& $config_mysql_cli --host=$config_mysql_host --user=$config_mysql_user --password=$config_mysql_password --batch --skip-column-names -e "SHOW 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
 }
 
 function Create-Backup([String]$database, [String]$target) {
-    & $config_mysqldump_cli --host=$config_mysql_host --user=$config_mysql_user --password=$config_mysql_password --single-transaction --result-file=$target $database
+    & $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) {
+    if($configRotate -le 0) {
+               return
+       }
+       
+       $keepBackupsCount = $configRotate
+       
+       Get-ChildItem $backupDir -File | Where-Object {($_.Name -match "^backup-.+-\d{8,}-\d{6}\.sql$")} | Sort-Object -Descending |
+       Foreach-Object {
+               if($keepBackupsCount -ge 0) {
+                       $keepBackupsCount--
+               }
+               
+               if($keepBackupsCount -eq -1) {
+                       Write-Output "Deleting backup $($_.FullName)"
+                       Remove-Item -Force $_.FullName
+               }
+       }
+}
+
+$defaultExclusions = @("information_schema", "performance_schema")
+
+$currDaytime = Get-Date -format "yyyyMMdd-HHmmss"
+
+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 = @()
+
+if($configDbBackup -and $configDbBackup.count -gt 0) {
+    $databasesToBackup = $configDbBackup
 }
+else {
+    $databasesToBackup = $databases
+}
+
+foreach($d in $databasesToBackup) {
+    $databaseBackupDir = Join-Path -Path $configBackupDir -ChildPath $d
 
-$databases = Get-Databases | Where-Object { $_ -ne "information_schema" -and $_ -ne "performance_schema"}
+    if(!(Test-Path $databaseBackupDir)) {
+        New-Item -ItemType directory -Path $databaseBackupDir -ErrorAction Stop | Out-Null
+    }
 
-foreach($d in $databases) {
-    $backupFile = $config_backup_dir + "\" + $d + ".sql"
-    Write-Output "Backing up $d to $backupFile..."
-    Create-Backup $d $backupFile
+    $databaseBackupFile = Join-Path -Path $databaseBackupDir -ChildPath "backup-$d-$currDaytime.sql"
+    Write-Output "Backing up $d to $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