From deeb93ed553db78df6fb7e69e81265655a728424 Mon Sep 17 00:00:00 2001
From: xujianxie <xujianxie@163.com>
Date: Wed, 1 Jun 2022 00:32:36 +0000
Subject: [PATCH] =?UTF-8?q?update=20src/main/java/io/jboot/codegen/model/J?=
 =?UTF-8?q?bootModelGenerator.java.=20=20=20=20=20//=E5=A2=9E=E5=8A=A0?=
 =?UTF-8?q?=E7=99=BD=E5=90=8D=E5=8D=95=E5=8A=9F=E8=83=BD=20=20=20=20=20pub?=
 =?UTF-8?q?lic=20JbootModelGenerator=20addWhitelist(String...=20tableNames?=
 =?UTF-8?q?)=20{=20=20=20=20=20=20=20=20=20if=20(tableNames=20!=3D=20null)?=
 =?UTF-8?q?=20{=20=20=20=20=20=20=20=20=20=20=20=20=20this.metaBuilder.add?=
 =?UTF-8?q?Whitelist(tableNames);=20=20=20=20=20=20=20=20=20}=20=20=20=20?=
 =?UTF-8?q?=20=20=20=20=20return=20this;=20=20=20=20=20}?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/io/jboot/codegen/model/JbootModelGenerator.java | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/main/java/io/jboot/codegen/model/JbootModelGenerator.java b/src/main/java/io/jboot/codegen/model/JbootModelGenerator.java
index 41a95958..3b9ac67a 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;
+    }
 }
-- 
Gitee