| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636 |
- using System.Diagnostics;
- using System.Text;
- using System.Data;
- using SqlSugar;
- namespace Admin.NET.Plugin.AiDOP.Infrastructure;
- /// <summary>
- /// S0 Excel 字段口径迁移:将现有旧列名重命名为 Excel 字段名。
- /// 仅处理文档中已确认映射的字段,不自动补 Excel 独有字段。
- /// </summary>
- public static class AidopS0ExcelColumnMigration
- {
- private sealed record ColumnRename(string TableName, string OldColumnName, string NewColumnName);
- private sealed record PrimaryKeyRepair(string TableName, string PkColumnName, bool IsIdentity, string PkComment, params string[] ShadowColumns);
- private sealed record MySqlColumnMeta(
- string ColumnName,
- string ColumnType,
- string IsNullable,
- string? ColumnDefault,
- string? Extra,
- string? ColumnComment,
- string? CollationName,
- string? CharacterSetName);
- private static readonly ColumnRename[] Renames =
- [
- new("CostCtrMaster", "cost_ctr", "CostCtr"),
- new("CostCtrMaster", "create_time", "CreateTime"),
- new("CostCtrMaster", "create_user", "CreateUser"),
- new("CostCtrMaster", "descr", "Descr"),
- new("CostCtrMaster", "domain_code", "Domain"),
- new("CostCtrMaster", "eff_time", "EffTime"),
- new("CostCtrMaster", "rec_id", "RecID"),
- new("CostCtrMaster", "ufld1", "Ufld1"),
- new("CostCtrMaster", "update_time", "UpdateTime"),
- new("CostCtrMaster", "update_user", "UpdateUser"),
- new("CustMaster", "create_time", "CreateTime"),
- new("CustMaster", "create_user", "CreateUser"),
- new("CustMaster", "credit_rating", "CreditRating"),
- new("CustMaster", "curr", "Curr"),
- new("CustMaster", "cust", "Cust"),
- new("CustMaster", "domain_code", "Domain"),
- new("CustMaster", "is_active", "IsActive"),
- new("CustMaster", "is_confirm", "IsConfirm"),
- new("CustMaster", "rec_id", "RecID"),
- new("CustMaster", "remarks", "Remarks"),
- new("CustMaster", "ship_to", "ShipTo"),
- new("CustMaster", "slspsn1", "Slspsn1"),
- new("CustMaster", "slspsn2", "Slspsn2"),
- new("CustMaster", "sort_name", "SortName"),
- new("CustMaster", "tax_class", "TaxClass"),
- new("CustMaster", "tax_in", "TaxIn"),
- new("CustMaster", "update_time", "UpdateTime"),
- new("CustMaster", "update_user", "UpdateUser"),
- new("DepartmentMaster", "create_time", "CreateTime"),
- new("DepartmentMaster", "create_user", "CreateUser"),
- new("DepartmentMaster", "department", "Department"),
- new("DepartmentMaster", "descr", "Descr"),
- new("DepartmentMaster", "domain_code", "Domain"),
- new("DepartmentMaster", "is_active", "IsActive"),
- new("DepartmentMaster", "rec_id", "RecID"),
- new("DepartmentMaster", "update_time", "UpdateTime"),
- new("DepartmentMaster", "update_user", "UpdateUser"),
- new("DevMonitor", "code_descr", "CodeDescr"),
- new("DevMonitor", "code_type", "CodeType"),
- new("DevMonitor", "create_time", "CreateTime"),
- new("DevMonitor", "create_user", "CreateUser"),
- new("DevMonitor", "device_descr", "DeviceDescr"),
- new("DevMonitor", "device_number", "DeviceNumber"),
- new("DevMonitor", "domain", "Domain"),
- new("DevMonitor", "id", "RecID"),
- new("DevMonitor", "low_limit", "LowLimit"),
- new("DevMonitor", "message_code", "MessageCode"),
- new("DevMonitor", "std_value", "StdValue"),
- new("DevMonitor", "top_limit", "TopLimit"),
- new("DevMonitor", "ufld1", "Ufld1"),
- new("DevMonitor", "um", "UM"),
- new("DevMonitor", "update_time", "UpdateTime"),
- new("DevMonitor", "update_user", "UpdateUser"),
- new("DevMonitor", "value_type", "ValueType"),
- new("EmpSkills", "create_time", "CreateTime"),
- new("EmpSkills", "create_user", "CreateUser"),
- new("EmpSkills", "domain", "Domain"),
- new("EmpSkills", "efficiency_coefficient", "EfficiencyCoefficient"),
- new("EmpSkills", "employee", "Employee"),
- new("EmpSkills", "employee_id", "EmployeeID"),
- new("EmpSkills", "employee_name", "EmployeeName"),
- new("EmpSkills", "id", "RecID"),
- new("EmpSkills", "site", "Site"),
- new("EmpSkills", "skill_description", "SkillDescription"),
- new("EmpSkills", "skill_level", "SkillLevel"),
- new("EmpSkills", "skill_no", "SkillNo"),
- new("EmpSkills", "update_time", "UpdateTime"),
- new("EmpSkills", "update_user", "UpdateUser"),
- new("EmpWorkDutyMaster", "create_time", "CreateTime"),
- new("EmpWorkDutyMaster", "create_user", "CreateUser"),
- new("EmpWorkDutyMaster", "domain_code", "Domain"),
- new("EmpWorkDutyMaster", "duty", "Duty"),
- new("EmpWorkDutyMaster", "emp_type", "EmpType"),
- new("EmpWorkDutyMaster", "employee", "Employee"),
- new("EmpWorkDutyMaster", "item_num1", "ItemNum1"),
- new("EmpWorkDutyMaster", "item_num2", "ItemNum2"),
- new("EmpWorkDutyMaster", "location", "Location"),
- new("EmpWorkDutyMaster", "prod_line", "ProdLine"),
- new("EmpWorkDutyMaster", "rec_id", "RecID"),
- new("EmpWorkDutyMaster", "ufld2", "Ufld2"),
- new("EmpWorkDutyMaster", "update_time", "UpdateTime"),
- new("EmpWorkDutyMaster", "update_user", "UpdateUser"),
- new("EmployeeMaster", "birth_date", "BirthDate"),
- new("EmployeeMaster", "car_id", "CarId"),
- new("EmployeeMaster", "create_time", "CreateTime"),
- new("EmployeeMaster", "create_user", "CreateUser"),
- new("EmployeeMaster", "date_employed", "DateEmployed"),
- new("EmployeeMaster", "date_terminated", "DateTerminated"),
- new("EmployeeMaster", "default_work_location", "DefaultWorkLocation"),
- new("EmployeeMaster", "department", "Department"),
- new("EmployeeMaster", "domain_code", "Domain"),
- new("EmployeeMaster", "email", "Email"),
- new("EmployeeMaster", "employee", "Employee"),
- new("EmployeeMaster", "employment_status", "EmploymentStatus"),
- new("EmployeeMaster", "is_active", "IsActive"),
- new("EmployeeMaster", "job_title", "JobTitle"),
- new("EmployeeMaster", "marital_status", "MaritalStatus"),
- new("EmployeeMaster", "name", "Name"),
- new("EmployeeMaster", "phone", "Phone"),
- new("EmployeeMaster", "rec_id", "RecID"),
- new("EmployeeMaster", "sex", "Sex"),
- new("EmployeeMaster", "update_time", "UpdateTime"),
- new("EmployeeMaster", "update_user", "UpdateUser"),
- new("EmployeeMaster", "work_ctr", "WorkCtr"),
- new("ItemMaster", "allocate_single_lot", "AllocateSingleLot"),
- new("ItemMaster", "auto_lot_nums", "AutoLotNums"),
- new("ItemMaster", "commodity_code", "CommodityCode"),
- new("ItemMaster", "create_time", "CreateTime"),
- new("ItemMaster", "create_user", "CreateUser"),
- new("ItemMaster", "days_between_pm", "DaysBetweenPM"),
- new("ItemMaster", "default_shelf", "DefaultShelf"),
- new("ItemMaster", "descr", "Descr"),
- new("ItemMaster", "descr1", "Descr1"),
- new("ItemMaster", "domain_code", "Domain"),
- new("ItemMaster", "drawing", "Drawing"),
- new("ItemMaster", "emt_type", "EMTType"),
- new("ItemMaster", "ins_lt", "InsLT"),
- new("ItemMaster", "inspect", "Inspect"),
- new("ItemMaster", "install", "Install"),
- new("ItemMaster", "is_active", "IsActive"),
- new("ItemMaster", "is_confirm", "IsConfirm"),
- new("ItemMaster", "is_main_mas", "IsMainMas"),
- new("ItemMaster", "issue_policy", "IssuePolicy"),
- new("ItemMaster", "item_num", "ItemNum"),
- new("ItemMaster", "item_type", "ItemType"),
- new("ItemMaster", "key_item", "KeyItem"),
- new("ItemMaster", "length", "Length"),
- new("ItemMaster", "location", "Location"),
- new("ItemMaster", "location_type", "LocationType"),
- new("ItemMaster", "lot_serial_control", "LotSerialControl"),
- new("ItemMaster", "max_ord", "MaxOrd"),
- new("ItemMaster", "mfg_lt", "MfgLT"),
- new("ItemMaster", "mfgmttr", "MFGMTTR"),
- new("ItemMaster", "min_ord", "MinOrd"),
- new("ItemMaster", "min_ord_sales", "MinOrdSales"),
- new("ItemMaster", "net_weight", "NetWeight"),
- new("ItemMaster", "net_weight_um", "NetWeightUM"),
- new("ItemMaster", "ord_mult", "OrdMult"),
- new("ItemMaster", "owner_application", "OwnerApplication"),
- new("ItemMaster", "pkg_code", "PkgCode"),
- new("ItemMaster", "planner", "Planner"),
- new("ItemMaster", "pur_lt", "PurLT"),
- new("ItemMaster", "pur_mfg", "PurMfg"),
- new("ItemMaster", "rec_id", "RecID"),
- new("ItemMaster", "remark", "Remark"),
- new("ItemMaster", "rev", "Rev"),
- new("ItemMaster", "safety_stk", "SafetyStk"),
- new("ItemMaster", "size", "Size"),
- new("ItemMaster", "size_um", "SizeUM"),
- new("ItemMaster", "status", "Status"),
- new("ItemMaster", "stock_turn_over", "StockTurnOver"),
- new("ItemMaster", "trace_detail", "TraceDetail"),
- new("ItemMaster", "um", "UM"),
- new("ItemMaster", "update_time", "UpdateTime"),
- new("ItemMaster", "update_user", "UpdateUser"),
- new("ItemOpCondition", "code_type", "CodeType"),
- new("ItemOpCondition", "create_time", "CreateTime"),
- new("ItemOpCondition", "create_user", "CreateUser"),
- new("ItemOpCondition", "descr", "Descr"),
- new("ItemOpCondition", "domain", "Domain"),
- new("ItemOpCondition", "id", "RecID"),
- new("ItemOpCondition", "is_main", "IsMain"),
- new("ItemOpCondition", "item_code", "ItemCode"),
- new("ItemOpCondition", "item_num", "ItemNum"),
- new("ItemOpCondition", "line", "Line"),
- new("ItemOpCondition", "op", "Op"),
- new("ItemOpCondition", "typed", "Typed"),
- new("ItemOpCondition", "update_time", "UpdateTime"),
- new("ItemOpCondition", "update_user", "UpdateUser"),
- new("ItemPackMaster", "create_time", "CreateTime"),
- new("ItemPackMaster", "create_user", "CreateUser"),
- new("ItemPackMaster", "cust_item", "CustItem"),
- new("ItemPackMaster", "domain_code", "Domain"),
- new("ItemPackMaster", "high", "High"),
- new("ItemPackMaster", "is_active", "IsActive"),
- new("ItemPackMaster", "iss_specific", "IssSpecific"),
- new("ItemPackMaster", "item_num", "ItemNum"),
- new("ItemPackMaster", "length", "Length"),
- new("ItemPackMaster", "net_weight", "NetWeight"),
- new("ItemPackMaster", "packing_qty", "PackingQty"),
- new("ItemPackMaster", "packing_type", "PackingType"),
- new("ItemPackMaster", "rec_id", "RecID"),
- new("ItemPackMaster", "remark", "Remark"),
- new("ItemPackMaster", "small_packing_qty", "SmallPackingQty"),
- new("ItemPackMaster", "update_time", "UpdateTime"),
- new("ItemPackMaster", "update_user", "UpdateUser"),
- new("ItemPackMaster", "weight_um", "WeightUM"),
- new("ItemPackMaster", "width", "Width"),
- new("ItemSubstituteDetail", "create_time", "CreateTime"),
- new("ItemSubstituteDetail", "create_user", "CreateUser"),
- new("ItemSubstituteDetail", "domain", "Domain"),
- new("ItemSubstituteDetail", "id", "RecID"),
- new("ItemSubstituteDetail", "item_num", "ItemNum"),
- new("ItemSubstituteDetail", "parent_item", "ParentItem"),
- new("ItemSubstituteDetail", "remark", "Remark"),
- new("ItemSubstituteDetail", "substitute_item", "SubstituteItem"),
- new("ItemSubstituteDetail", "substitute_qty", "SubstituteQty"),
- new("ItemSubstituteDetail", "update_time", "UpdateTime"),
- new("ItemSubstituteDetail", "update_user", "UpdateUser"),
- new("LineMaster", "create_time", "CreateTime"),
- new("LineMaster", "create_user", "CreateUser"),
- new("LineMaster", "domain", "Domain"),
- new("LineMaster", "id", "RecID"),
- new("LineMaster", "is_active", "IsActive"),
- new("LineMaster", "line", "Line"),
- new("LineMaster", "line_category", "LineCategory"),
- new("LineMaster", "line_type", "LineType"),
- new("LineMaster", "location", "Location"),
- new("LineMaster", "mid_location", "MidLocation"),
- new("LineMaster", "picking_location", "PickingLocation"),
- new("LineMaster", "update_time", "UpdateTime"),
- new("LineMaster", "update_user", "UpdateUser"),
- new("LineMaster", "v_location", "VLocation"),
- new("LineMaster", "workshop", "Workshop"),
- new("LineSkillDetail", "create_time", "CreateTime"),
- new("LineSkillMaster", "create_time", "CreateTime"),
- new("LineSkillMaster", "create_user", "CreateUser"),
- new("LineSkillMaster", "domain", "Domain"),
- new("LineSkillMaster", "id", "RecID"),
- new("LineSkillMaster", "job_descr", "JOBDescr"),
- new("LineSkillMaster", "job_no", "JOBNo"),
- new("LineSkillMaster", "prod_line", "ProdLine"),
- new("LineSkillMaster", "update_time", "UpdateTime"),
- new("LineSkillMaster", "update_user", "UpdateUser"),
- new("PriorityCode", "create_time", "CreateTime"),
- new("PriorityCode", "create_user", "CreateUser"),
- new("PriorityCode", "descr", "Descr"),
- new("PriorityCode", "domain_code", "Domain"),
- new("PriorityCode", "is_active", "IsActive"),
- new("PriorityCode", "priority", "Priority"),
- new("PriorityCode", "rec_id", "RecID"),
- new("PriorityCode", "source_column", "SourceColumn"),
- new("PriorityCode", "source_id", "SourceID"),
- new("PriorityCode", "source_table", "SourceTable"),
- new("PriorityCode", "source_type", "SourceType"),
- new("PriorityCode", "update_time", "UpdateTime"),
- new("PriorityCode", "update_user", "UpdateUser"),
- new("PriorityCode", "value", "Value"),
- new("PriorityCode", "value_id", "ValueID"),
- new("PriorityCode", "value_type", "ValueType"),
- new("ProdLineDetail", "create_time", "CreateTime"),
- new("ProdLineDetail", "create_user", "CreateUser"),
- new("ProdLineDetail", "domain", "Domain"),
- new("ProdLineDetail", "id", "RecID"),
- new("ProdLineDetail", "is_active", "IsActive"),
- new("ProdLineDetail", "is_confirm", "IsConfirm"),
- new("ProdLineDetail", "line", "Line"),
- new("ProdLineDetail", "mold_type_code", "MoldTypeCode"),
- new("ProdLineDetail", "op", "Op"),
- new("ProdLineDetail", "part", "Part"),
- new("ProdLineDetail", "rate", "Rate"),
- new("ProdLineDetail", "setup_time", "SetupTime"),
- new("ProdLineDetail", "site", "Site"),
- new("ProdLineDetail", "skill_no", "SkillNo"),
- new("ProdLineDetail", "standard_staff_count", "StandardStaffCount"),
- new("ProdLineDetail", "start", "Start"),
- new("ProdLineDetail", "update_time", "UpdateTime"),
- new("ProdLineDetail", "update_user", "UpdateUser"),
- new("ProductStructureMaster", "end_eff", "EndEff"),
- new("ProductStructureMaster", "id", "RecID"),
- new("ProductStructureMaster", "qty", "Qty"),
- new("ProductStructureMaster", "qty_consumed", "QtyConsumed"),
- new("ProductStructureMaster", "refs", "Refs"),
- new("ProductStructureMaster", "remark", "Remark"),
- new("ProductStructureMaster", "scrap", "Scrap"),
- new("ProductStructureMaster", "start_eff", "StartEff"),
- new("ProductStructureMaster", "structure_type", "StructureType"),
- new("ProductStructureMaster", "um", "UM"),
- new("ProductStructureOp", "id", "RecID"),
- new("ProductStructureOp", "op", "Op"),
- new("RoutingOpDetail", "comment_index", "CommentIndex"),
- new("RoutingOpDetail", "create_user", "CreateUser"),
- new("RoutingOpDetail", "id", "RecID"),
- new("RoutingOpDetail", "is_confirm", "IsConfirm"),
- new("RoutingOpDetail", "u_deci1", "UDeci1"),
- new("RoutingOpDetail", "u_deci2", "UDeci2"),
- new("RoutingOpDetail", "u_deci3", "UDeci3"),
- new("RoutingOpDetail", "u_deci4", "UDeci4"),
- new("RoutingOpDetail", "u_deci5", "UDeci5"),
- new("RoutingOpDetail", "ufld1", "Ufld1"),
- new("RoutingOpDetail", "ufld2", "Ufld2"),
- new("RoutingOpDetail", "ufld3", "Ufld3"),
- new("RoutingOpDetail", "ufld4", "Ufld4"),
- new("RoutingOpDetail", "ufld5", "Ufld5"),
- new("RoutingOpDetail", "ufld6", "Ufld6"),
- new("RoutingOpDetail", "ufld7", "Ufld7"),
- new("RoutingOpDetail", "ufld8", "Ufld8"),
- new("RoutingOpDetail", "ufld9", "Ufld9"),
- new("RoutingOpDetail", "update_user", "UpdateUser"),
- new("SuppMaster", "ap_end_day1", "APEndDay1"),
- new("SuppMaster", "ap_end_day2", "APEndDay2"),
- new("SuppMaster", "ap_start_day1", "APStartDay1"),
- new("SuppMaster", "ap_start_day2", "APStartDay2"),
- new("SuppMaster", "cr_terms", "CrTerms"),
- new("SuppMaster", "create_time", "CreateTime"),
- new("SuppMaster", "create_user", "CreateUser"),
- new("SuppMaster", "curr", "Curr"),
- new("SuppMaster", "domain_code", "Domain"),
- new("SuppMaster", "is_active", "IsActive"),
- new("SuppMaster", "pur_contact", "PurContact"),
- new("SuppMaster", "rec_id", "RecID"),
- new("SuppMaster", "remark", "Remark"),
- new("SuppMaster", "sort_name", "SortName"),
- new("SuppMaster", "supp", "Supp"),
- new("SuppMaster", "tax_class", "TaxClass"),
- new("SuppMaster", "tax_in", "TaxIn"),
- new("SuppMaster", "type", "Type"),
- new("SuppMaster", "update_time", "UpdateTime"),
- new("SuppMaster", "update_user", "UpdateUser"),
- new("WorkCtrMaster", "create_time", "CreateTime"),
- new("WorkCtrMaster", "create_user", "CreateUser"),
- new("WorkCtrMaster", "department", "Department"),
- new("WorkCtrMaster", "descr", "Descr"),
- new("WorkCtrMaster", "domain", "Domain"),
- new("WorkCtrMaster", "id", "RecID"),
- new("WorkCtrMaster", "is_active", "IsActive"),
- new("WorkCtrMaster", "update_time", "UpdateTime"),
- new("WorkCtrMaster", "update_user", "UpdateUser"),
- new("WorkCtrMaster", "work_ctr", "WorkCtr"),
- new("ic_substitute", "id", "Id"),
- new("ic_substitute", "is_deleted", "IsDeleted"),
- new("srm_purchase", "id", "Id"),
- new("srm_purchase", "is_require_goods", "IsRequireGoods"),
- ];
- private static readonly PrimaryKeyRepair[] PrimaryKeyRepairs =
- [
- new("ItemSubstituteDetail", "RecID", true, "物料替代关系主键", "Id"),
- new("CostCtrMaster", "RecID", true, "主键"),
- new("CustMaster", "RecID", true, "主键"),
- new("DepartmentMaster", "RecID", true, "主键"),
- new("DevMonitor", "RecID", true, "主键", "Id"),
- new("EmpSkills", "RecID", true, "主键", "Id"),
- new("EmpWorkDutyMaster", "RecID", true, "主键"),
- new("EmployeeMaster", "RecID", true, "主键"),
- new("ItemMaster", "RecID", true, "主键"),
- new("ItemOpCondition", "RecID", true, "物料工序生产要素主键", "Id"),
- new("ItemPackMaster", "RecID", true, "主键"),
- new("LineMaster", "RecID", true, "生产线主键", "Id"),
- new("LineSkillMaster", "RecID", true, "产线岗位主键", "Id"),
- new("PriorityCode", "RecID", true, "主键"),
- new("ProdLineDetail", "RecID", true, "生产线物料主键"),
- new("ProductStructureMaster", "RecID", true, "主键", "Id"),
- new("ProductStructureOp", "RecID", true, "主键", "Id"),
- new("RoutingOpDetail", "RecID", true, "主键", "Id"),
- new("SuppMaster", "RecID", true, "主键"),
- new("WorkCtrMaster", "RecID", true, "工作中心主键"),
- new("ic_substitute", "Id", false, "主键"),
- new("srm_purchase", "Id", true, "货源清单主键")
- ];
- public static void Migrate(ISqlSugarClient db)
- {
- var databaseName = db.Ado.SqlQuerySingle<string>("SELECT DATABASE();");
- foreach (var group in Renames.GroupBy(x => x.TableName))
- {
- var tableName = group.Key;
- var tables = db.DbMaintenance.GetTableInfoList(false);
- if (!tables.Any(t => t.Name.Equals(tableName, StringComparison.OrdinalIgnoreCase)))
- continue;
- try
- {
- var columns = db.DbMaintenance.GetColumnInfosByTableName(tableName, false) ?? [];
- foreach (var rename in group)
- {
- var oldColumn = columns.FirstOrDefault(c => c.DbColumnName.Equals(rename.OldColumnName, StringComparison.Ordinal));
- var newColumn = columns.FirstOrDefault(c => c.DbColumnName.Equals(rename.NewColumnName, StringComparison.Ordinal));
- if (oldColumn == null)
- continue;
- if (newColumn != null)
- {
- if (oldColumn.IsPrimarykey)
- {
- db.Ado.ExecuteCommand($"ALTER TABLE `{tableName}` DROP COLUMN `{rename.NewColumnName}`");
- Trace.TraceInformation($"Ai-DOP S0 drop duplicate new PK shadow column: {tableName}.{rename.NewColumnName}");
- RenameColumnMySql(db, databaseName, tableName, rename.OldColumnName, rename.NewColumnName);
- Trace.TraceInformation($"Ai-DOP S0 rename PK column: {tableName}.{rename.OldColumnName} -> {rename.NewColumnName}");
- }
- else
- {
- db.Ado.ExecuteCommand($"ALTER TABLE `{tableName}` DROP COLUMN `{rename.OldColumnName}`");
- Trace.TraceInformation($"Ai-DOP S0 drop duplicate old column: {tableName}.{rename.OldColumnName}");
- }
- columns = db.DbMaintenance.GetColumnInfosByTableName(tableName, false) ?? [];
- continue;
- }
- RenameColumnMySql(db, databaseName, tableName, rename.OldColumnName, rename.NewColumnName);
- Trace.TraceInformation($"Ai-DOP S0 rename column: {tableName}.{rename.OldColumnName} -> {rename.NewColumnName}");
- columns = db.DbMaintenance.GetColumnInfosByTableName(tableName, false) ?? [];
- }
- }
- catch (Exception ex)
- {
- Trace.TraceError($"Ai-DOP S0 rename columns failed for {tableName}: {ex}");
- throw;
- }
- }
- RepairPrimaryKeys(db, databaseName);
- }
- private static void RenameColumnMySql(ISqlSugarClient db, string databaseName, string tableName, string oldColumnName, string newColumnName)
- {
- var meta = GetMySqlColumnMeta(db, databaseName, tableName, oldColumnName);
- if (meta == null)
- return;
- var ddl = $"ALTER TABLE `{tableName}` CHANGE COLUMN `{oldColumnName}` `{newColumnName}` {BuildColumnDefinition(meta)}";
- db.Ado.ExecuteCommand(ddl);
- }
- private static void RepairPrimaryKeys(ISqlSugarClient db, string databaseName)
- {
- var tableList = db.DbMaintenance.GetTableInfoList(false);
- foreach (var repair in PrimaryKeyRepairs)
- {
- var tableName = tableList.FirstOrDefault(t => t.Name.Equals(repair.TableName, StringComparison.OrdinalIgnoreCase))?.Name;
- if (string.IsNullOrEmpty(tableName))
- continue;
- var tableColumns = db.DbMaintenance.GetColumnInfosByTableName(tableName, false) ?? [];
- if (tableColumns.Count == 0)
- continue;
- foreach (var shadowColumn in repair.ShadowColumns)
- {
- if (tableColumns.Any(c => c.DbColumnName.Equals(shadowColumn, StringComparison.Ordinal)))
- {
- db.Ado.ExecuteCommand($"ALTER TABLE `{tableName}` DROP COLUMN `{shadowColumn}`");
- }
- }
- SugarParameter[] pkParameters =
- [
- new SugarParameter("@db", databaseName),
- new SugarParameter("@table", tableName)
- ];
- var pkTable = db.Ado.GetDataTable(
- """
- SELECT COLUMN_NAME
- FROM information_schema.KEY_COLUMN_USAGE
- WHERE TABLE_SCHEMA = @db
- AND TABLE_NAME = @table
- AND CONSTRAINT_NAME = 'PRIMARY'
- ORDER BY ORDINAL_POSITION;
- """,
- pkParameters);
- var primaryKeyColumns = pkTable.Rows.Cast<DataRow>()
- .Select(r => r["COLUMN_NAME"]?.ToString())
- .Where(x => !string.IsNullOrWhiteSpace(x))
- .ToList();
- if (primaryKeyColumns.Count == 0)
- {
- primaryKeyColumns = tableColumns
- .Where(c => c.IsPrimarykey)
- .OrderBy(c => c.DbColumnName)
- .Select(c => c.DbColumnName)
- .Where(x => !string.IsNullOrWhiteSpace(x))
- .Select(x => (string?)x)
- .ToList();
- }
- var pkMatches = primaryKeyColumns.Count == 1
- && primaryKeyColumns[0]!.Equals(repair.PkColumnName, StringComparison.OrdinalIgnoreCase);
- if (!pkMatches)
- {
- if (primaryKeyColumns.Count > 0)
- db.Ado.ExecuteCommand($"ALTER TABLE `{tableName}` DROP PRIMARY KEY");
- db.Ado.ExecuteCommand($"ALTER TABLE `{tableName}` ADD PRIMARY KEY(`{repair.PkColumnName}`)");
- }
- if (!repair.IsIdentity)
- continue;
- var meta = GetMySqlColumnMeta(db, databaseName, tableName, repair.PkColumnName)
- ?? GetMySqlColumnMeta(db, databaseName, tableName, primaryKeyColumns.FirstOrDefault() ?? repair.PkColumnName);
- if (meta == null || ContainsToken(meta.Extra, "auto_increment"))
- continue;
- var ddl = $"ALTER TABLE `{tableName}` MODIFY COLUMN `{meta.ColumnName}` {BuildColumnDefinition(meta, forceNotNull: true, forceAutoIncrement: true, skipDefault: true, commentOverride: repair.PkComment)}";
- db.Ado.ExecuteCommand(ddl);
- }
- }
- private static MySqlColumnMeta? GetMySqlColumnMeta(ISqlSugarClient db, string databaseName, string tableName, string columnName)
- {
- var sql = """
- SELECT
- COLUMN_NAME AS ColumnName,
- COLUMN_TYPE AS ColumnType,
- IS_NULLABLE AS IsNullable,
- COLUMN_DEFAULT AS ColumnDefault,
- EXTRA AS Extra,
- COLUMN_COMMENT AS ColumnComment,
- COLLATION_NAME AS CollationName,
- CHARACTER_SET_NAME AS CharacterSetName
- FROM information_schema.COLUMNS
- WHERE TABLE_SCHEMA = @db
- AND TABLE_NAME = @table
- AND BINARY COLUMN_NAME = @column
- LIMIT 1;
- """;
- SugarParameter[] parameters =
- [
- new SugarParameter("@db", databaseName),
- new SugarParameter("@table", tableName),
- new SugarParameter("@column", columnName)
- ];
- var table = db.Ado.GetDataTable(sql, parameters);
- if (table.Rows.Count == 0)
- return null;
- var row = table.Rows[0];
- return new MySqlColumnMeta(
- row["ColumnName"]?.ToString() ?? columnName,
- row["ColumnType"]?.ToString() ?? throw new InvalidOperationException($"Missing ColumnType for {tableName}.{columnName}"),
- row["IsNullable"]?.ToString() ?? "YES",
- row["ColumnDefault"] == DBNull.Value ? null : row["ColumnDefault"]?.ToString(),
- row["Extra"] == DBNull.Value ? null : row["Extra"]?.ToString(),
- row["ColumnComment"] == DBNull.Value ? null : row["ColumnComment"]?.ToString(),
- row["CollationName"] == DBNull.Value ? null : row["CollationName"]?.ToString(),
- row["CharacterSetName"] == DBNull.Value ? null : row["CharacterSetName"]?.ToString());
- }
- private static string BuildColumnDefinition(MySqlColumnMeta meta, bool forceNotNull = false, bool forceAutoIncrement = false, bool skipDefault = false, string? commentOverride = null)
- {
- var builder = new StringBuilder();
- builder.Append(meta.ColumnType);
- if (!string.IsNullOrWhiteSpace(meta.CharacterSetName))
- builder.Append($" CHARACTER SET {meta.CharacterSetName}");
- if (!string.IsNullOrWhiteSpace(meta.CollationName))
- builder.Append($" COLLATE {meta.CollationName}");
- var isNullable = !forceNotNull && meta.IsNullable.Equals("YES", StringComparison.OrdinalIgnoreCase);
- builder.Append(isNullable ? " NULL" : " NOT NULL");
- if (!skipDefault && meta.ColumnDefault != null)
- {
- builder.Append(" DEFAULT ");
- builder.Append(FormatDefaultValue(meta.ColumnType, meta.ColumnDefault));
- }
- else if (!skipDefault && isNullable &&
- !ContainsToken(meta.Extra, "auto_increment") &&
- !ContainsToken(meta.Extra, "DEFAULT_GENERATED"))
- {
- builder.Append(" DEFAULT NULL");
- }
- if (forceAutoIncrement || ContainsToken(meta.Extra, "auto_increment"))
- builder.Append(" AUTO_INCREMENT");
- if (ContainsToken(meta.Extra, "on update CURRENT_TIMESTAMP"))
- builder.Append(" ON UPDATE CURRENT_TIMESTAMP");
- var comment = commentOverride ?? meta.ColumnComment;
- if (!string.IsNullOrWhiteSpace(comment))
- builder.Append($" COMMENT '{EscapeSqlLiteral(comment)}'");
- return builder.ToString();
- }
- private static string FormatDefaultValue(string columnType, string columnDefault)
- {
- if (columnDefault.Equals("NULL", StringComparison.OrdinalIgnoreCase))
- return "NULL";
- if (columnDefault.StartsWith("CURRENT_TIMESTAMP", StringComparison.OrdinalIgnoreCase))
- return columnDefault;
- return IsNumericLikeType(columnType)
- ? columnDefault
- : $"'{EscapeSqlLiteral(columnDefault)}'";
- }
- private static bool IsNumericLikeType(string columnType)
- {
- var normalized = columnType.ToLowerInvariant();
- return normalized.StartsWith("tinyint") ||
- normalized.StartsWith("smallint") ||
- normalized.StartsWith("mediumint") ||
- normalized.StartsWith("int") ||
- normalized.StartsWith("bigint") ||
- normalized.StartsWith("decimal") ||
- normalized.StartsWith("numeric") ||
- normalized.StartsWith("float") ||
- normalized.StartsWith("double") ||
- normalized.StartsWith("real") ||
- normalized.StartsWith("bit") ||
- normalized.StartsWith("bool") ||
- normalized.StartsWith("boolean");
- }
- private static bool ContainsToken(string? value, string token)
- => !string.IsNullOrWhiteSpace(value) && value.Contains(token, StringComparison.OrdinalIgnoreCase);
- private static string EscapeSqlLiteral(string value)
- => value.Replace("\\", "\\\\").Replace("'", "\\'");
- }
|