要“鱼”也要“渔”-java手写代码生成器

小尘哥

发布于 2018-10-18 10:17:03

6270

发布于 2018-10-18 10:17:03

遇到重复工作一定想办法将其转化为自动化，我觉得这才是合格的程序猿。我们日常开发中遇到的80%的代码都是增删改查，当然，已经有一些插件，比如MP提供了一整套的代码生成方案，但是提供的基本都是java代码，因为模板语言有太多种，根据每个人每个项目的不同而不同，但是，模板中一样也基本是列表和表单提交，怎么来解决这个重复工作呢？

思路：依然依赖mp组件，因为mp已经提供了对应的BaseService，BaseMapper等，但是本文重点不在于“怎么用”，而在于“怎么写”。“怎么写”属于自己的代码生成器，方便自己随意扩展前端页面生成。

前提-数据库设计约定：表名、字段名多个单词时下划线分割
读取数据库，解析各个表字段及备注
实体类生成：根据字段名解析为对应实体类属性名，同时注释为属性注释
定义生成模板（xxx.ftl）
前端页面：数据库字段备注为表单中的字段名，根据字段类型设置输入项的html标签
分别生成实体类、Mapper、Service、前端页面等
替换（xxx.ftl）中的变量

一、TableInfo对象

一、JDBC读取数据库表结构

public Map<String, TableInfo> getTables() {
        try {
            DatabaseMetaData databaseMetaData = getConn().getMetaData();

            ResultSet resultSet = databaseMetaData.getColumns(null, "%", "%", "%");

            Map<String, TableInfo> tableInfoMap = new HashMap<>(8);
            while (resultSet.next()) {
                String tableName = resultSet.getString("TABLE_NAME");
                TableInfo tableInfo = tableInfoMap.get(tableName);
                if (tableInfo == null) {
                    tableInfo = new TableInfo(tableName, StringUtil.toHump(tableName,true));
                }

                String colName = resultSet.getString("COLUMN_NAME");
                String remark = resultSet.getString("REMARKS");
                String colType = resultSet.getString("TYPE_NAME");
                List<ColInfo> colInfos = tableInfo.getColInfos();
                if (colInfos == null || colInfos.size() == 0) {
                    colInfos = new ArrayList<>();
                }

                ColInfo colInfo = new ColInfo(colName, colType, remark);
                colInfos.add(colInfo);
                tableInfo.setColInfos(colInfos);

                tableInfoMap.put(tableName, tableInfo);
            }

            return tableInfoMap;
        } catch (SQLException e) {
            e.printStackTrace();
        }

        return null;
    }

二、生成实体类

dataMap 存放渲染ftl模板时的数据。

public void generateModel(String diskPath, TableInfo tableInfo) {
        String templateName = "model.ftl";
        String path = diskPath + File.separator + tableInfo.getModelName() + TableInfo.GEN_SUFFIX;
        Map<String, Object> dataMap = new HashMap<>();
        dataMap.put("ModelName", tableInfo.getModelName());
        dataMap.put("packageName", "com.mos.aaa");
        List<ColInfo> colInfos = tableInfo.getColInfos();
        dataMap.put("fields", colInfos);
        List<String> pcks = new ArrayList<>();
        for (ColInfo info : colInfos){
            if (StringUtil.isNotBlank(info.getFieldType().getPkg()) && !pcks.contains(info.getFieldType().getPkg())){
                pcks.add(info.getFieldType().getPkg());
            }
        }
        dataMap.put("pcks", pcks);
        generateFileByTemplate(templateName, path, dataMap);

    }

三、Model模板定义（用来生成Entity）

package ${packageName};

import com.baomidou.mybatisplus.activerecord.Model;
import java.io.Serializable;
<#if pcks??>
<#list pcks as pck>
import ${pck!};
</#list>
</#if>

public class ${ModelName} extends Model<${ModelName}> {
    private static final long serialVersionUID = 1L;

<#list fields as field>
    /**
    * ${field.remark!}
    */
    private ${field.fieldType.type} ${field.changedColNameLower};
</#list>

<#list fields as field>

    public ${field.fieldType.type} get${field.changedColName}(){
        return ${field.changedColNameLower};
    }

    public void set${field.changedColName}(${field.fieldType.type} ${field.changedColNameLower}){
        this.${field.changedColNameLower} = ${field.changedColNameLower};
    }
</#list>

    @Override
    protected Serializable pkVal() {
        return this.id;
    }

}

四、最终文件生成

private void generateFileByTemplate(final String templateName, String genPath, Map<String, Object> dataMap) {
        try {
            Template template = FreeMarkerTemplateUtils.getTemplate(templateName);
            FileOutputStream fos = new FileOutputStream(new File(genPath));
            Writer out = new BufferedWriter(new OutputStreamWriter(fos, "utf-8"), 10240);
            template.process(dataMap, out);
        } catch (Exception e) {
            System.out.println(e.getMessage());
        }

    }