From e775f4514803b04659e4ef08fa2473417cbcbc95 Mon Sep 17 00:00:00 2001
From: zhouhongfa <zhouhongfa@gz-yibo.com>
Date: Wed, 31 Jul 2019 21:08:15 +0800
Subject: [PATCH] =?UTF-8?q?add:=20=E5=A2=9E=E5=8A=A0=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E6=89=80=E6=9C=89=E5=8F=AF=E7=94=A8=E8=A1=A8=E5=90=8D=E6=96=B9?=
 =?UTF-8?q?=E6=B3=95=EF=BC=8C=E5=8F=AA=E5=8F=96=E8=A1=A8=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/wugui/tool/query/BaseQueryTool.java   | 14 ++++++++++++++
 .../com/wugui/tool/query/QueryToolInterface.java   |  8 ++++++++
 .../com/wugui/tool/query/MySQLQueryToolTest.java   |  6 ++++++
 3 files changed, 28 insertions(+)

diff --git a/datax-web/src/main/java/com/wugui/tool/query/BaseQueryTool.java b/datax-web/src/main/java/com/wugui/tool/query/BaseQueryTool.java
index aa1bea46..73ba9d77 100644
--- a/datax-web/src/main/java/com/wugui/tool/query/BaseQueryTool.java
+++ b/datax-web/src/main/java/com/wugui/tool/query/BaseQueryTool.java
@@ -275,4 +275,18 @@ public abstract class BaseQueryTool implements QueryToolInterface {
         }
         return columns;
     }
+
+    @Override
+    public List<String> getTableNames() {
+        List<String> res = Lists.newArrayList();
+        List<Map<String, Object>> tables = getTables();
+        //这里只取表名
+        tables.forEach(e -> {
+            //表名，注释
+            List tValues = new ArrayList(e.values());
+            //第一个总是表名
+            res.add((String) tValues.get(0));
+        });
+        return res;
+    }
 }
diff --git a/datax-web/src/main/java/com/wugui/tool/query/QueryToolInterface.java b/datax-web/src/main/java/com/wugui/tool/query/QueryToolInterface.java
index e18a9d9d..2efee159 100644
--- a/datax-web/src/main/java/com/wugui/tool/query/QueryToolInterface.java
+++ b/datax-web/src/main/java/com/wugui/tool/query/QueryToolInterface.java
@@ -52,4 +52,12 @@ public interface QueryToolInterface {
      * @return2
      */
     public List<String> getColumnNames(String tableName);
+
+
+    /**
+     * 获取所有可用表名
+     *
+     * @return2
+     */
+    public List<String> getTableNames();
 }
diff --git a/datax-web/src/test/java/com/wugui/tool/query/MySQLQueryToolTest.java b/datax-web/src/test/java/com/wugui/tool/query/MySQLQueryToolTest.java
index b87ce729..20e3f86a 100644
--- a/datax-web/src/test/java/com/wugui/tool/query/MySQLQueryToolTest.java
+++ b/datax-web/src/test/java/com/wugui/tool/query/MySQLQueryToolTest.java
@@ -55,4 +55,10 @@ public class MySQLQueryToolTest {
         List<String> columns = queryTool.getColumnNames("datax_plugin");
         log.info(columns.toString());
     }
+
+    @Test
+    public void getTableNames() {
+        List<String> tableNames = queryTool.getTableNames();
+        tableNames.forEach(System.out::println);
+    }
 }
\ No newline at end of file