From de4d377065be5e1bce95c910d7fca32ba77242e1 Mon Sep 17 00:00:00 2001 From: smallchill Date: Thu, 25 Aug 2022 01:47:06 +0800 Subject: [PATCH] =?UTF-8?q?:zap:=20=E5=8A=A0=E5=BF=AB=E7=89=A9=E7=90=86?= =?UTF-8?q?=E8=A1=A8=E4=BF=A1=E6=81=AF=E8=AF=BB=E5=8F=96=E9=80=9F=E5=BA=A6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../develop/controller/ModelController.java | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) diff --git a/blade-ops/blade-develop/src/main/java/org/springblade/develop/controller/ModelController.java b/blade-ops/blade-develop/src/main/java/org/springblade/develop/controller/ModelController.java index fb522633..f5f3e2cb 100644 --- a/blade-ops/blade-develop/src/main/java/org/springblade/develop/controller/ModelController.java +++ b/blade-ops/blade-develop/src/main/java/org/springblade/develop/controller/ModelController.java @@ -197,7 +197,7 @@ public class ModelController extends BladeController { */ private TableInfo getTableInfo(String tableName, Long datasourceId) { Datasource datasource = datasourceService.getById(datasourceId); - ConfigBuilder config = getConfigBuilder(datasource); + ConfigBuilder config = getConfigBuilder(datasource, tableName); List tableInfoList = config.getTableInfoList(); TableInfo tableInfo = null; Iterator iterator = tableInfoList.stream().filter(table -> table.getName().equals(tableName)).collect(Collectors.toList()).iterator(); @@ -214,8 +214,21 @@ public class ModelController extends BladeController { * @param datasource 数据源信息 */ private ConfigBuilder getConfigBuilder(Datasource datasource) { - StrategyConfig strategyConfig = new StrategyConfig.Builder() - .entityBuilder() + return getConfigBuilder(datasource, null); + } + + /** + * 获取表配置信息 + * + * @param datasource 数据源信息 + * @param tableName 表名 + */ + private ConfigBuilder getConfigBuilder(Datasource datasource, String tableName) { + StrategyConfig.Builder builder = new StrategyConfig.Builder(); + if (StringUtil.isNotBlank(tableName)) { + builder.addInclude(tableName); + } + StrategyConfig strategyConfig = builder.entityBuilder() .naming(NamingStrategy.underline_to_camel) .columnNaming(NamingStrategy.underline_to_camel).build(); DataSourceConfig datasourceConfig = new DataSourceConfig.Builder(