diff --git a/src/main/java/io/jboot/codegen/model/JbootModelGenerator.java b/src/main/java/io/jboot/codegen/model/JbootModelGenerator.java index 41a95958992902ce8aff1f0ef7345cc0e24401a6..3b9ac67a04c3cd9abe34fb509a1906f6003fcb7f 100644 --- a/src/main/java/io/jboot/codegen/model/JbootModelGenerator.java +++ b/src/main/java/io/jboot/codegen/model/JbootModelGenerator.java @@ -55,6 +55,11 @@ public class JbootModelGenerator extends ModelGenerator { metaBuilder.setGenerateRemarks(generateRemarks); return this; } - - + //增加白名单功能 + public JbootModelGenerator addWhitelist(String... tableNames) { + if (tableNames != null) { + this.metaBuilder.addWhitelist(tableNames); + } + return this; + } }