如何自定义代码生成器（下）

4.2 编码

4.2.1 常量类

package freemarker.constant;

public class TemplateConstant {
   

    //实体类模板
    public static final String entityTemplate = "templates/Entity.ftl";
    //Mapper模板
    public static final String mapperTemplate = "templates/Mapper.ftl";
    //Mapper映射文件模版
    public static final String mapperXmlTemplate = "templates/MapperXml.ftl";
    //service模版
    public static final String serviceTemplate = "templates/Service.ftl";
    //service实现类模版
    public static final String serviceImplTemplate = "templates/ServiceImpl.ftl";
    //controller模版
    public static final String controllerTemplate = "templates/Controller.ftl";
    //vo模版
    public static final String entityVoTemplate = "templates/EntityVo.ftl";
    //dto模版
    public static final String entityDtoTemplate = "templates/EntityDto.ftl";
    //pom模版
    public static final String pomTemplate = "templates/pom.ftl";
    //application.yml模版
    public static final String applicationTemplate = "templates/application.ftl";
}

4.2.2 工具类

（1）DbUtil数据库工具类

package freemarker.util;

import freemarker.param.ColumnClass;
import freemarker.param.TableClass;
import lombok.Getter;
import lombok.Setter;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

/**
 * 数据库工具类
 */
@Setter
@Getter
public class DbUtil {
   

    //数据库连接地址
    private String url = "jdbc:mysql://localhost:3306/heima_leadnews_wemedia?useSSL=false&nullCatalogMeansCurrent=true&serverTimezone=UTC";
    //数据库用户名
    private String username = "root";
    //数据库密码

    private String password = "123456";
    //数据库驱动
    private String driver = "com.mysql.jdbc.Driver";

    //数据库名称
    private String dbName = null;

    private Connection connection =null;

    /**
     * 获取jdbc链接
     * @return
     * @throws Exception
     */
    public Connection getConnection() throws Exception{
   
        Properties pro = new Properties();
        pro.setProperty("user", username);
        pro.setProperty("password", password);
        pro.put("useInformationSchema", "true");//获取mysql表注释
        //pro.setProperty("remarksReporting","true");//获取oracle表注释
        Class.forName(driver);
        connection = DriverManager.getConnection(url, pro);
        return connection;
    }

    /**
     * 获取当前数据库下的所有表名称及注释
     * @return
     * @throws Exception
     */
    public List<TableClass> getAllTables() throws Exception {
   
        List<TableClass> list = new ArrayList<>();
        //获取链接
        Connection conn = getConnection();
        //获取元数据
        DatabaseMetaData metaData = conn.getMetaData();
        //获取所有的数据库表信息
        ResultSet rs = metaData.getTables(dbName!=null?dbName:conn.getCatalog(), "%", "%", new String[]{
   "TABLE"});
        while (rs.next()) {
   
            TableClass tableClass = new TableClass();
            tableClass.setTableName(rs.getString(3));
            tableClass.setTableComment(rs.getString(5));
            list.add(tableClass);
        }
        return list;
    }

    /**
     *  获取某张表的所有列
     * @param tableName
     * @return
     * @throws Exception
     */
    public List<ColumnClass> getAllColumns(String tableName) throws Exception {
   
        List<ColumnClass> list = new ArrayList<>();
        //获取链接
        Connection conn = getConnection();
        //获取元数据
        DatabaseMetaData metaData = conn.getMetaData();
        //获取所有的数据库某张表所有列信息
        ResultSet rs = metaData.getColumns(dbName!=null?dbName:conn.getCatalog(), "%", tableName,"%");

        while(rs.next()) {
   
            ColumnClass columnClass = new ColumnClass();
            columnClass.setColumnName(rs.getString("COLUMN_NAME"));
            columnClass.setColumnType(rs.getString("TYPE_NAME"));
            columnClass.setColumnComment(rs.getString("REMARKS"));
            columnClass.setChangeColumnName(StrUtil.changeColumnStr(rs.getString("COLUMN_NAME")));
            list.add(columnClass);
        }
        return list;
    }

    /**
     * 关闭链接
     */
    public void closeConnection(){
   
        try {
   
            connection.close();
        } catch (SQLException e) {
   
            e.printStackTrace();
        }
    };

}

（2）字符串处理工具类

package freemarker.util;

/**
 * 字符串处理工具类
 */
public class StrUtil {
   
    /**
     * 去掉下划线转驼峰  user_name  -> userName
     * @param str
     * @return
     */
    public static String changeColumnStr(String str) {
   
        String name = str;
        if (name.indexOf("_") > 0 && name.length() != name.indexOf("_") + 1) {
   
            int lengthPlace = name.indexOf("_");
            name = name.replaceFirst("_", "");
            String s = name.substring(lengthPlace, lengthPlace + 1);
            s = s.toUpperCase();
            str = name.substring(0, lengthPlace) + s + name.substring(lengthPlace + 1);
        } else {
   
            return str;
        }
        return changeColumnStr(str);
    }
    /**
     * 去掉下划线转驼峰  tb_user  -> TbUser
     * @param str
     * @return
     */
    public static String changeTableStr(String str) {
   
        String s = changeColumnStr(str);
        return s.substring(0,1).toUpperCase()+s.substring(1);
    }
}

（3）FreeMarker模版工具类

package freemarker.util;

import freemarker.cache.FileTemplateLoader;
import freemarker.cache.NullCacheStorage;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateExceptionHandler;

import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;

/**
 * FreeMarker模版工具类
 */
public class FreeMarkerTemplateUtils {
   
    private FreeMarkerTemplateUtils(){
   }
    private static final Configuration CONFIGURATION = new Configuration(Configuration.VERSION_2_3_22);
    static{
   
        //ClassTemplateLoader方式：需要将模版放在FreeMarkerTemplateUtils类所在的包，加载模版时会从该包下加载
        //CONFIGURATION.setTemplateLoader(new ClassTemplateLoader(FreeMarkerTemplateUtils.class,""));

        try {
   
            String path = java.net.URLDecoder.decode(FreeMarkerTemplateUtils.class.getClassLoader().getResource("").getPath(),"utf-8");
            //FileTemplateLoader方式：需要将模版放置在classpath目录下 目录有中文也可以
            CONFIGURATION.setTemplateLoader(new FileTemplateLoader(new File(path)));
        } catch (UnsupportedEncodingException e) {
   
            e.printStackTrace();
        } catch (IOException e) {
   
            e.printStackTrace();
        }
        CONFIGURATION.setDefaultEncoding("UTF-8");
        CONFIGURATION.setTemplateExceptionHandler(TemplateExceptionHandler.RETHROW_HANDLER);
        CONFIGURATION.setCacheStorage(NullCacheStorage.INSTANCE);
    }
 
    public static Template getTemplate(String templateName) throws IOException {
   
        try {
   
            return CONFIGURATION.getTemplate(templateName);
        } catch (IOException e) {
   
            throw e;
        }
    }
 
    public static void clearCache() {
   
        CONFIGURATION.clearTemplateCache();
    }
}

4.2.3 实体类

（1）表实体类

package freemarker.param;

import lombok.Data;

@Data
public class TableClass {
   
    /**
     * 表名 tb_user
     **/
    private String tableName;

    /**
     *  表注释
     **/
    private String tableComment;

}

（2）列实体类

package freemarker.param;

import lombok.Data;

@Data
public class ColumnClass {
   
    /**
     * 数据库字段名称 user_name
     **/
    private String columnName;
    /**
     * 数据库字段类型
     **/
    private String columnType;
    /**
     * 数据库字段首字母小写且去掉下划线字符串 userName
     **/
    private String changeColumnName;
    /**
     * 数据库字段注释
     **/
    private String columnComment;

}

（3）模版相关参数类

package freemarker.param;

import lombok.Data