From 77582e67a094ec3ebbeba038e5b2a01441c23d18 Mon Sep 17 00:00:00 2001 From: Suomm <1474983351@qq.com> Date: Wed, 30 Aug 2023 12:24:42 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E6=9B=B4=E5=A4=9A?= =?UTF-8?q?=E4=BF=A1=E6=81=AF=E8=BE=93=E5=87=BA=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../main/java/com/mybatisflex/codegen/Generator.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/Generator.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/Generator.java index 45a0dda9..f4c8f48d 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/Generator.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/Generator.java @@ -30,6 +30,7 @@ import java.sql.SQLException; import java.util.ArrayList; import java.util.Collection; import java.util.List; +import java.util.stream.Collectors; /** * 代码生成器。 @@ -55,11 +56,18 @@ public class Generator { } public void generate() { - try (Connection conn = dataSource.getConnection();) { + try (Connection conn = dataSource.getConnection()) { dbMeta = conn.getMetaData(); List tables = buildTables(conn); + if (tables.isEmpty()) { + System.err.printf("table %s not found.%n", globalConfig.getGenerateTables()); + return; + } else { + System.out.printf("find tables: %s%n", tables.stream().map(Table::getName).collect(Collectors.toSet())); + } + for (Table table : tables) { Collection generators = GeneratorFactory.getGenerators(); for (IGenerator generator : generators) {