Merge branch 'master' of https://gitee.com/leshalv/screw

2021-06-06 17:31:46 +08:00
parent 9bcae7a487 a0666bc959
commit 4a8c84f97b
10 changed files with 724 additions and 8 deletions
--- a/screw-core/src/main/java/cn/smallbun/screw/core/mapping/Mapping.java
+++ b/screw-core/src/main/java/cn/smallbun/screw/core/mapping/Mapping.java
@@ -120,6 +120,48 @@ public class Mapping {
        return list;
    }

+    /**
+     * @param resultSet {@link ResultSet} 对象
+     * @param clazz     领域类型
+     * @param <T>       领域泛型
+     * @return 领域对象
+     * @throws MappingException MappingException
+     */
+    public static <T> List<T> convertListByColumnLabel(ResultSet resultSet,
+                                          Class<T> clazz) throws MappingException {
+        //存放列名和结果
+        List<Map<String, Object>> values = new ArrayList<>(16);
+        //结果集合
+        List<T> list = new ArrayList<>();
+        try {
+            //处理 ResultSet
+            ResultSetMetaData metaData = resultSet.getMetaData();
+            int columnCount = metaData.getColumnCount();
+            //迭代
+            while (resultSet.next()) {
+                //map object
+                HashMap<String, Object> value = new HashMap<>(16);
+                //循环所有的列，获取列名，根据列名获取值
+                for (int i = 1; i <= columnCount; i++) {
+                    String columnName = metaData.getColumnLabel(i);
+                    value.put(columnName, resultSet.getString(i));
+                }
+                //add object
+                values.add(value);
+            }
+            //获取类数据
+            List<FieldMethod> fieldMethods = getFieldMethods(clazz);
+            //循环集合，根据类型反射构建对象
+            for (Map<String, Object> map : values) {
+                T rsp = getObject(clazz, fieldMethods, map);
+                list.add(rsp);
+            }
+        } catch (Exception e) {
+            throw new MappingException(e);
+        }
+        return list;
+    }
+
    /**
     * 获取对象
     *
--- a/screw-core/src/main/java/cn/smallbun/screw/core/query/AbstractDatabaseQuery.java
+++ b/screw-core/src/main/java/cn/smallbun/screw/core/query/AbstractDatabaseQuery.java
@@ -72,6 +72,7 @@ public abstract class AbstractDatabaseQuery implements DatabaseQuery {
        try {
            //不为空
            if (!Objects.isNull(connection) && !connection.isClosed()) {
+                System.out.println(connection);
                return connection;
            }
            //同步代码块
@@ -115,7 +116,7 @@ public abstract class AbstractDatabaseQuery implements DatabaseQuery {
        try {
            String schema;
            //获取数据库URL 用于判断数据库类型
-            String url = this.getDataSource().getConnection().getMetaData().getURL();
+            String url = this.getConnection().getMetaData().getURL();
            //获取数据库名称
            String name = JdbcUtils.getDbType(url).getName();
            if (DatabaseType.CACHEDB.getName().equals(name)) {
--- a/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/Db2DataBaseQuery.java
+++ b/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/Db2DataBaseQuery.java
@@ -18,18 +18,36 @@
 package cn.smallbun.screw.core.query.db2;

 import cn.smallbun.screw.core.exception.QueryException;
+import cn.smallbun.screw.core.mapping.Mapping;
 import cn.smallbun.screw.core.metadata.Column;
 import cn.smallbun.screw.core.metadata.Database;
 import cn.smallbun.screw.core.metadata.PrimaryKey;
 import cn.smallbun.screw.core.metadata.Table;
 import cn.smallbun.screw.core.query.AbstractDatabaseQuery;
+import cn.smallbun.screw.core.query.db2.model.Db2ColumnModel;
+import cn.smallbun.screw.core.query.db2.model.Db2DatabaseModel;
+import cn.smallbun.screw.core.query.db2.model.Db2PrimaryKeyModel;
+import cn.smallbun.screw.core.query.db2.model.Db2TableModel;
+import cn.smallbun.screw.core.query.mysql.model.MySqlTableModel;
+import cn.smallbun.screw.core.query.oracle.model.OracleColumnModel;
+import cn.smallbun.screw.core.query.oracle.model.OraclePrimaryKeyModel;
+import cn.smallbun.screw.core.query.oracle.model.OracleTableModel;
 import cn.smallbun.screw.core.util.Assert;
+import cn.smallbun.screw.core.util.CollectionUtils;
 import cn.smallbun.screw.core.util.ExceptionUtils;
+import cn.smallbun.screw.core.util.JdbcUtils;
+import org.apache.commons.lang3.StringUtils;

 import javax.sql.DataSource;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;

 import static cn.smallbun.screw.core.constant.DefaultConstants.NOT_SUPPORTED;
+import static cn.smallbun.screw.core.constant.DefaultConstants.PERCENT_SIGN;

 /**
 * db2 数据库查询
@@ -54,7 +72,9 @@ public class Db2DataBaseQuery extends AbstractDatabaseQuery {
     */
    @Override
    public Database getDataBase() throws QueryException {
-        throw ExceptionUtils.mpe(NOT_SUPPORTED);
+        Db2DatabaseModel model = new Db2DatabaseModel();
+        model.setDatabase(StringUtils.trim(this.getSchema()));
+        return  model;
    }

    /**
@@ -63,8 +83,19 @@ public class Db2DataBaseQuery extends AbstractDatabaseQuery {
     * @return {@link List} 所有表信息
     */
    @Override
-    public List<Table> getTables() {
-        throw ExceptionUtils.mpe(NOT_SUPPORTED);
+    public List<Db2TableModel> getTables() {
+        ResultSet resultSet = null;
+        try {
+            //查询
+            resultSet = getMetaData().getTables(getCatalog(), getSchema(), PERCENT_SIGN,
+                    new String[] { "TABLE" });
+            //映射
+            return Mapping.convertList(resultSet, Db2TableModel.class);
+        } catch (SQLException e) {
+            throw ExceptionUtils.mpe(e);
+        } finally {
+            JdbcUtils.close(resultSet,connection);
+        }
    }

    /**
@@ -75,9 +106,59 @@ public class Db2DataBaseQuery extends AbstractDatabaseQuery {
     * @throws QueryException QueryException
     */
    @Override
-    public List<Column> getTableColumns(String table) throws QueryException {
+    public List<Db2ColumnModel> getTableColumns(String table) throws QueryException {
        Assert.notEmpty(table, "Table name can not be empty!");
-        throw ExceptionUtils.mpe(NOT_SUPPORTED);
+        ResultSet resultSet = null;
+        try {
+            //查询所有
+            List<Db2ColumnModel> list = null;
+            if(PERCENT_SIGN.equals(table)){
+                List<String> tableNames = getTables().stream().map(Table::getTableName)
+                        .collect(Collectors.toList()).stream().distinct().collect(Collectors.toList());
+                //db2 getMetaData().getColumns 中 无法获取表名,循环单张表获取
+                list = new ArrayList<>();
+                for (String tableName : tableNames) {
+                    list.addAll(getTableColumns(tableName));
+                }
+            }else {
+                if(!this.columnsCaching.containsKey(table)){
+                    //查询
+                    resultSet = getMetaData().getColumns(getCatalog(), getSchema(), table, PERCENT_SIGN);
+                    //映射
+                    list = Mapping.convertList(resultSet, Db2ColumnModel.class);
+                    //单表查询
+                    String sql = "SELECT COLNAME as NAME,TABNAME as TABLE_NAME,REMARKS,LENGTH as COLUMN_LENGTH, TYPENAME ||'('|| LENGTH ||')'  as COLUMN_TYPE,SCALE as DECIMAL_DIGITS   FROM SYSCAT.COLUMNS WHERE TABSCHEMA='"+StringUtils.trim(getSchema())+"' and TABNAME = '%s' ORDER BY COLNO";
+                    resultSet = prepareStatement(String.format(sql, table)).executeQuery();
+                    //db2 ColumnName 获取的不是 as column 值,使用ColumnLabel获取
+                    List<Db2ColumnModel> inquires = Mapping.convertListByColumnLabel(resultSet, Db2ColumnModel.class);
+                    for (Db2ColumnModel i : list) {
+                        inquires.forEach(j -> {
+                            //列名一致
+                            if (i.getColumnName().equals(j.getColumnName())) {
+                                //放入备注
+                                i.setRemarks(j.getRemarks());
+                                i.setColumnLength(j.getColumnLength());
+                                i.setColumnType(j.getColumnType());
+                                i.setTableName(j.getTableName());
+                                i.setDecimalDigits(j.getDecimalDigits());
+                                if("NO".equals(i.getNullable())){
+                                    i.setNullable("0");
+                                }
+                            }
+                        });
+                    }
+                    this.columnsCaching.put(table,list.stream()
+                            .filter(i ->  StringUtils.isNotBlank(i.getColumnName())).collect(Collectors.toList()));
+                }else{
+                    list =  this.columnsCaching.get(table).stream().map(c -> (Db2ColumnModel)c).collect(Collectors.toList());
+                }
+            }
+            return list;
+        } catch (SQLException e) {
+            throw ExceptionUtils.mpe(e);
+        } finally {
+            JdbcUtils.close(resultSet, this.connection);
+        }
    }

    /**
@@ -88,7 +169,7 @@ public class Db2DataBaseQuery extends AbstractDatabaseQuery {
     */
    @Override
    public List<? extends Column> getTableColumns() throws QueryException {
-        throw ExceptionUtils.mpe(NOT_SUPPORTED);
+        return getTableColumns(PERCENT_SIGN);
    }

    /**
@@ -100,7 +181,39 @@ public class Db2DataBaseQuery extends AbstractDatabaseQuery {
     */
    @Override
    public List<? extends PrimaryKey> getPrimaryKeys(String table) throws QueryException {
-        throw ExceptionUtils.mpe(NOT_SUPPORTED);
+        ResultSet resultSet = null;
+        try {
+            //查询
+            resultSet = getMetaData().getPrimaryKeys(getCatalog(), getSchema(), table);
+            //映射
+            return Mapping.convertList(resultSet, Db2PrimaryKeyModel.class);
+        } catch (SQLException e) {
+            throw ExceptionUtils.mpe(e);
+        } finally {
+            JdbcUtils.close(resultSet, this.connection);
+        }
    }

+    /**
+     * 根据表名获取主键
+     *
+     * @return {@link List}
+     * @throws QueryException QueryException
+     */
+    @Override
+    public List<? extends PrimaryKey> getPrimaryKeys() throws QueryException {
+        ResultSet resultSet = null;
+        try {
+            // 由于单条循环查询存在性能问题，所以这里通过自定义SQL查询数据库主键信息
+            String sql = "SELECT TABSCHEMA as TABLE_SCHEM, TABNAME as TABLE_NAME ,COLNAME as COLUMN_NAME,KEYSEQ as KEY_SEQ   FROM SYSCAT.COLUMNS WHERE TABSCHEMA = '%s' AND KEYSEQ IS NOT NULL ORDER BY KEYSEQ";
+            resultSet = prepareStatement(String.format(sql, getDataBase().getDatabase())).executeQuery();
+            return Mapping.convertListByColumnLabel(resultSet, Db2PrimaryKeyModel.class);
+        } catch (SQLException e) {
+            throw new QueryException(e);
+        } finally {
+            JdbcUtils.close(resultSet);
+        }
+    }
+
+
 }
--- a/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/model/Db2ColumnModel.java
+++ b/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/model/Db2ColumnModel.java
@@ -0,0 +1,175 @@
+/*
+ * screw-core - 简洁好用的数据库表结构文档生成工具
+ * Copyright © 2020 SanLi (qinggang.zuo@gmail.com)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package cn.smallbun.screw.core.query.db2.model;
+
+import cn.smallbun.screw.core.mapping.MappingField;
+import cn.smallbun.screw.core.metadata.Column;
+import lombok.Data;
+
+/**
+ * db2 table column
+ *
+ * @author hexw
+ * Created by 964104818@qq.com on 2021/05/31 8:44
+ */
+@Data
+public class Db2ColumnModel implements Column {
+    /**
+     * 表名称，它是引用属性的作用域（如果 DATA_TYPE 不是 REF，则为 null）
+     */
+    @MappingField(value = "SCOPE_TABLE")
+    private Object scopeTable;
+    /**
+     * 表类别
+     */
+    @MappingField(value = "TABLE_CAT")
+    private Object tableCat;
+    /**
+     * 未被使用
+     */
+    @MappingField(value = "BUFFER_LENGTH")
+    private String bufferLength;
+    /**
+     * ISO 规则用于确定列是否包括 null。
+     */
+    @MappingField(value = "IS_NULLABLE")
+    private String isNullable;
+    /**
+     * 数据源依赖的类型名称，对于 UDT，该类型名称是完全限定的
+     */
+    @MappingField(value = "TABLE_NAME")
+    private String tableName;
+    /**
+     * 该列的默认值，当值在单引号内时应被解释为一个字符串（可为 null）
+     */
+    @MappingField(value = "COLUMN_DEF")
+    private String columnDef;
+    /**
+     * 表的类别，它是引用属性的作用域（如果 DATA_TYPE 不是 REF，则为 null）
+     */
+    @MappingField(value = "SCOPE_CATALOG")
+    private Object scopeCatalog;
+    /**
+     * 表模式
+     */
+    @MappingField(value = "TABLE_SCHEM")
+    private String tableSchem;
+    /**
+     * 列名称
+     */
+    @MappingField(value = "NAME")
+    private String columnName;
+    /**
+     * 是否允许使用 NULL。
+     */
+    @MappingField(value = "NULLABLE")
+    private String nullable;
+    /**
+     * 描述列的注释（可为 null）
+     */
+    @MappingField(value = "REMARKS")
+    private String remarks;
+    /**
+     * 小数部分的位数。对于 DECIMAL_DIGITS 不适用的数据类型，则返回 Null。
+     */
+    @MappingField(value = "DECIMAL_DIGITS")
+    private String decimalDigits;
+    /**
+     * 基数（通常为 10 或 2）
+     */
+    @MappingField(value = "NUM_PREC_RADIX")
+    private String numPrecRadix;
+    /**
+     *
+     */
+    @MappingField(value = "SQL_DATETIME_SUB")
+    private String sqlDatetimeSub;
+    /**
+     *
+     */
+    @MappingField(value = "IS_GENERATEDCOLUMN")
+    private String isGeneratedColumn;
+    /**
+     * 指示此列是否自动增加
+     * YES --- 如果该列自动增加
+     * NO --- 如果该列不自动增加
+     */
+    @MappingField(value = "IS_AUTOINCREMENT")
+    private String isAutoIncrement;
+    /**
+     * SQL数据类型
+     */
+    @MappingField(value = "SQL_DATA_TYPE")
+    private String sqlDataType;
+    /**
+     * 对于 char 类型，该长度是列中的最大字节数
+     */
+    @MappingField(value = "CHAR_OCTET_LENGTH")
+    private String charOctetLength;
+    /**
+     * 表中的列的索引（从 1 开始）
+     */
+    @MappingField(value = "ORDINAL_POSITION")
+    private String ordinalPosition;
+    /**
+     * 表的模式，它是引用属性的作用域（如果 DATA_TYPE 不是 REF，则为 null）
+     */
+    @MappingField(value = "SCOPE_SCHEMA")
+    private String scopeSchema;
+    /**
+     * 不同类型或用户生成 Ref 类型、来自 java.sql.Types 的 SQL 类型的源类型（如果 DATA_TYPE 不是 DISTINCT 或用户生成的 REF，则为 null）
+     */
+    @MappingField(value = "SOURCE_DATA_TYPE")
+    private String sourceDataType;
+    /**
+     * 来自 java.sql.Types 的 SQL 类型
+     */
+    @MappingField(value = "DATA_TYPE")
+    private String dataType;
+    /**
+     * 数据源依赖的类型名称，对于 UDT，该类型名称是完全限定的
+     */
+    @MappingField(value = "TYPE_NAME")
+    private String typeName;
+    /**
+     * 列表示给定列的指定列大小。
+     * 对于数值数据，这是最大精度。
+     * 对于字符数据，这是字符长度。
+     * 对于日期时间数据类型，这是 String 表示形式的字符长度（假定允许的最大小数秒组件的精度）。
+     * 对于二进制数据，这是字节长度。
+     * 对于 ROWID 数据类型，这是字节长度。对于列大小不适用的数据类型，则返回 Null。
+     */
+    @MappingField(value = "COLUMN_SIZE")
+    private String columnSize;
+
+    /**
+     * 是否主键
+     */
+    private String primaryKey;
+    /**
+     * 列类型（带长度）
+     */
+    @MappingField(value = "COLUMN_TYPE")
+    private String columnType;
+
+    /**
+     * 列长度
+     */
+    @MappingField(value = "COLUMN_LENGTH")
+    private String columnLength;
+}
--- a/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/model/Db2DatabaseModel.java
+++ b/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/model/Db2DatabaseModel.java
@@ -0,0 +1,39 @@
+/*
+ * screw-core - 简洁好用的数据库表结构文档生成工具
+ * Copyright © 2020 SanLi (qinggang.zuo@gmail.com)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package cn.smallbun.screw.core.query.db2.model;
+
+import cn.smallbun.screw.core.mapping.MappingField;
+import cn.smallbun.screw.core.metadata.Database;
+import lombok.Data;
+
+/**
+ * db2数据库信息
+ *
+ * @author hexw
+ * Created by 964104818@qq.com on 2021/05/31 8:44
+ */
+@Data
+public class Db2DatabaseModel implements Database {
+
+    private static final long serialVersionUID = 931210775266917894L;
+    /**
+     * 数据库名称
+     */
+    @MappingField(value = "TABLE_CAT")
+    private String            database;
+}
--- a/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/model/Db2PrimaryKeyModel.java
+++ b/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/model/Db2PrimaryKeyModel.java
@@ -0,0 +1,58 @@
+/*
+ * screw-core - 简洁好用的数据库表结构文档生成工具
+ * Copyright © 2020 SanLi (qinggang.zuo@gmail.com)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package cn.smallbun.screw.core.query.db2.model;
+
+import cn.smallbun.screw.core.mapping.MappingField;
+import cn.smallbun.screw.core.metadata.PrimaryKey;
+import lombok.Data;
+
+/**
+ * db2 table primary
+ *
+ * @author hexw
+ * Created by 964104818@qq.com on 2021/05/31 8:44
+ */
+@Data
+public class Db2PrimaryKeyModel implements PrimaryKey {
+
+    /**
+     * 表名
+     */
+    @MappingField(value = "TABLE_NAME")
+    private String tableName;
+    /**
+     * pk name
+     */
+    @MappingField(value = "PK_NAME")
+    private String pkName;
+    /**
+     * 表模式
+     */
+    @MappingField(value = "TABLE_SCHEM")
+    private String tableSchem;
+    /**
+     * 列名
+     */
+    @MappingField(value = "COLUMN_NAME")
+    private String columnName;
+    /**
+     * 键序列
+     */
+    @MappingField(value = "KEY_SEQ")
+    private String keySeq;
+}
--- a/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/model/Db2TableModel.java
+++ b/screw-core/src/main/java/cn/smallbun/screw/core/query/db2/model/Db2TableModel.java
@@ -0,0 +1,57 @@
+/*
+ * screw-core - 简洁好用的数据库表结构文档生成工具
+ * Copyright © 2020 SanLi (qinggang.zuo@gmail.com)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package cn.smallbun.screw.core.query.db2.model;
+
+import cn.smallbun.screw.core.mapping.MappingField;
+import cn.smallbun.screw.core.metadata.Table;
+import lombok.Data;
+
+/**
+ * db2 table
+ *
+ * @author hexw
+ * Created by 964104818@qq.com on 2021/05/31 8:44
+ */
+@Data
+public class Db2TableModel implements Table {
+    /**
+     * TABLE_CAT
+     */
+    @MappingField(value = "TABLE_CAT")
+    private String tableCat;
+    /**
+     * 表名
+     */
+    @MappingField(value = "NAME")
+    private String tableName;
+    /**
+     * 表模式
+     */
+    @MappingField(value = "TABLE_SCHEM")
+    private String tableSchem;
+    /**
+     * 表类型
+     */
+    @MappingField(value = "TABLE_TYPE")
+    private String tableType;
+    /**
+     * 备注
+     */
+    @MappingField(value = "REMARKS")
+    private String remarks;
+}
--- a/screw-core/src/main/resources/properties/db2.properties
+++ b/screw-core/src/main/resources/properties/db2.properties
@@ -0,0 +1,26 @@
+#
+# screw-core - \u7B80\u6D01\u597D\u7528\u7684\u6570\u636E\u5E93\u8868\u7ED3\u6784\u6587\u6863\u751F\u6210\u5DE5\u5177
+# Copyright \u00A9 2020 SanLi (qinggang.zuo@gmail.com)
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+#URL
+url=jdbc:db2://xxx.xxx.xxx.xxx:50000/xxx:currentSchema=xxx;
+#\u9A71\u52A8
+driver=com.ibm.db2.jcc.DB2Driver
+#\u7528\u6237\u540D
+username=xxx
+#\u5BC6\u7801
+password=xxx
--- a/screw-core/src/test/java/cn/smallbun/screw/core/produce/Db2DocumentationBuilderTest.java
+++ b/screw-core/src/test/java/cn/smallbun/screw/core/produce/Db2DocumentationBuilderTest.java
@@ -0,0 +1,86 @@
+/*
+ * screw-core - 简洁好用的数据库表结构文档生成工具
+ * Copyright © 2020 SanLi (qinggang.zuo@gmail.com)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package cn.smallbun.screw.core.produce;
+
+import cn.smallbun.screw.core.Configuration;
+import cn.smallbun.screw.core.engine.EngineConfig;
+import cn.smallbun.screw.core.engine.EngineFileType;
+import cn.smallbun.screw.core.engine.EngineTemplateType;
+import com.zaxxer.hikari.HikariConfig;
+import com.zaxxer.hikari.HikariDataSource;
+import org.junit.jupiter.api.Test;
+
+import java.io.IOException;
+import java.util.Properties;
+
+import static cn.smallbun.screw.core.common.Constants.fileOutputDir;
+
+/**
+ * SqlServer 文档生成测试
+ *
+ * @author SanLi
+ * Created by qinggang.zuo@gmail.com / 2689170096@qq.com on 2020/3/30 18:33
+ */
+public class Db2DocumentationBuilderTest extends AbstractDocumentationExecute {
+
+    /**
+     * 构建
+     */
+    @Test
+    void build() throws IOException {
+        //数据源
+        HikariConfig hikariConfig = new HikariConfig();
+        hikariConfig.setDriverClassName(getDriver());
+        hikariConfig.setJdbcUrl(getUrl());
+        hikariConfig.setUsername(getUserName());
+        hikariConfig.setPassword(getPassword());
+        hikariConfig.setMinimumIdle(2);
+        hikariConfig.setMaximumPoolSize(5);
+        HikariDataSource dataSource = new HikariDataSource(hikariConfig);
+        //生成配置
+        EngineConfig engineConfig = EngineConfig.builder()
+            //生成文件路径
+            .fileOutputDir(fileOutputDir)
+            //文件类型
+            .fileType(EngineFileType.MD)
+            //打开文件夹
+            .openOutputDir(true)
+            //生成模板实现
+            .produceType(EngineTemplateType.velocity).build();
+        //配置
+        Configuration config = Configuration.builder()
+            //版本
+            .version("1.0.0")
+            //描述
+            .description("数据库设计文档生成")
+            //数据源
+            .dataSource(dataSource)
+            //生成配置
+            .engineConfig(engineConfig).build();
+        execute(config);
+    }
+
+    /**
+     * 获取配置文件
+     * @return {@link Properties}
+     */
+    @Override
+    public String getConfigProperties() {
+        return System.getProperty("user.dir") + "/src/main/resources/properties/db2.properties";
+    }
+}
--- a/screw-core/src/test/java/cn/smallbun/screw/core/querys/Db2DataBaseQueryTest.java
+++ b/screw-core/src/test/java/cn/smallbun/screw/core/querys/Db2DataBaseQueryTest.java
@@ -0,0 +1,119 @@
+/*
+ * screw-core - 简洁好用的数据库表结构文档生成工具
+ * Copyright © 2020 SanLi (qinggang.zuo@gmail.com)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package cn.smallbun.screw.core.querys;
+
+import cn.smallbun.screw.core.common.Properties;
+import cn.smallbun.screw.core.exception.QueryException;
+import cn.smallbun.screw.core.metadata.Column;
+import cn.smallbun.screw.core.metadata.Database;
+import cn.smallbun.screw.core.metadata.PrimaryKey;
+import cn.smallbun.screw.core.metadata.Table;
+import cn.smallbun.screw.core.query.DatabaseQuery;
+import cn.smallbun.screw.core.query.DatabaseQueryFactory;
+import com.alibaba.fastjson.JSON;
+import com.zaxxer.hikari.HikariConfig;
+import com.zaxxer.hikari.HikariDataSource;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import javax.sql.DataSource;
+import java.io.IOException;
+import java.util.List;
+import java.util.UUID;
+
+/**
+ * db2 database test
+ *
+ * @author SanLi
+ * Created by qinggang.zuo@gmail.com / 2689170096@qq.com on 2020/3/25 16:59
+ */
+public class Db2DataBaseQueryTest implements Properties {
+    /**
+     * 日志
+     */
+    private final Logger  logger = LoggerFactory.getLogger(this.getClass());
+    /**
+     * DataBaseQuery
+     */
+    private DatabaseQuery query;
+
+    /**
+     * before
+     */
+    @BeforeEach
+    void before() throws IOException {
+        HikariConfig config = new HikariConfig();
+        config.setDriverClassName(getDriver());
+        config.setJdbcUrl(getUrl());
+        config.setUsername(getUserName());
+        config.setPassword(getPassword());
+        //config.addDataSourceProperty("remarks", "false");
+
+        config.setPoolName(UUID.randomUUID().toString().replace("-", ""));
+        DataSource dataSource = new HikariDataSource(config);
+        query = new DatabaseQueryFactory(dataSource).newInstance();
+    }
+
+    /**
+     * 查询数据库
+     */
+    @Test
+    void databases() throws QueryException {
+        Database dataBasesInfo = query.getDataBase();
+        logger.info(JSON.toJSONString(dataBasesInfo, true));
+    }
+
+    /**
+     * 查询表
+     */
+    @Test
+    void tables() throws QueryException {
+        List<? extends Table> tables = query.getTables();
+        logger.info(JSON.toJSONString(tables, true));
+    }
+
+    /**
+     * 查询列
+     */
+    @Test
+    void column() throws QueryException {
+        List<? extends Column> columns = query.getTableColumns();
+        logger.info("{}", columns.size());
+        logger.info(JSON.toJSONString(columns, true));
+    }
+
+    /**
+     * 查询主键
+     */
+    @Test
+    void primaryKeys() {
+        List<? extends PrimaryKey> primaryKeys = query.getPrimaryKeys();
+        logger.info(JSON.toJSONString(primaryKeys, true));
+    }
+
+    /**
+     * 获取配置文件
+     * @return {@link java.util.Properties}
+     */
+    @Override
+    public String getConfigProperties() {
+        return System.getProperty("user.dir") + "/src/main/resources/properties/db2.properties";
+    }
+}