diff --git a/src/main/java/io/jboot/codegen/model/JbootBaseModelGenerator.java b/src/main/java/io/jboot/codegen/model/JbootBaseModelGenerator.java index f54d0bc2729812024f0ce077eabab0c7491db3fd..78cae4b83f25a3d3d71c32be1a65fe24df3ae446 100644 --- a/src/main/java/io/jboot/codegen/model/JbootBaseModelGenerator.java +++ b/src/main/java/io/jboot/codegen/model/JbootBaseModelGenerator.java @@ -59,6 +59,11 @@ public class JbootBaseModelGenerator extends BaseModelGenerator { metaBuilder.setGenerateRemarks(generateRemarks); return this; } - + public JbootBaseModelGenerator addWhitelist(String... tableNames) { + if (tableNames != null) { + this.metaBuilder.addWhitelist(tableNames); + } + return this; + } }