opensource/mateclaw
2
0
Fork 0
mirror of https://gitee.com/mateos/mateclaw.git synced 2026-09-15 03:55:09 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(guard): check column existence before ALTER TABLE to suppress H2 duplicate column errors

Browse Source
This commit is contained in:
matevip 2026-04-06 12:37:12 +08:00
parent 08d1a89c0e
commit 647fa9ab94
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mateclaw-server/src/main/java/vip/mate/tool/guard/config/ToolGuardSchemaMigration.java
View File

@ -113,12 +113,26 @@ public class ToolGuardSchemaMigration implements ApplicationRunner {
* 为 mate_tool_guard_config 补充审计配置列（向已有表兼容迁移） * 为 mate_tool_guard_config 补充审计配置列（向已有表兼容迁移）
*/ */
private void migrateGuardConfigAuditColumns() { private void migrateGuardConfigAuditColumns() {
safeExecute("ALTER TABLE mate_tool_guard_config ADD COLUMN audit_enabled BOOLEAN NOT NULL DEFAULT TRUE"); safeAddColumn("mate_tool_guard_config", "audit_enabled", "BOOLEAN NOT NULL DEFAULT TRUE");
safeExecute("ALTER TABLE mate_tool_guard_config ADD COLUMN audit_min_severity VARCHAR(16) NOT NULL DEFAULT 'INFO'"); safeAddColumn("mate_tool_guard_config", "audit_min_severity", "VARCHAR(16) NOT NULL DEFAULT 'INFO'");
safeExecute("ALTER TABLE mate_tool_guard_config ADD COLUMN audit_retention_days INT NOT NULL DEFAULT 90"); safeAddColumn("mate_tool_guard_config", "audit_retention_days", "INT NOT NULL DEFAULT 90");
log.info("[ToolGuardSchemaMigration] audit columns migration done"); log.info("[ToolGuardSchemaMigration] audit columns migration done");
} }
private void safeAddColumn(String table, String column, String definition) {
try {
Integer count = jdbcTemplate.queryForObject(
"SELECT COUNT(*) FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = ? AND COLUMN_NAME = ?",
Integer.class, table.toUpperCase(), column.toUpperCase());
if (count != null && count > 0) {
return;
}
jdbcTemplate.execute("ALTER TABLE " + table + " ADD COLUMN " + column + " " + definition);
} catch (Exception e) {
log.debug("[ToolGuardSchemaMigration] Column migration skipped: {}", e.getMessage());
}
}
private void safeExecute(String sql) { private void safeExecute(String sql) {
try { try {
jdbcTemplate.execute(sql); jdbcTemplate.execute(sql);