110 lines
3.9 KiB
PowerShell
110 lines
3.9 KiB
PowerShell
#!PowerShell
|
|
|
|
# Function: Import data from script files or RAR script files into MySQL specific libraries
|
|
# Author: JiangJun
|
|
# Create-Time: 2025-8-10 15:57:26
|
|
# Version: 0.3
|
|
|
|
# Set-ExecutionPolicy RemoteSigned -Scope Process -Force
|
|
|
|
$logFile = "ScriptExecutionLog_$(Get-Date -Format 'yyyyMMdd_HHmmss').txt"
|
|
|
|
function Write-Log {
|
|
param (
|
|
[string]$message
|
|
)
|
|
$timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
|
|
$logMessage = "[$timestamp] $message"
|
|
# [debug]
|
|
# Write-Output $logMessage | Out-File -FilePath $logFile -Append
|
|
Write-Host $logMessage
|
|
}
|
|
|
|
function randomText {
|
|
$chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789'.ToCharArray()
|
|
$length = 12
|
|
$random = New-Object System.Random
|
|
$string = -join (1..$length | ForEach-Object { $chars[$random.Next(0, $chars.Length)] })
|
|
return $string
|
|
}
|
|
|
|
function Import-Script {
|
|
param (
|
|
[string]$dbServer,
|
|
[string]$dbPort,
|
|
[string]$dbUser,
|
|
[string]$dbPassword
|
|
)
|
|
if (Test-Path -Path ".\script" -PathType Container) {
|
|
Write-Log "Found script folder, starting to process .."
|
|
$subFolders = Get-ChildItem -Path ".\script" -Directory
|
|
if ($subFolders.Count -eq 0) {
|
|
Write-Log "There are no subfolders in the script folder, skipping processing."
|
|
exit
|
|
}
|
|
|
|
foreach ($folder in $subFolders) {
|
|
Write-Log "Database connection parameters - Please modify according to the actual situation $($folder.Name)"
|
|
$rarFiles = Get-ChildItem -Path $folder.FullName -Filter "*.rar"
|
|
if ($rarFiles.Count -eq 0) {
|
|
Write-Log "There are no SQL files in the folder $($folder.Name), skipping."
|
|
continue
|
|
}
|
|
|
|
foreach ($rarFile in $rarFiles) {
|
|
Write-Log "Processing file: $($rarFile.Name) (located in $($folder.Name))"
|
|
$rarName = $rarFile.BaseName
|
|
$rarDirectoryName = (Get-Item $rarFile.FullName).DirectoryName
|
|
try {
|
|
Start-Process -FilePath "rar" -ArgumentList "x -pshzyh!234 `"$($rarFile.FullName)`" `"$rarDirectoryName`"" -NoNewWindow -Wait
|
|
if (Test-Path $rarFile.FullName) {
|
|
Remove-Item $rarFile.FullName -Force
|
|
}
|
|
} catch {
|
|
Write-Log "Error unrar $($rarFile.Name):$_"
|
|
Write-Log "=> $($rarFile.FullName)"
|
|
}
|
|
}
|
|
}
|
|
|
|
foreach ($folder in $subFolders) {
|
|
Write-Log "Database connection parameters - Please modify according to the actual situation $($folder.Name)"
|
|
$sqlFiles = Get-ChildItem -Path $folder.FullName -Filter "*.sql"
|
|
if ($sqlFiles.Count -eq 0) {
|
|
Write-Log "There are no SQL files in the folder $($folder.Name), skipping."
|
|
continue
|
|
}
|
|
foreach ($sqlFile in $sqlFiles) {
|
|
Write-Log "Processing file: $($sqlFile.Name) (located in $($folder.Name))"
|
|
$dbName = $sqlFile.BaseName
|
|
try {
|
|
Write-Log "Creating database $dbName and importing $($sqlFile.Name)"
|
|
Start-Process -FilePath "mysql" -ArgumentList "-h$dbServer -u$dbUser -p$dbPassword -P$dbPort $($folder.Name)" -RedirectStandardInput $sqlFile.FullName -NoNewWindow -Wait
|
|
Write-Log "Successfully imported $($sqlFile.Name) into database $($folder.Name)"
|
|
|
|
$date = $(Get-Date -Format "yyyy-MM-dd")
|
|
$random = randomText
|
|
$target = ".\patch\$($date)"
|
|
New-Item -Path "$target" -ItemType Directory -Force
|
|
Start-Process -FilePath "rar" -ArgumentList "a -pPassword#$date! -hp $target\$($random).rar `"$($sqlFile.FullName)`"" -NoNewWindow -Wait
|
|
|
|
if (Test-Path $sqlFile.FullName) {
|
|
Remove-Item $sqlFile.FullName -Force
|
|
}
|
|
} catch {
|
|
Write-Log "Error importing $($sqlFile.Name):$_"
|
|
Write-Log "=> $($sqlFile.FullName)"
|
|
}
|
|
}
|
|
}
|
|
|
|
Write-Log "All script processing has been completed."
|
|
}
|
|
}
|
|
|
|
# Param 1: MySQL IP
|
|
# Param 2: MySQL Port
|
|
# Param 3: MySQL root
|
|
# Param 4: MySQL root_password
|
|
Import-Script "127.0.0.1" "33306" "root" "Root@2025"
|