MySql模型生成类

package com.digital.web.front; /**
 * pom依赖
 * 
 * mysql
 * mysql-connector-java
 * 5.1.27
 * 
 * 
 * org.projectlombok
 * lombok
 * 1.16.10
 * 
 */


import com.alibaba.fastjson.JSONObject;
import com.mchange.v2.collection.MapEntry;
import lombok.Data;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.sql.*;
import java.util.*;

/**
 * 自定义框架
 * 生成mysql表对应的实体类
 */
public class MySqlMapper {
    /**
     * 配制参数
     */
    static class Config {//mysql地址,数据库,用户名,密码
        final static private String address = "127.0.0.1:3306";
        final static private String dbName = "henan";
        final static private String username = "root";
        final static private String password = "111QSC3RDXqe@$%";
        //模型保存的位置
        final static private String modelPath = "G:\\IdeaProject\\中文\\src\\main\\java\\com\\digital\\web\\front\\model";
        //mysql数据类型与java数据类型转换
        final static MapEntry[] map = {
                new MapEntry("INT", "Integer"),
                new MapEntry("VARCHAR", "String"),
                new MapEntry("TIMESTAMP", "String"),
                new MapEntry("DOUBLE", "Double")
        };
    }

    /**
     * 内存参数
     */
    static class Variable {
        final static private String packagePath = Config.modelPath.split("java\\\\")[1].replace("\\", ".");
        static private Map mapper = new HashMap<>();

        static {
            for (MapEntry mapEntry : Config.map) {
                mapper.put(String.valueOf(mapEntry.getKey()), String.valueOf(mapEntry.getValue()));
            }
        }

    }

    /**
     * 获取连接
     *
     * @return
     * @throws SQLException
     * @throws ClassNotFoundException
     */
    public static Connection getConn() throws SQLException, ClassNotFoundException {
        Class.forName("com.mysql.jdbc.Driver");
        String url = "jdbc:mysql://" + Config.address + "/" + Config.dbName + "?useUnicode=true&characterEncoding=UTF-8";
        Connection conn = DriverManager.getConnection(url, Config.username, Config.password);
        return conn;
    }


    public static void main(String[] args) throws Exception {
        Connection conn = getConn();
        List modelTableList = Business.getModelTable(conn);//获取数据库下的表结构
        conn.close();
        Business.foreach(modelTableList);//遍历结构数据


    }

    static class Business {//业务类


        /**
         * 获取数据库中表与字段的集合
         *
         * @param conn
         * @return
         */
        public static List getModelTable(Connection conn) throws Exception {
            List modelTableList = new LinkedList<>();
            Map tableMsg = Business.getTableNameList(conn);
            for (Map.Entry line : tableMsg.entrySet()) {
                ModelTable tableFieldMap = getTableFieldMap(conn, line);
                modelTableList.add(tableFieldMap);
            }
            return modelTableList;
        }


        /**
         * 获取表名称
         *
         * @param conn
         * @return
         * @throws Exception
         */
        public static Map getTableNameList(Connection conn) throws Exception {
            String sql = "select TABLE_NAME,TABLE_COMMENT from information_schema.tables where table_schema=\"henan\"";
            List> tableMap = Util.select(sql, conn);
            Map tableMsg = new HashMap<>();
            for (Map line : tableMap) {
                String tableName = String.valueOf(line.get("TABLE_NAME"));
                String tableComment = String.valueOf(line.get("TABLE_COMMENT"));
                tableMsg.put(tableName, tableComment);
            }
            return tableMsg;
        }

        /**
         * 获取表中的字段信息
         *
         * @param conn
         * @return
         * @throws Exception
         */
        public static ModelTable getTableFieldMap(Connection conn, Map.Entry line) throws Exception {
            DatabaseMetaData meta = conn.getMetaData();
            String tableName = line.getKey();
            String tableComment = line.getValue();
            ResultSet rs = meta.getColumns(null, "%", tableName, "%");
            Map tableFieldMap = new HashMap<>();
            ModelTable modelTable = new ModelTable();
            modelTable.setTableName(tableName);
            modelTable.setFields(tableFieldMap);
            modelTable.setTableComment(tableComment);
            while (rs.next()) {
                String columnName = rs.getString("COLUMN_NAME");
                String dataTypeName = rs.getString("TYPE_NAME");
                String remarks = rs.getString("REMARKS");
                tableFieldMap.put(columnName, new MapEntry(dataTypeName, remarks));//字段名称,字段类型
            }
            return modelTable;
        }

        /**
         * 遍历model信息
         *
         * @param modelTableList
         */
        public static void foreach(List modelTableList) throws IOException {
            for (ModelTable modelTable : modelTableList) {
                writerModelTable(modelTable);
            }
        }

        /**
         * 写入model信息到文件
         *
         * @param modelTable
         */
        public static void writerModelTable(ModelTable modelTable) throws IOException {
            String msg = getMsg(modelTable);//计算需要写入的信息
            String filePath = Config.modelPath + "/" + modelTable.tableName + ".java";
            Util.writer(filePath, msg);
        }

        /**
         * 生成model的信息
         *
         * @param modelTable
         * @return
         */
        public static String getMsg(ModelTable modelTable) {
            String packagePath = "package " + Variable.packagePath + ";";
            String fastjson = "import com.alibaba.fastjson.JSONObject;";
            String lombok = "import lombok.Data;";
            String javaUtil = "import java.util.*;";
            String data = "@Data";
            String headClass = "public class " + modelTable.tableName + "{";
            String tailClass = "}";
            StringBuilder sbr = new StringBuilder();
            sbr.append(packagePath).append("\n").append("\n");
            sbr.append(fastjson).append("\n");
            sbr.append(lombok).append("\n");
            sbr.append(javaUtil).append("\n").append("\n");
            sbr.append("/**").append("\n");//添加类备注
            sbr.append("/*").append(modelTable.tableComment).append("\n");
            sbr.append("*/").append("\n");
            sbr.append(data).append("\n");
            sbr.append(headClass).append("\n");
            //添加成员变量与备注
            for (Map.Entry entry : modelTable.fields.entrySet()) {
                String fieldName = entry.getKey();
                MapEntry value = entry.getValue();
                String fieldType = String.valueOf(value.getKey());
                String remakes = String.valueOf(value.getValue());
                sbr.append("\tprivate ").append(Variable.mapper.get(fieldType)).append(" ").append(fieldName).append(";").append("//").append(remakes).append("\n");
            }
            sbr.append("\n");
            //定义方法
            sbr.append(ModelFunc.getToJson(modelTable)).append("\n");
            sbr.append(ModelFunc.getToMap(modelTable)).append("\n");
            sbr.append(ModelFunc.setToJson(modelTable)).append("\n");
            sbr.append(ModelFunc.setToMap(modelTable)).append("\n");
            sbr.append(ModelFunc.getInsertSql(modelTable)).append("\n");


            //结束
            sbr.append(tailClass).append("\n");
            return sbr.toString();
        }
    }

    static class ModelFunc {//方法构造

        public static String getToJson(ModelTable modelTable) {//将成员变量转化为json对象
            StringBuilder sbr = new StringBuilder();
            sbr.append("\t").append("public JSONObject getToJson(){\n");
            Map fields = modelTable.fields;
            sbr.append("\t\tJSONObject result = new JSONObject();").append("\n");
            for (Map.Entry entry : fields.entrySet()) {
                String fieldName = entry.getKey();
                sbr.append("\t\t").append("result.put(\"").append(fieldName).append("\",").append(fieldName).append(");").append("\n");
            }
            sbr.append("\t\treturn result;\n");
            sbr.append("\t}\n");
            return sbr.toString();
        }

        public static String getToMap(ModelTable modelTable) {//将成员变量转化为Map对象
            StringBuilder sbr = new StringBuilder();
            sbr.append("\t").append("public Map getToMap(){\n");
            sbr.append("\t\t").append("Map result = new HashMap<>();\n");
            for (Map.Entry entry : modelTable.fields.entrySet()) {
                String fieldName = entry.getKey();
                sbr.append("\t\t").append("result.put(\"").append(fieldName).append("\",").append(fieldName).append(");").append("\n");
            }
            sbr.append("\t\treturn result;\n");
            sbr.append("\t}\n");
            return sbr.toString();
        }

        public static String setToJson(ModelTable modelTable) {//json赋值
            StringBuilder sbr = new StringBuilder();
            sbr.append("\t").append("public void setToJson(JSONObject model){\n");
            JSONObject model = null;
            for (Map.Entry entry : modelTable.fields.entrySet()) {
                String fieldName = entry.getKey();
                String fieldType = String.valueOf(entry.getValue().getKey());
                sbr.append("\t\t").append("this.").append(fieldName).append("=model.get").append(Variable.mapper.get(fieldType)).append("(\"").append(fieldName).append("\");").append("\n");
            }
            sbr.append("\t}\n");
            return sbr.toString();
        }

        public static String setToMap(ModelTable modelTable) {//Map赋值
            StringBuilder sbr = new StringBuilder();
            sbr.append("\t").append("public void setToMap(Map model){\n");
            for (Map.Entry entry : modelTable.fields.entrySet()) {
                String fieldName = entry.getKey();
                String fieldType = String.valueOf(entry.getValue().getKey());
                sbr.append("\t\t").append("this.").append(fieldName).append("=").append(Variable.mapper.get(fieldType)).append(".valueOf(model.get(\"").append(fieldName).append("\").toString());").append("\n");
            }
            sbr.append("\t}\n");
            return sbr.toString();
        }

        public static String getInsertSql(ModelTable modelTable) {//获取新增sql
            StringBuilder sbr = new StringBuilder();
            sbr.append("\t").append("public String getInsertSql(){\n");
            StringBuilder keys = new StringBuilder();
            StringBuilder values = new StringBuilder();
            sbr.append("\t\tStringBuilder values = new StringBuilder();").append("\n");
            for (Map.Entry entry : modelTable.fields.entrySet()) {
                String fieldName = entry.getKey();
                keys.append(fieldName).append(",");
                values.append(".append(" +
                        "\"this.").append(fieldName).append("/.").append(");");
            }
            sbr.append("\t\tString sql = values.append(").append("\"insert into ").append(modelTable.tableName).append("(").append(keys.substring(0,keys.length()-1)).append(")value(").append("\")").append(values.substring(0,values.length()-1));
            sbr.append("insert into ").append(modelTable.tableName).append("(").append(keys.substring(0,keys.length()-1)).append(")value(").append(values.substring(0,values.length()-1)).append(")\";\n");
            sbr.append("\treturn sql;");
            sbr.append("\t}\n");
            return sbr.toString();
        }
    }


    static class Util {//工具类

        /**
         * 查询数据
         *
         * @param sql  请求sql
         * @param conn 数据库连接
         * @return 查询的响应数据
         * @throws Exception
         */
        public static List> select(String sql, Connection conn) throws Exception {
            List> result = new LinkedList<>();
            PreparedStatement preparedStatement = null;//支出传递问号,防止sql注入
            ResultSet resultSet = null;
            try {
                preparedStatement = conn.prepareStatement(sql);//获取PreparedStatement
                resultSet = preparedStatement.executeQuery();//获取查询结果的字段信息
                ResultSetMetaData metaData = resultSet.getMetaData();
                int fieldCount = metaData.getColumnCount();//字段数量
                List fields = new ArrayList<>(fieldCount);
                for (int i = 1; i <= fieldCount; i++) {
                    fields.add(metaData.getColumnName(i));
                }
                //5、从结果集获取结果数据
                while (resultSet.next()) {
                    Map line = new HashMap();
                    for (String field : fields) {
                        line.put(field, resultSet.getObject(field));
                    }
                    result.add(line);
                }
            } catch (SQLException e) {
                e.printStackTrace();
            } finally {
                resultSet.close();
                preparedStatement.close();
            }
            return result;
        }

        /**
         * 写入内容到文件
         *
         * @param filePath 文件路径
         * @param msg      写入信息
         * @throws IOException
         */
        public static void writer(String filePath, String msg) throws IOException {
            if (msg == null || "".equals(msg)) return;
            File file = new File(filePath);
            if (!file.exists()) file.createNewFile();
            FileOutputStream fileOutputStream = null;
            try {
                fileOutputStream = new FileOutputStream(file);
                fileOutputStream.write(msg.getBytes("UTF-8"));
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                fileOutputStream.close();//先打开的后关闭
            }
        }

        /**
         * 获取json内容的类型转换
         *
         * @param fieldType
         * @return
         */
        public static String getType(String fieldType) {
            String javaFieldType = Variable.mapper.get(fieldType);
            switch (javaFieldType) {
                case "Integer":
                    return "getInteger";
                case "String":
                case "TIMESTAMP":
                    return "getString";
                case "DOUBLE":
                    return "getDouble";
                default:
                    return "getObject";

            }
        }
    }

    @Data
    static class ModelTable {//数据库中表与字段的集合
        private String tableName;//表名称
        private String tableComment;//表备注
        private Map fields;//字段列表,MapEntry<字段类型,备注>
    }
}