41 lines
1.1 KiB
PowerShell
41 lines
1.1 KiB
PowerShell
#!/usr/bin/env pwsh
|
|
# 查询所有素材类型
|
|
|
|
param(
|
|
[Parameter(Mandatory=$false)]
|
|
[ValidateSet('production', 'shadow')]
|
|
[string]$Target = 'production'
|
|
)
|
|
|
|
$ErrorActionPreference = 'Stop'
|
|
$ServerIP = "192.168.58.20"
|
|
|
|
if ($Target -eq 'production') {
|
|
$DatabaseName = "lobechat"
|
|
} else {
|
|
$DatabaseName = "lobechat_stage"
|
|
}
|
|
|
|
$SqlQuery = @'
|
|
SELECT DISTINCT "materialType", COUNT(*) as count
|
|
FROM materials
|
|
WHERE "deletedAt" IS NULL
|
|
GROUP BY "materialType"
|
|
ORDER BY "materialType";
|
|
'@
|
|
|
|
Write-Host "查询所有素材类型 ($Target)..." -ForegroundColor Cyan
|
|
|
|
$QueryBytes = [System.Text.Encoding]::UTF8.GetBytes($SqlQuery)
|
|
$QueryBase64 = [Convert]::ToBase64String($QueryBytes)
|
|
$RemoteCommand = "echo '$QueryBase64' | base64 -d | docker exec -i gongxiangpan-postgres-1 psql -U gongxiangpan -d $DatabaseName"
|
|
|
|
$Result = ssh -o StrictHostKeyChecking=no root@$ServerIP $RemoteCommand
|
|
|
|
if ($LASTEXITCODE -eq 0) {
|
|
Write-Host $Result -ForegroundColor Green
|
|
} else {
|
|
Write-Host "查询失败！" -ForegroundColor Red
|
|
Write-Host $Result -ForegroundColor Red
|
|
}
|