140 lines
4.5 KiB
PowerShell
140 lines
4.5 KiB
PowerShell
# 设置输出编码为UTF-8
|
||
[Console]::OutputEncoding = [System.Text.Encoding]::UTF8
|
||
$OutputEncoding = [System.Text.Encoding]::UTF8
|
||
[Console]::InputEncoding = [System.Text.Encoding]::UTF8
|
||
|
||
# 启用详细输出和错误处理
|
||
$ErrorActionPreference = "Stop"
|
||
$VerbosePreference = "Continue"
|
||
|
||
# 检查并创建必要的目录
|
||
if (-not (Test-Path "./model")) {
|
||
Write-Output "Creating model directory..."
|
||
New-Item -ItemType Directory -Path "./model" | Out-Null
|
||
}
|
||
|
||
if (-not (Test-Path "../../app/user/model")) {
|
||
Write-Output "Creating target directory..."
|
||
New-Item -ItemType Directory -Path "../../app/user/model" -Force | Out-Null
|
||
}
|
||
|
||
# 将表名转换为驼峰命名法的函数
|
||
function ConvertToCamelCase {
|
||
param (
|
||
[string]$tableName
|
||
)
|
||
try {
|
||
# 将表名按_分割，并将每个部分首字母大写
|
||
$parts = $tableName -split '_'
|
||
$camelCase = ""
|
||
foreach ($part in $parts) {
|
||
if ($part.Length -gt 0) {
|
||
$camelCase += $part.Substring(0, 1).ToUpper() + $part.Substring(1).ToLower()
|
||
}
|
||
}
|
||
return $camelCase
|
||
}
|
||
catch {
|
||
Write-Output "Error in ConvertToCamelCase for table: $tableName"
|
||
Write-Output $_.Exception.Message
|
||
return $tableName # 出错时返回原表名
|
||
}
|
||
}
|
||
|
||
# 数据库连接信息
|
||
$DB_URL = "qnc:5vg67b3UNHu8@tcp(127.0.0.1:21001)/qnc"
|
||
$OUTPUT_DIR = "./model"
|
||
$TEMPLATE_DIR = "../template"
|
||
$TARGET_DIR = "../../app/user/model"
|
||
|
||
# 表名列表 - 每个元素后必须有逗号分隔
|
||
$tables = @(
|
||
"agent",
|
||
# "agent_active_stat",
|
||
"agent_audit",
|
||
"agent_real_name"
|
||
# "agent_closure",
|
||
# "agent_commission",
|
||
# "agent_commission_deduction",
|
||
# "agent_link",
|
||
# "agent_membership_config",
|
||
# "agent_membership_recharge_order",
|
||
# "agent_membership_user_config",
|
||
# "agent_order",
|
||
# "agent_platform_deduction",
|
||
# "agent_product_config",
|
||
# "agent_rewards",
|
||
# "agent_wallet",
|
||
# "agent_withdrawal",
|
||
# "feature",
|
||
# "global_notifications",
|
||
# "order",
|
||
# "product",
|
||
# "product_feature",
|
||
# "query",
|
||
# "user",
|
||
# "user_auth",
|
||
# "example",
|
||
# "authorization",
|
||
# "authorization_face"
|
||
)
|
||
|
||
# 为每个表生成模型
|
||
foreach ($table in $tables) {
|
||
try {
|
||
Write-Output "========================================="
|
||
Write-Output "Processing table: $table"
|
||
|
||
# 生成模型
|
||
Write-Output "Generating model..."
|
||
$command = "goctl model mysql datasource -url=`"$DB_URL`" -table=`"$table`" -dir=`"$OUTPUT_DIR`" --home=`"$TEMPLATE_DIR`" -cache=true --style=goZero"
|
||
Write-Output "Running command: $command"
|
||
Invoke-Expression $command
|
||
|
||
# 将表名转换为驼峰命名法
|
||
$camelCaseName = ConvertToCamelCase -tableName $table
|
||
Write-Output "Table name converted to: $camelCaseName"
|
||
|
||
# 定义源文件和目标文件路径
|
||
$sourceModelFile = "$OUTPUT_DIR/${camelCaseName}Model.go"
|
||
$sourceModelGenFile = "$OUTPUT_DIR/${camelCaseName}Model_gen.go"
|
||
$targetModelFile = "$TARGET_DIR/${camelCaseName}Model.go"
|
||
$targetModelGenFile = "$TARGET_DIR/${camelCaseName}Model_gen.go"
|
||
|
||
Write-Output "Source files:"
|
||
Write-Output " - $sourceModelFile"
|
||
Write-Output " - $sourceModelGenFile"
|
||
Write-Output "Target files:"
|
||
Write-Output " - $targetModelFile"
|
||
Write-Output " - $targetModelGenFile"
|
||
|
||
# 检查源文件是否存在
|
||
if (-not (Test-Path $sourceModelFile)) {
|
||
Write-Output "WARNING: Source file not found: $sourceModelFile"
|
||
}
|
||
if (-not (Test-Path $sourceModelGenFile)) {
|
||
Write-Output "WARNING: Source file not found: $sourceModelGenFile"
|
||
}
|
||
|
||
# 移动文件
|
||
if (Test-Path $sourceModelFile) {
|
||
Write-Output "Moving $sourceModelFile to $targetModelFile"
|
||
Move-Item -Path $sourceModelFile -Destination $targetModelFile -Force
|
||
}
|
||
if (Test-Path $sourceModelGenFile) {
|
||
Write-Output "Moving $sourceModelGenFile to $targetModelGenFile"
|
||
Move-Item -Path $sourceModelGenFile -Destination $targetModelGenFile -Force
|
||
}
|
||
|
||
Write-Output "Processing completed for table: $table"
|
||
}
|
||
catch {
|
||
Write-Output "ERROR processing table: $table"
|
||
Write-Output $_.Exception.Message
|
||
Write-Output $_.ScriptStackTrace
|
||
}
|
||
}
|
||
|
||
Write-Output "========================================="
|
||
Write-Output "Script execution completed."
|