告別編寫基礎的增刪改查代碼

時間  2019-12-12
標籤 告別 編寫 基礎 增刪 代碼 简体版
原文   原文鏈接

 

　　告別編寫基礎的增刪改查代碼java

  

1、 說明web

　　1.本文基本幫你實現不用編寫基礎的增刪改查代碼 注意是「基礎」的，你只須要寫一個實體類，其餘均可以自動生成包括 mapper、dao、service、controller、建表sql。spring

        2.springboot版本：2.0.5 、mybatis：1.3.2 、swagger：2.8.0。sql

　　3.本文代碼目前爲定製化代碼 例如mapper文件命名規範、controller請求路徑等可根據本身項目的風格自行調整。apache

 

2、代碼springboot

　　1.生成mapper文件mybatis

          文件命名規則可根據本身須要更改、默認採用「id」做爲主鍵。app

import java.lang.reflect.Field;

/**
 * mapper.xml生成類
 * 注意：全部實體主鍵假定爲id
 */
public class MapperUtils {
    public static boolean generateMapper(Class clazz){
        StringBuilder sb = new StringBuilder();
        sb.append("<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n")
            .append("<!DOCTYPE mapper PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\" ")
            .append("\"http://mybatis.org/dtd/mybatis-3-mapper.dtd\">\n")
            .append("\n");
        sb.append("<mapper namespace=\"").append(clazz.getSimpleName()).append("\">\n");
        sb.append("\t<resultMap id=\"resultMap\" type=\"").append(clazz.getName()).append("\">\n");
        sb.append("\t\t<id column=\"id\" property=\"id\"/>\n");
        Field[] fields = clazz.getDeclaredFields();
        StringBuilder associations = new StringBuilder();
        /**生成resultMap中result*/
        for (Field field : fields) {
            String fieldName = field.getName();
            if(field.getGenericType().getTypeName().indexOf(".model.") != -1){ //外鍵
                associations.append("\t\t<association property=\"").append(fieldName)
                    .append("\" javaType=\"").append(field.getGenericType().getTypeName()).append("\">\n");
                associations.append("\t\t\t<id column=\"").append(BaseGeneratorUtils.camel2Underline(fieldName)+"_id").append("\" property=\"id\"/>\n");
                associations.append("\t\t</association>\n");
            }else if(!"id".equals(fieldName)){
                sb.append("\t\t<result column=\"").append(BaseGeneratorUtils.camel2Underline(fieldName))
                    .append("\" property=\"").append(fieldName).append("\"/>\n");
            }
        }
        sb.append(associations);
        sb.append("\t</resultMap>\n\n");
        /**生成 sql columns*/
        sb.append("\t<sql id=\"columns\">\n\t\t");
        for (int i = 0; i < fields.length; i++) {
            Field field = fields[i];
            String columnName = BaseGeneratorUtils.getColumn(field);
            sb.append("`").append(columnName).append("`");
            if(i < fields.length-1){
                sb.append(",");
            }
        }
        sb.append("\n\t</sql>\n\n");
        /**生成 sql properties*/
        sb.append("\t<sql id=\"properties\">\n\t\t");
        for (int i = 0; i < fields.length; i++) {
            Field field = fields[i];
            String columnName = BaseGeneratorUtils.getFieldName(field);
            sb.append("#{").append(columnName).append("}");
            if(i < fields.length-1){
                sb.append(",");
            }
        }
        sb.append("\n\t</sql>\n\n");
        
        /**生成insert*/
        sb.append("\t<insert id=\"insert\">\n");
        sb.append("\t\tinsert into ").append(BaseGeneratorUtils.camel2Underline(clazz.getSimpleName()))
            .append(" (<include refid=\"columns\"></include>)\n");
        sb.append("\t\tvalues (<include refid=\"properties\"></include>)\n");
        sb.append("\t</insert>\n\n");
        
        /**生成updateByPrimaryKey*/
        sb.append("\t<update id=\"updateByPrimaryKey\">\n");
        sb.append("\t\tupdate ").append(BaseGeneratorUtils.camel2Underline(clazz.getSimpleName())).append("\n");
        sb.append("\t\t<set>\n");
        for (Field field : fields) {
            sb.append("\t\t\t").append(BaseGeneratorUtils.getColumn(field)).append(" = ")
                .append("#{").append(BaseGeneratorUtils.getFieldName(field)).append("},\n");
        }
        sb.append("\t\t</set>\n");
        sb.append("\t\twhere id = #{id}\n");
        sb.append("\t</update>\n\n");
        
        /**生成updateByPrimaryKeySelective*/
        sb.append("\t<update id=\"updateByPrimaryKeySelective\">\n");
        sb.append("\t\tupdate ").append(BaseGeneratorUtils.camel2Underline(clazz.getSimpleName())).append("\n");
        sb.append("\t\t<set>\n");
        for (Field field : fields) {
            sb.append("\t\t\t<if test=\"");
            if(field.getGenericType().getTypeName().indexOf(".model.") != -1){
                sb.append(field.getName()).append(" != null and ").append(field.getName()).append(".id != null");
            }else{
                sb.append(field.getName()).append(" != null");
            }
            sb.append("\">\n");
            
            sb.append("\t\t\t\t").append(BaseGeneratorUtils.getColumn(field)).append(" = ")
                .append("#{").append(BaseGeneratorUtils.getFieldName(field)).append("},\n");
            sb.append("\t\t\t</if>\n");
        }
        sb.append("\t\t</set>\n");
        sb.append("\t\twhere id = #{id}\n");
        sb.append("\t</update>\n\n");
        
        /**生成deleteByPrimaryKey*/
        sb.append("\t<delete id=\"deleteByPrimaryKey\">\n");
        sb.append("\t\tdelete from ").append(BaseGeneratorUtils.camel2Underline(clazz.getSimpleName())).append(" where id = #{id}\n");
        sb.append("\t</delete>\n");
        
        /**生成selectByPrimaryKey*/
        sb.append("\t<select id=\"selectByPrimaryKey\" resultMap=\"resultMap\">\n");
        sb.append("\t\tselect * from ").append(BaseGeneratorUtils.camel2Underline(clazz.getSimpleName())).append(" where id = #{id}\n");
        sb.append("\t</select>\n\n");
        
        /**生成selectByExample*/
        sb.append("\t<select id=\"selectByExample\" resultMap=\"resultMap\">\n");
        sb.append("\t\tselect * from ").append(BaseGeneratorUtils.camel2Underline(clazz.getSimpleName())).append("\n");
        sb.append("\t\t<where>\n");
        for (Field field : fields) {
            sb.append("\t\t\t<if test=\"");
            if(field.getGenericType().getTypeName().indexOf(".model.") != -1){
                sb.append(field.getName()).append(" != null and ").append(field.getName()).append(".id != null");
            }else{
                sb.append(field.getName()).append(" != null");
            }
            sb.append("\">\n");
            
            sb.append("\t\t\t\tand ").append(BaseGeneratorUtils.getColumn(field)).append(" = ")
                .append("#{").append(BaseGeneratorUtils.getFieldName(field)).append("}\n");
            sb.append("\t\t\t</if>\n");
        }
        sb.append("\t\t</where>\n");
        sb.append("\t</select>\n\n");
        
        sb.append("</mapper>");
        
        //保存至文件中
        String folderPath = clazz.getResource("").toString().replace("/target/classes", "/src/main/resources/mybatis");
        folderPath=folderPath.replace("com/surfilter/eyes/", "").replace("entity/", "");//文件生成路徑定製化  替換字符串
        String filePath = folderPath+clazz.getSimpleName()+"Dao.xml";
        return BaseGeneratorUtils.writeIntoFile(filePath, sb.toString());
    }
    
}
MapperUtils

　　2.BaseDaoide

　　   全部dao層默認繼承BaseDao。測試

import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.lang.reflect.TypeVariable;
import java.util.List;
import javax.annotation.Resource;
import org.mybatis.spring.SqlSessionTemplate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;


@Component
public class BaseDao<E> {
    
    private static final Logger log = LoggerFactory.getLogger(BaseDao.class);
    
    private Class<E> entityClass;
    
    @Resource
    private SqlSessionTemplate st;
    
    
    private final static String INSERT_KEY = "insert";
    private final static String INSERT_SELECTIVE_KEY = "insertSelective";
    private final static String UPDATE_BY_PRIMARYKEY_KEY = "updateByPrimaryKey";
    private final static String UPDATE_BY_PRIMARYKEY_SELECTIVE_KEY = "updateByPrimaryKeySelective";
    private final static String DELETE_BY_PRIMARYKEY_KEY = "deleteByPrimaryKey";
    private final static String SELECT_BY_PRIMARYKEY_KEY = "selectByPrimaryKey";
    private final static String SELECT_BY_PRIMARYKEY_KEY_FOR_UPDATE = "selectByPrimaryKeyForUpdate";
    
    private final static String SELECT_BY_EXAMPLE_KEY = "selectByExample";
    
    /**
     * 新增長
     * @param entity
     */
    public int insert(E entity){
        if(log.isDebugEnabled()){
            log.debug("Base dao ,insert, entity = " + getRealKey(INSERT_KEY) );
        }
        return st.insert( getRealKey(INSERT_KEY),entity );
    }
    
    public int insertSelective(E entity){
        return st.insert( getRealKey(INSERT_SELECTIVE_KEY),entity );
    }
    
    public int updateByPrimaryKey(E entity){
        return st.update(getRealKey(UPDATE_BY_PRIMARYKEY_KEY), entity);
    }

    public int updateByPrimaryKeySelective(E entity){
        return st.update(getRealKey(UPDATE_BY_PRIMARYKEY_SELECTIVE_KEY), entity);
    }
    
    public int deleteByPrimaryKey(Object id){
        return st.delete(getRealKey(DELETE_BY_PRIMARYKEY_KEY), id);
    }

    /**
     * 經過實體主鍵獲取該實體 
     * @see SELECT_BY_PRIMARYKEY_KEY
     * @param id
     * @return
     */
    public E selectByPrimaryKey(Object id){
        return st.selectOne(getRealKey(SELECT_BY_PRIMARYKEY_KEY), id);
    }
    /**
     * 經過實體主鍵獲取該實體，並加鎖 
     */
    public E selectByPrimaryKeyForUpdate(Object id){
        return st.selectOne(getRealKey(SELECT_BY_PRIMARYKEY_KEY_FOR_UPDATE), id);
    }
    
    public <E> List<E> selectByFilter(Object filter){
        return st.selectList(getRealKey(SELECT_BY_EXAMPLE_KEY), filter);
    }
    
    
    String getEntityClassName() {
        return getEntityClass().getSimpleName();
    }
    
    public String getRealKey(String key){
        StringBuilder sb = new StringBuilder();
        sb.append(getEntityClassName()).append(".").append(key);
        return sb.toString();
    }
    
    public Class<E> getEntityClass() {
        if (entityClass != null) {
            return entityClass;
        }
        Type type = getClass().getGenericSuperclass();
        if (type instanceof ParameterizedType) {
            ParameterizedType paramType = (ParameterizedType) type;
            if (paramType.getActualTypeArguments().length == 2) {
                // likely dealing with -> new
                // EntityHome<Person>().getEntityClass()
                if (paramType.getActualTypeArguments()[1] instanceof TypeVariable) {
                    throw new IllegalArgumentException(
                            "Could not guess entity class by reflection");
                }
                // likely dealing with -> new Home<EntityManager, Person>() {
                // ... }.getEntityClass()
                else {
                    entityClass = (Class<E>) paramType.getActualTypeArguments()[1];
                }
            } else {
                // likely dealing with -> new PersonHome().getEntityClass()
                // where PersonHome extends EntityHome<Person>
                entityClass = (Class<E>) paramType.getActualTypeArguments()[0];
            }
        } else {
            throw new IllegalArgumentException(
                    "Could not guess entity class by reflection");
        }
        return entityClass;
    }
    
}
BaseDao

　　3.BaseService

　　   默認注入BaseDao

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import com.surfilter.eyes.common.core.dao.BaseDao;

@Component
public class BaseService {

    @Autowired
    private BaseDao baseDao;
    
    public <E> void createEntity(E entity){
        baseDao.insert(entity);
    }
    
    public <E> void deleteEntityById(final String id){
        baseDao.deleteByPrimaryKey(id);
    }
    
    public <E> void updateEntityById(E entity){
        baseDao.updateByPrimaryKey(entity);
    }
    
    public <E> void updateEntitySelectiveById(E entity){
        baseDao.updateByPrimaryKeySelective(entity);
    }
    
    public <E> E getEntityById(Class<E> c,final String id){
        return (E) baseDao.selectByPrimaryKey(id);
    }
    
}
BaseService

　　4.生成dao、service、controller文件

　　　本文默認實現了新增方法其餘方法可仿照本身增長或者直接使用BaseService代碼，controller層默認注入相關service，service默認注入相關dao，service默認繼承BaseService。

/**
 * 生成java文件
 *
 */
public class javaUtils {
    /**
     * 生成contrller文件
     * @Title: generateController   
     * @Description: TODO  
     * @param: @param clazz
     * @param: @return      
     * @return: boolean 
     * @author: haoqingshuang     
     * @date: 2018年11月19日 下午5:59:13
     */
    public static boolean generateController(Class clazz){
        StringBuilder sb = new StringBuilder();
        String javaTxt = BaseGeneratorUtils.readJavaFile(clazz);
        String lowerJavaName = BaseGeneratorUtils.captureName(clazz.getSimpleName());
        String prevPackage = clazz.getName().split("\\.")[clazz.getName().split("\\.").length-3];
        String chineseName = BaseGeneratorUtils.getJavaAnnotation(javaTxt, clazz.getSimpleName());
        String crePackage = clazz.getName().replace("entity", "controller");
        sb.append("package ").append(crePackage.substring(0,crePackage.lastIndexOf("."))).append(";\n\n");
        sb.append("import ").append(clazz.getName()).append(";\n");
        sb.append("import org.springframework.web.bind.annotation.RestController;\n");
        sb.append("import org.springframework.web.bind.annotation.RequestMapping;\n");
        sb.append("import io.swagger.annotations.Api;\n\n");
        sb.append("import org.slf4j.Logger;\n");
        sb.append("import org.slf4j.LoggerFactory;\n");
        sb.append("import "+clazz.getName().replace("entity", "service")).append("Service;\n");
        sb.append("import org.springframework.beans.factory.annotation.Autowired;\n");
        sb.append("import com.surfilter.eyes.common.core.entity.PageResult;\n");
        sb.append("import io.swagger.annotations.Api;\nimport io.swagger.annotations.ApiOperation;\n");
        sb.append("import io.swagger.annotations.ApiResponse;\nimport io.swagger.annotations.ApiResponses;\n");
        sb.append("import org.springframework.web.bind.annotation.PostMapping;\n");
        sb.append("import com.surfilter.eyes.dago.exception.CommonException;\n");
        sb.append("/**\n");
        sb.append(" * ").append(chineseName).append("\tcontroller\n");
        sb.append(" */\n");
        
        sb.append("@RestController\n");
        sb.append("@RequestMapping(\"${versionName}/").append(BaseGeneratorUtils.camel2Slantline(lowerJavaName)).append("\")\n");
        sb.append("@Api(tags = \""+chineseName+"\")\n");
        sb.append("public class ").append(clazz.getSimpleName()).append("Controller {\n\n");
        sb.append("\tprivate static final Logger logger = LoggerFactory.getLogger("+clazz.getSimpleName()+"Controller.class);\n\n");
        sb.append("\n");
        sb.append("\t@Autowired\n");
        sb.append("\tprivate ").append(clazz.getSimpleName()+"Service ").append(BaseGeneratorUtils.captureName(clazz.getSimpleName()+"Service;\n\n"));
        
        sb.append("\t@PostMapping(\"/create\")\n");
        sb.append("\t@ApiOperation(\"新增\")\n");
        sb.append("\t@ApiResponses(value = {@ApiResponse(code = 200, message = \"請求成功\", response = String.class, responseContainer = \"String\")})\n");
        sb.append("\tpublic PageResult create"+clazz.getSimpleName()+"("+clazz.getSimpleName()+" model)  {\n");
        sb.append("\t\tlong st = System.currentTimeMillis();\n");
        sb.append("\t\tPageResult pageResult = new PageResult();\n");
        sb.append("\t\ttry {\n");
        sb.append("\t\t\tpageResult.setMsg("+BaseGeneratorUtils.captureName(clazz.getSimpleName()+"Service.save")).append(clazz.getSimpleName()+"(model));\n");
        sb.append("\t\t\tpageResult.setStatus(1);\n");
        sb.append("\t\t} catch (CommonException e) {\n");
        sb.append("\t\t\tpageResult.setMsg(e.getMessage());\n");
        sb.append("\t\t\tpageResult.setStatus(2);\n");
        sb.append("\t\t}\n");
        sb.append("\t\tpageResult.setTime(System.currentTimeMillis() - st);\n");
        sb.append("\t\treturn pageResult;\n");
        sb.append("\t}\n");
        
        sb.append("}");
        
        String folderPath = clazz.getResource("").toString().replace("/target/classes", "/src/main/java").replace("/model", "").replace("entity", "controller");
        String filePath = folderPath + clazz.getSimpleName() + "Controller.java";
        return BaseGeneratorUtils.writeIntoFile(filePath, sb.toString());
    }
    
    /**
     * 生成dao 繼承BaseDao接口
     * @Title: generateDao   
     * @Description: TODO  
     * @param: @param clazz
     * @param: @return      
     * @return: boolean 
     * @author: haoqingshuang     
     * @date: 2018年11月19日 下午3:36:58
     */
    public static boolean generateDao(Class clazz){
        StringBuilder sb = new StringBuilder();
        String javaTxt = BaseGeneratorUtils.readJavaFile(clazz);
        String lowerJavaName = BaseGeneratorUtils.captureName(clazz.getSimpleName());
        String prevPackage = clazz.getName().split("\\.")[clazz.getName().split("\\.").length-3];
        String chineseName = BaseGeneratorUtils.getJavaAnnotation(javaTxt, clazz.getSimpleName())+"dao";
        //sb.append("package ").append(clazz.getName().substring(0, clazz.getName().indexOf("Model")).replace("entity", "repository")).append("Dao;\n\n");
        String crePackage = clazz.getName().replace("entity", "repository");
        sb.append("package ").append(crePackage.substring(0,crePackage.lastIndexOf("."))).append(";\n\n");
        sb.append("import ").append(clazz.getName()).append(";\n");
        sb.append("import com.surfilter.eyes.common.core.dao.BaseDao;\n");
        sb.append("import org.springframework.stereotype.Repository;\n");
        sb.append("/**\n");
        sb.append(" * ").append(chineseName).append("\n");
        sb.append(" */\n");
        sb.append("@Repository\n");
        sb.append("public class ").append(clazz.getSimpleName()).append("Dao extends BaseDao<").append(clazz.getSimpleName()).append(">{\n");
        sb.append("\n");
        sb.append("}");
        String folderPath = clazz.getResource("").toString().replace("/target/classes", "/src/main/java").replace("/model", "").replace("entity", "repository");
        String filePath = folderPath + clazz.getSimpleName() + "Dao.java";
        return BaseGeneratorUtils.writeIntoFile(filePath, sb.toString());
    }
    
    /**
     * 生成service文件 目前只寫了insert方法 並注入了dao層
     * @Title: generateService   
     * @Description: TODO  
     * @param: @param clazz
     * @param: @return      
     * @return: boolean 
     * @author: haoqingshuang     
     * @date: 2018年11月19日 下午4:40:05
     */
    public static boolean generateService(Class clazz){
        StringBuilder sb = new StringBuilder();
        String javaTxt = BaseGeneratorUtils.readJavaFile(clazz);
        String lowerJavaName = BaseGeneratorUtils.captureName(clazz.getSimpleName());
        String prevPackage = clazz.getName().split("\\.")[clazz.getName().split("\\.").length-3];
        String chineseName = BaseGeneratorUtils.getJavaAnnotation(javaTxt, clazz.getSimpleName());
        //sb.append("package ").append(clazz.getName().substring(0, clazz.getName().indexOf("Model")).replace("entity", "repository")).append("Dao;\n\n");
        String crePackage = clazz.getName().replace("entity", "service");
        sb.append("package ").append(crePackage.substring(0,crePackage.lastIndexOf("."))).append(";\n\n");
        sb.append("import ").append(clazz.getName()).append(";\n");
        sb.append("import org.springframework.stereotype.Service;\n");
        sb.append("import org.springframework.beans.factory.annotation.Autowired;\n");
        sb.append("import org.slf4j.Logger;\n");
        sb.append("import org.slf4j.LoggerFactory;\n");
        sb.append("import org.springframework.transaction.annotation.Transactional;\n");
        sb.append("import com.surfilter.eyes.dago.exception.CommonException;\n");
        sb.append("import com.surfilter.eyes.common.core.service.BaseService;\n");
        //import com.surfilter.eyes.dago.entity.integration.scheduling.HaoTestModel;
        //import com.surfilter.eyes.dago.repository.integration.scheduling.DagoTestModelDao;
        sb.append("import "+clazz.getName().replace("entity", "repository")).append("Dao;\n");
        
        sb.append("/**\n");
        sb.append(" * ").append(chineseName).append("\tservice\n");
        sb.append(" */\n");
        sb.append("@Service\n");
        sb.append("public class ").append(clazz.getSimpleName()).append("Service extends BaseService");
        sb.append("{\n\n");
        sb.append("\tprivate static final Logger logger = LoggerFactory.getLogger("+clazz.getSimpleName()+"Service.class);\n\n");
        sb.append("\t@Autowired\n");
        sb.append("\tprivate ").append(clazz.getSimpleName()+"Dao ").append(BaseGeneratorUtils.captureName(clazz.getSimpleName()+"Dao;\n\n"));
        
        sb.append("\t/**\n");
        sb.append("\t * ").append("新增"+chineseName+"\n");
        sb.append("\t */\n");
        sb.append("\t@Transactional(rollbackFor = Exception.class)\n");
        sb.append("\tpublic String save"+clazz.getSimpleName()+"("+clazz.getSimpleName()+" model)throws CommonException {\n");
        sb.append("\t\tif(null==model){\n");
        sb.append("\t\t\tthrow new CommonException(\"參數錯誤，請檢查！\");\n");
        sb.append("\t\t}\n");
        sb.append("\t\t"+BaseGeneratorUtils.captureName(clazz.getSimpleName()+"Dao")+".insert(model);\n");
        sb.append("\t\treturn \"新增成功！\";\n");
        sb.append("\t}\n\n");
        
        sb.append("}");
        String folderPath = clazz.getResource("").toString().replace("/target/classes", "/src/main/java").replace("/model", "").replace("entity", "service");
        String filePath = folderPath + clazz.getSimpleName() + "Service.java";
        return BaseGeneratorUtils.writeIntoFile(filePath, sb.toString());
    }
}
javaUtils

　　5.生成建表sql語句

　　   權限語句、菜單語句可參考建表語句自行編寫。

import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.lang.reflect.Type;
import org.apache.commons.lang.StringUtils;
import org.hibernate.validator.constraints.Length;

public class SqlUtils {
    public static String generateDdl(Class clazz) {
        StringBuilder sb = new StringBuilder("CREATE TABLE `").append(BaseGeneratorUtils.camel2Underline(clazz.getSimpleName()))
                .append("` (").append("\n");
        String javaTxt = BaseGeneratorUtils.readJavaFile(clazz);
        Field[] fields = clazz.getDeclaredFields();
        for (Field field : fields) {
            String fieldName = BaseGeneratorUtils.camel2Underline(field.getName());
            if(field.getGenericType().getTypeName().indexOf(".model.") != -1){
                fieldName += "_id";
            }
            sb.append("  `").append(fieldName).append("` ");
            /** 判斷類型信息 */
            Type type = field.getType();
            switch (type.getTypeName()) {
            case "java.lang.String":
                sb.append("varchar(");
                // 對於字符串考慮註解
                int max = 255;
                Annotation[] annotations = field.getAnnotations();
                for(Annotation annotation : annotations){
                    if(annotation.annotationType().getName() == "org.hibernate.validator.constraints.Length"){
                        Length length = (Length) annotation;
                        max = length.max()>0?length.max():max;
                    }
                }
                sb.append(max).append(") ");
                break;
            case "java.util.Date":
                sb.append("datetime ");
                break;
            case "int":
            case "java.lang.Integer":
            case "long":
            case "java.lang.Long":
                sb.append("int(11) ");
                break;
            case "double":
            case "java.lang.Double":
                sb.append("double ");
                break;
            default: // 關聯對象、玫舉
                sb.append("varchar(32) ");
            }
            //註解
            Annotation[] annotations = field.getAnnotations();
            for(Annotation annotation : annotations){
                switch(annotation.annotationType().getName()){
                case "org.hibernate.validator.constraints.NotEmpty":
                case "org.hibernate.validator.constraints.NotBlank":
                case "javax.validation.constraints.NotNull":
                    sb.append("NOT NULL ");
                }
            }
            
            // 註釋
            String note = BaseGeneratorUtils.getJavaAnnotation(javaTxt, field.getName());
            if(StringUtils.isNotEmpty(note)){
                sb.append("COMMENT '").append(note).append("'");
            }
            
            sb.append(",\n");
        }
        sb.append("  PRIMARY KEY (`id`)").append("\n").append(") ENGINE=InnoDB DEFAULT CHARSET=utf8 ");
        // 表註釋
        String clazzNote = BaseGeneratorUtils.getJavaAnnotation(javaTxt, clazz.getSimpleName());
        if(StringUtils.isNotEmpty(clazzNote)){
            sb.append("COMMENT='").append(clazzNote).append("';");
        }else{
            sb.append(";");
        }
        return sb.toString();
    }

}
SqlUtils

　　6.基礎文件生成類

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * 根據實體生成相應資源
 * 1. 全部實體類均默認主鍵爲id
 * 2. 生成的相關的sql
 * 3. 生成代碼過程當中，會參考實體類的註釋及註解（\@Length \@NotBlank \@NotEmpty)，因此先完善註解再生成
 * 目前可生成的代碼有：
 *  mapper、controller、sql(建表語句)
 */
public class SourceGenerateUtils {
    
    private static final Logger logger = LoggerFactory.getLogger(SourceGenerateUtils.class);
    
    public static void generate(Class clazz){
        
        if(MapperUtils.generateMapper(clazz)){
            logger.info(clazz.getSimpleName()+"Dao.xml 文件已生成");
        }else{
            logger.info(clazz.getSimpleName()+"Dao.xml 文件已存在，未從新生成");
        }
        if(javaUtils.generateDao(clazz)){
            logger.info(clazz.getSimpleName()+"Dao.java 文件已生成");
        }else{
            logger.info(clazz.getSimpleName()+"Dao.java 文件已存在，未從新生成");
        }
        if(javaUtils.generateService(clazz)){
            logger.info(clazz.getSimpleName()+"Service.java 文件已生成");
        }else {
            logger.info(clazz.getSimpleName()+"Service.java 文件已存在，未從新生成");
        }
        if(javaUtils.generateController(clazz)){
            logger.info(clazz.getSimpleName()+"Controller.java 文件已生成");
        }else {
            logger.info(clazz.getSimpleName()+"Controller.java 文件已存在，未從新生成");
        }
        
        logger.info("正在生成Sql語句，sql僅供參考，注意審查...");
        logger.info("#建表語句\n"+SqlUtils.generateDdl(clazz));
        
    }

}
SourceGenerateUtils

　　7.對象轉化爲Map形式

　　   把對象轉化爲Map形式，key類型String；value類型Object（只推薦在Mybatis dao層中使用）。

import java.beans.PropertyDescriptor;
import java.lang.reflect.InvocationTargetException;
import java.util.HashMap;
import java.util.Map;
import org.apache.commons.beanutils.PropertyUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * 把對象轉化爲Map形式，key類型String；value類型Object（只推薦在Mybatis dao層中使用）
 *
 */
public class MybatisExample2MapUtils {
    
    private final static Log log = LogFactory.getLog(MybatisExample2MapUtils.class);
    
    /**
     * 把對象轉化爲Map形式，key String；value Object
     * @param bean
     * @return
     */
    public static Map<?,?> example2Map(final Object bean){
        if(bean == null){
            return (new HashMap<Object, Object>());
        }
        
        final Map<String, Object> result = new HashMap<String, Object>();
        
        final PropertyDescriptor[] descriptors =
                PropertyUtils.getPropertyDescriptors(bean);
            
            for (PropertyDescriptor descriptor : descriptors) {
                final String name = descriptor.getName();
                
                if(PropertyUtils.getReadMethod(descriptor) != null){
                    try {
                        final Object tempObj = PropertyUtils.getProperty(bean, name);
                        //log.debug("第二層對象解析：" + tempObj);
                        if(tempObj == null){
                            continue;
                        }
                        //log.debug("屬性名稱：" + tempObj.getClass().getName());
                        if(tempObj.getClass().getName().startsWith("com.surfilter")){
                            
                            final PropertyDescriptor[] descriptors2 =
                                    PropertyUtils.getPropertyDescriptors(tempObj);
                               
                            for (PropertyDescriptor descriptor2 : descriptors2) {
                                final String name2 = descriptor2.getName();
                                if(PropertyUtils.getReadMethod(descriptor) != null){
                                    Object obj2 = PropertyUtils.getProperty(tempObj, name2);
                                    if(obj2 == null){
                                        continue;
                                    }
                                    result.put(name+"_"+name2, obj2);
                                }
                            }
                                
                        }else{
                            result.put(name, tempObj);
                        }
                        
                    } catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
                        log.error("Bean轉化爲Map錯誤！ Message："+ e.getMessage());
                        e.printStackTrace();
                    }
                }
            }
        
        if(log.isDebugEnabled()){
            log.debug("轉化結果=" + result);
        }
        return result ;
    }
}
MybatisExample2MapUtils

　　8.實體類

　　   注意本文使用了lombok

　　   實體類註釋、屬性註釋請採用以下格式，其餘格式的註釋可能存在取不到的狀況(未測試),類註釋用於swagger接口說明、controller、service、dao註釋。屬性註釋主要用於生成sql語句的字段描述。

　　　/**

　　　 *註釋內容XXX

　　　 */

　　　請在相關屬性上添加@Length 、@NotBlank 、@NotEmpty、@Length(max = 200)註解，用於生成sql判斷是否爲空、及長度，還可做爲實體類驗證使用。 

　　   測試實體類以下：直接在該類下運行main方法便可 注意查看控制檯輸出。

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import javax.validation.constraints.NotEmpty;
import javax.validation.constraints.NotNull;
import lombok.Data;
import org.hibernate.validator.constraints.Length;
import com.surfilter.eyes.dago.utils.generator.SourceGenerateUtils;

/**
 * 測試實體
 */
@Data
@ApiModel(value = "HaoTestModel", description = "測試實體")
public class HaoTestModel {
    
    /**
     * 編號
     */
    @NotEmpty
    @Length(max = 200)
    @ApiModelProperty(value = "編號", required = true)
    private String id;
    
    /**
     * 姓名
     */
    @NotEmpty
    @Length(max = 200)
    @ApiModelProperty(value = "姓名", required = true)
    private String name;
    
    /**
     * 年齡
     */
    @NotNull
    @ApiModelProperty(value = "用戶類別", required = true)
    private Integer userType;
    
    public static void main(String[] args) {
        SourceGenerateUtils.generate(HaoTestModel.class);
    }

}
HaoTestModel

 

3、生成文件展現

　　1.生成的mapper文件

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">

<mapper namespace="HaoTestModel">
    <resultMap id="resultMap" type="com.surfilter.eyes.dago.entity.integration.scheduling.HaoTestModel">
        <id column="id" property="id"/>
        <result column="name" property="name"/>
        <result column="user_type" property="userType"/>
    </resultMap>

    <sql id="columns">
        `id`,`name`,`user_type`
    </sql>

    <sql id="properties">
        #{id},#{name},#{userType}
    </sql>

    <insert id="insert">
        insert into hao_test_model (<include refid="columns"></include>)
        values (<include refid="properties"></include>)
    </insert>

    <update id="updateByPrimaryKey">
        update hao_test_model
        <set>
            id = #{id},
            name = #{name},
            user_type = #{userType},
        </set>
        where id = #{id}
    </update>

    <update id="updateByPrimaryKeySelective">
        update hao_test_model
        <set>
            <if test="id != null">
                id = #{id},
            </if>
            <if test="name != null">
                name = #{name},
            </if>
            <if test="userType != null">
                user_type = #{userType},
            </if>
        </set>
        where id = #{id}
    </update>

    <delete id="deleteByPrimaryKey">
        delete from hao_test_model where id = #{id}
    </delete>
    <select id="selectByPrimaryKey" resultMap="resultMap">
        select * from hao_test_model where id = #{id}
    </select>

    <select id="selectByExample" resultMap="resultMap">
        select * from hao_test_model
        <where>
            <if test="id != null">
                and id = #{id}
            </if>
            <if test="name != null">
                and name = #{name}
            </if>
            <if test="userType != null">
                and user_type = #{userType}
            </if>
        </where>
    </select>

</mapper>
HaoTestModel

　　2.生成的dao文件

import com.surfilter.eyes.dago.entity.integration.scheduling.HaoTestModel;
import com.surfilter.eyes.common.core.dao.BaseDao;
import org.springframework.stereotype.Repository;
/**
 * 測試實體dao
 */
@Repository
public class HaoTestModelDao extends BaseDao<HaoTestModel>{

}
HaoTestModelDao

　　3.生成的service文件

import com.surfilter.eyes.dago.entity.integration.scheduling.HaoTestModel;
import org.springframework.stereotype.Service;
import org.springframework.beans.factory.annotation.Autowired;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.transaction.annotation.Transactional;
import com.surfilter.eyes.dago.exception.CommonException;
import com.surfilter.eyes.common.core.service.BaseService;
import com.surfilter.eyes.dago.repository.integration.scheduling.HaoTestModelDao;
/**
 * 測試實體    service
 */
@Service
public class HaoTestModelService extends BaseService{

    private static final Logger logger = LoggerFactory.getLogger(HaoTestModelService.class);

    @Autowired
    private HaoTestModelDao haoTestModelDao;

    /**
     * 新增測試實體
     */
    @Transactional(rollbackFor = Exception.class)
    public String saveHaoTestModel(HaoTestModel model)throws CommonException {
        if(null==model){
            throw new CommonException("參數錯誤，請檢查！");
        }
        haoTestModelDao.insert(model);
        return "新增成功！";
    }

}
HaoTestModelService

　　4.生成的controller文件

import com.surfilter.eyes.dago.entity.integration.scheduling.HaoTestModel;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.bind.annotation.RequestMapping;
import io.swagger.annotations.Api;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.surfilter.eyes.dago.service.integration.scheduling.HaoTestModelService;
import org.springframework.beans.factory.annotation.Autowired;
import com.surfilter.eyes.common.core.entity.PageResult;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiResponse;
import io.swagger.annotations.ApiResponses;
import org.springframework.web.bind.annotation.PostMapping;
import com.surfilter.eyes.dago.exception.CommonException;
/**
 * 測試實體    controller
 */
@RestController
@RequestMapping("${versionName}/hao/test/model")
@Api(tags = "測試實體")
public class HaoTestModelController {

    private static final Logger logger = LoggerFactory.getLogger(HaoTestModelController.class);


    @Autowired
    private HaoTestModelService haoTestModelService;

    @PostMapping("/create")
    @ApiOperation("新增")
    @ApiResponses(value = {@ApiResponse(code = 200, message = "請求成功", response = String.class, responseContainer = "String")})
    public PageResult createHaoTestModel(HaoTestModel model)  {
        long st = System.currentTimeMillis();
        PageResult pageResult = new PageResult();
        try {
            pageResult.setMsg(haoTestModelService.saveHaoTestModel(model));
            pageResult.setStatus(1);
        } catch (CommonException e) {
            pageResult.setMsg(e.getMessage());
            pageResult.setStatus(2);
        }
        pageResult.setTime(System.currentTimeMillis() - st);
        return pageResult;
    }
}
HaoTestModelController

　　5.生成的sql文件

CREATE TABLE `hao_test_model` (
  `id` varchar(200) COMMENT '編號',
  `name` varchar(200) COMMENT '姓名',
  `user_type` int(11) NOT NULL COMMENT '年齡',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='測試實體';
hao_test_model

　　6.文件生成明細

           

4、注意

　　　　1.若是XXXDao、XXXService、XXXContrller、mapper文件已生成 則不會再生成，防止發生覆蓋後悔莫及，請放心運行main方法。

　　　　2.類註釋請必定書寫，不少地方都用到了類註釋。

　　　　3.文中controller、service只寫了一個添加的方法，能夠寫一個BaseController、BaseService 並繼承該類。

　　　　4.若有問題歡迎評論交流。

 

 

 

相關文章
相關標籤/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公眾號
   歡迎關注本站公眾號,獲取更多信息
聯繫我們 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程