com.base4j.mybatis.sql.template.AbstractSqlTemplate Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of base4j-mybatis Show documentation
base4j封装mybatis包
The newest version!
package com.base4j.mybatis.sql.template;

import com.base4j.mybatis.config.datasource.ApplicationProperties;
import com.base4j.mybatis.sql.entity.EntityTable;
import com.base4j.mybatis.sql.helper.EntityHelper;
import com.base4j.mybatis.tool.NullUtil;
import com.base4j.mybatis.tool.SpringContextHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;
import java.nio.charset.Charset;

/**
 * Created by USER on 2017-03-20.
 */
public abstract class AbstractSqlTemplate {

    private Logger logger = LoggerFactory.getLogger(this.getClass());

    private ApplicationProperties applicationProperties = SpringContextHelper.getBeanById("applicationProperties");

    /**
     * 生成mybatis需要的mapper.xml文件
     *
     * @param mapperClass 实体类
     * @return mapper.xml
     */
    public InputStream buildSqlFile(Class mapperClass) {
        EntityTable entityTable = anayEntity(mapperClass);
        return sqlParse(entityTable);
    }

    /**
     * sql生成模板路
     *
     * @return 模板名称
     */
    public abstract String getSqlTemplateName();

    /**
     * 解析实体，获得对应的表结构
     *
     * @param mapperClass mapper类型
     * @return EntityTable
     */
    public EntityTable anayEntity(Class mapperClass) {
        return EntityHelper.getEntityTableByMapperClass(mapperClass);
    }

    /**
     * 根据表结构生成默认mapper.xml文件。
     *
     * @param entityTable
     * @return InputStream
     */
    private InputStream sqlParse(EntityTable entityTable) {
        String xmlContent = FreeMarkerUtil.process(getSqlTemplateName(), entityTable);
        String genPath = applicationProperties.getMybatisMapperGenPath();
        if (NullUtil.isNotEmpty(genPath)) {
            String xmlName = entityTable.getMapperClass().getSimpleName() + ".xml";
            //生成文件，以备查看
            genXmlFile(genPath, xmlName, xmlContent);
        }
        //加载mapper文件
        InputStream inputStream = new ByteArrayInputStream(xmlContent.getBytes(Charset.forName("UTF-8")));
        return inputStream;
    }

    /**
     * 生成xml文件，以便测试验证
     *
     * @param genPath    xml文件生成路径
     * @param xmlName    xml文件名称
     * @param xmlContent xml文件内容
     * @throws IOException
     */
    private void genXmlFile(String genPath, String xmlName, String xmlContent) {
        logger.debug("生成mybatis mapper调试文件 -> ", "路径:", genPath, ", 名称:", xmlName);
        try {
            if (NullUtil.isNotEmpty(genPath)) {
                File genFolder = new File(genPath);
                if (!genFolder.exists()) {
                    genFolder.mkdirs();
                }
                String xmlPath = genFolder + File.separator + xmlName;
                File xml = new File(xmlPath);
                FileWriter fileWriter = new FileWriter(xml);
                fileWriter.write(xmlContent);
                fileWriter.flush();
                fileWriter.close();
            }
        } catch (IOException e) {
            logger.error("生成mybatis mapper调试文件时出错", e);
        }
    }
}