diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java index 306afc57..fe6ac1c2 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/entity/Table.java @@ -131,12 +131,6 @@ public class Table { public void addColumn(Column column) { - //排除忽略列 - if (globalConfig.getStrategyConfig().getIgnoreColumns() != null && - globalConfig.getStrategyConfig().getIgnoreColumns().contains(column.getName().toLowerCase())) { - return; - } - //主键 if (primaryKeys != null && primaryKeys.contains(column.getName())) { column.setPrimaryKey(true); diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java index b7f87bf5..28600499 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/generator/impl/EntityGenerator.java @@ -62,7 +62,10 @@ public class EntityGenerator implements IGenerator { if (entityJavaFile.exists() && !entityConfig.isOverwriteEnable()) { return; } - + //排除忽略列 + if (globalConfig.getStrategyConfig().getIgnoreColumns() != null) { + table.getColumns().removeIf(column -> globalConfig.getStrategyConfig().getIgnoreColumns().contains(column.getName().toLowerCase())); + } Map params = new HashMap<>(4); params.put("table", table);