mybatisPlus自动代码生成实例(超级简单使用)
每次新项目的开发,都避免不了建很多的dao层、service等等的文件夹,显得很是麻烦,在此推荐一种可快速生成entity、service、mapper、controller文件,以便节省我们的时间,减少加班。。。。。。
1.pom中导入以下三个包
<!-- 模板 velocity --> <dependency> <groupId>org.apache.velocity</groupId> <artifactId>velocity</artifactId> <version>1.7</version> </dependency>
<!-- java工具类 --> <dependency> <groupId>cn.hutool</groupId> <artifactId>hutool-all</artifactId> <version>4.5.18</version> </dependency>
<!-- mybatis扩展插件包mybatis-plus --> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus</artifactId> <version>2.3</version> </dependency>
2.新建一个类AutoFile
package com.example.vueservice.controller; import cn.hutool.core.collection.CollUtil; import com.baomidou.mybatisplus.enums.FieldFill; import com.baomidou.mybatisplus.generator.AutoGenerator; import com.baomidou.mybatisplus.generator.config.*; import com.baomidou.mybatisplus.generator.config.po.TableFill; import com.baomidou.mybatisplus.generator.config.rules.DbType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import java.util.List; public class AutoFile { /** * 测试 run 执行 注意:不生成service接口 注意:不生成service接口 注意:不生成service接口 * <p> * 配置方法查看 {@link ConfigGenerator} * </p> */ public static void main(String[] args) { AutoGenerator mpg = new AutoGenerator(); // 全局配置 GlobalConfig gc = new GlobalConfig(); gc.setOutputDir("C://");//代码生成存放位置 gc.setFileOverride(true); gc.setActiveRecord(false); gc.setEnableCache(false);// XML 二级缓存 gc.setBaseResultMap(true);// XML ResultMap gc.setBaseColumnList(false);// XML columList gc.setOpen(true); gc.setAuthor("小明");//此处填写代码生成人姓名 // 自定义文件命名,注意 %s 会自动填充表实体属性! // gc.setMapperName("%sDao"); // gc.setXmlName("%sDao"); gc.setServiceImplName("%sService"); // gc.setServiceImplName("%sServiceDiy"); // gc.setControllerName("%sAction"); mpg.setGlobalConfig(gc); // 数据源配置 DataSourceConfig dsc = new DataSourceConfig(); dsc.setDbType(DbType.MYSQL); dsc.setDriverName("com.mysql.jdbc.Driver"); dsc.setUsername("xxx");//数据库用户名 dsc.setPassword("xxx");//数据库密码 dsc.setUrl("jdbc:mysql://xxx:xxxx/xxx?characterEncoding=utf8&allowMultiQueries=true");//数据库连接地址 /*dsc.setTypeConvert(new SqlServerTypeConvert(){ // 自定义数据库表字段类型转换【可选】 @Override public DbColumnType processTypeConvert(String fieldType) { if (fieldType.contains("tinyint") ) { return DbColumnType.BLOB; } // 注意!!processTypeConvert 存在默认类型转换, //如果不是你要的效果请自定义返回、非如下直接返回。 return super.processTypeConvert(fieldType); } });*/ mpg.setDataSource(dsc); // 策略配置 StrategyConfig strategy = new StrategyConfig(); // strategy.setTablePrefix("sys_");// 此处可以修改为您的表前缀 strategy.setNaming(NamingStrategy.underline_to_camel);// 表名生成策略 /** * 第一步 * 代码生成需要改动这里,将待生成的表名替换到下面 * * 2017.11.2 */ strategy.setInclude(new String[] { "t_user" }); // 需要生成的表 /** * 第二步 * 去掉表名前缀 * * 2017.11.2 */ strategy.setTablePrefix(new String[] { "t_" });// 此处可以修改为您的表前缀 // strategy.setExclude(new String[]{"test"}); // 排除生成的表 // 自定义实体父类 strategy.setSuperEntityClass("com.mayi.test.base.BaseEntity"); // 自定义实体,公共字段 strategy.setSuperEntityColumns( new String[] {}); // 自定义 mapper 父类 strategy.setSuperMapperClass("com.baomidou.mybatisplus.mapper.BaseMapper"); // 自定义 service 父类 strategy.setSuperServiceImplClass("com.mayi.test.base.BaseService"); // 自定义 controller 父类 strategy.setSuperControllerClass("com.mayi.test.base.BaseController"); //逻辑删除属性名称 strategy.setLogicDeleteFieldName("is_deleted"); List<TableFill> tableFillList= CollUtil.newArrayList(); TableFill fill=new TableFill("update_time", FieldFill.INSERT_UPDATE); tableFillList.add(fill); fill=new TableFill("create_time", FieldFill.INSERT); tableFillList.add(fill); strategy.setTableFillList(tableFillList); // 自定义 service 实现类父类 //strategy.setSuperServiceImplClass("org.ibase4j.core.base.BaseService"); // 自定义 controller 父类 //strategy.setSuperControllerClass("org.ibase4j.core.base.AbstractController"); // 【实体】是否生成字段常量(默认 false) // public static final String ID = "test_id"; // strategy.setEntityColumnConstant(true); // 【实体】是否为构建者模型(默认 false) // public User setName(String name) {this.name = name; return this;} // strategy.setEntityBuliderModel(true); mpg.setStrategy(strategy); // 包配置 PackageConfig pc = new PackageConfig(); pc.setParent("com.mayi.test");//此处填写包名 /** * 第三步 * 按模块调整代码放置目录 * * 2017.11.2 */ pc.setEntity("sys.entity"); pc.setMapper("sys.dao"); pc.setXml("sys.dao.mappers"); pc.setServiceImpl("sys.service"); pc.setController("sys.web"); mpg.setPackageInfo(pc); // 注入自定义配置,可以在 VM 中使用 cfg.abc 设置的值 /*InjectionConfig cfg = new InjectionConfig() { public void initMap() { Map<String, Object> map = new HashMap<String, Object>(); map.put("providerClass", "ISysProvider"); map.put("providerClassPackage", "org.ibase4j.provider.ISysProvider"); this.setMap(map); } }; mpg.setCfg(cfg);*/ // 自定义模板配置,可以 copy 源码 mybatis-plus/src/main/resources/template 下面内容修改, // 放置自己项目的 src/main/resources/template 目录下, 默认名称一下可以不配置,也可以自定义模板名称 TemplateConfig tc = new TemplateConfig(); tc.setEntity("templates/entity.java.vm"); tc.setMapper("templates/mapper.java.vm"); tc.setXml("templates/mapper.xml.vm"); tc.setServiceImpl("templates/service.java.vm"); tc.setController("templates/controller.java.vm"); mpg.setTemplate(tc); // 执行生成 mpg.execute(); System.out.println("代码生成完毕"); } }
3.导入模板,模板可根据自己的需求修改,在此只提供基础
controller.java.vm
package ${package.Controller};
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.plugins.Page;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import xxx.entity.RestResultGenerator;//1,2,3,4,5,6导包类在博客文最下边
import xx.beanvalidator.AddGroup;//2
import xx.beanvalidator.EditGroup;//3
import xx.beanvalidator.PageGroup;//4
import xx.BaseQuery;//5
import xx.util.ValidateUtil;//6
import ${package.Entity}.${entity};
#if(${restControllerStyle})
import org.springframework.web.bind.annotation.RestController;
#else
import org.springframework.stereotype.Controller;
#end
#if(${superControllerClassPackage})
import ${superControllerClassPackage};
#end
import ${package.ServiceImpl}.${entity}Service;
import cn.hutool.core.util.StrUtil;
/**
*
* @ClassName: ${table.controllerName}
* @Description: TODO(这里用一句话描述这个类的作用)
* @author ${author}
* @date ${date}
*/
#if(${restControllerStyle})
@RestController
#else
@Controller
#end
@RequestMapping(value = "#if(${package.ModuleName})/${package.ModuleName}#end/#if(${controllerMappingHyphenStyle})${controllerMappingHyphen}#else${table.entityPath}#end")
#if(${superControllerClass})
public class ${table.controllerName} extends ${superControllerClass}<${entity}> {
#else
public class ${table.controllerName} {
#end
@Autowired
private ${entity}Service ${table.entityPath}Service;
@RequestMapping(method = RequestMethod.POST, value = "/add")
public void add(${entity} ${table.entityPath}, HttpServletRequest request,
HttpServletResponse response) {
beanValidator(${table.entityPath}, AddGroup.class);
${table.entityPath}Service.insert(${table.entityPath});
renderResult(response);
}
@RequestMapping(method = RequestMethod.POST, value = "/edit")
public void edit(${entity} ${table.entityPath}, HttpServletRequest request,
HttpServletResponse response) {
beanValidator(${table.entityPath}, EditGroup.class);
${table.entityPath}Service.updateById(${table.entityPath});
renderResult(response);
}
@RequestMapping(method = RequestMethod.POST, value = "/delete")
public void delete(HttpServletRequest request, HttpServletResponse response,
String ids) {
//ValidateUtil.isNotBlank(ids, "主键参数非法,操作失败,请检查");
List<String> idList = StrUtil.split(ids, ',');
${table.entityPath}Service.deleteBatchIds(idList);
renderResult(response);
}
@RequestMapping(method = RequestMethod.GET, value = "/detail")
public void detail(HttpServletRequest request, HttpServletResponse response,
String id) {
//ValidateUtil.isNotBlank(id, "主键参数非法,操作失败,请检查");
${entity} ${table.entityPath} = ${table.entityPath}Service.selectById(id);
renderResult(response, RestResultGenerator.success(${table.entityPath}));
}
@RequestMapping(method = RequestMethod.GET, value = "/queryList")
public void queryList(HttpServletRequest request, HttpServletResponse response) {
EntityWrapper<${entity}> wrapper = new EntityWrapper<${entity}>();
List<${entity}> list = ${table.entityPath}Service.selectList(wrapper);
renderResult(response, RestResultGenerator.success(list));
}
@RequestMapping(method = RequestMethod.GET, value = "/queryPageList")
public void queryPageList(HttpServletRequest request, HttpServletResponse response,
BaseQuery query) {
beanValidator(query, PageGroup.class);
EntityWrapper<${entity}> wrapper = new EntityWrapper<${entity}>();
Page<${entity}> pg = new Page<${entity}>(query.getPage(), query.getRows());
Page<${entity}> list = ${table.entityPath}Service.selectPage(pg, wrapper);
renderResult(response, RestResultGenerator.success(list));
}
}
entity.java.vm
package ${package.Entity};
import xxx.beanvalidator.EditGroup;//在博客最下边
import java.io.Serializable;
import com.baomidou.mybatisplus.annotations.TableId;
#foreach($pkg in ${table.importPackages})
import ${pkg};
#end
#if(${entityLombokModel})
import com.baomidou.mybatisplus.annotations.Version;
import lombok.Data;
import lombok.experimental.Accessors;
#end
import org.hibernate.validator.constraints.NotBlank;
/**
*
* @ClassName: ${entity}
* @Description: TODO(这里用一句话描述这个类的作用)
* @author ${author}
* @date ${date}
*/
#if(${entityLombokModel})
@Data
@EqualsAndHashCode(callSuper = true)
@Accessors(chain = true)
#end
#if(${table.convert})
@TableName("${table.name}")
#end
#if(${superEntityClass})
public class ${entity} extends ${superEntityClass} implements Serializable {
#else
public class ${entity} implements Serializable {
#end
private static final long serialVersionUID = 1L;
## ---------- BEGIN 字段循环遍历 ----------
#foreach($field in ${table.fields})
#if(${field.keyFlag})
#set($keyPropertyName=${field.propertyName})
#end
#if("$!field.comment" != "")
/**
* ${field.comment}
*/
#end
#if(${field.keyFlag})
## 主键
#if(${field.keyIdentityFlag})
@TableId(value="${field.name}", type= IdType.ID_WORKER)
@NotBlank(message = "主键不能为空", groups={EditGroup.class})
#elseif(${field.convert})
@TableId("${field.name}")
@NotBlank(message = "主键不能为空", groups={EditGroup.class})
#else
@TableId("${field.name}")
@NotBlank(message = "主键不能为空", groups={EditGroup.class})
#end
## 普通字段
#elseif(${field.fill})
## ----- 存在字段填充设置 -----
#if(${field.convert})
@TableField(value = "${field.name}", fill = FieldFill.${field.fill})
#else
@TableField(fill = FieldFill.${field.fill})
#end
#elseif(${field.convert})
@TableField("${field.name}")
#else
@TableField("${field.name}")
#end
## 乐观锁注解
#if(${versionFieldName}==${field.name})
@Version
#end
## 逻辑删除注解
#if(${logicDeleteFieldName}==${field.name})
@TableLogic
#end
private ${field.propertyType} ${field.propertyName};
#end
## ---------- END 字段循环遍历 ----------
#if(!${entityLombokModel})
#foreach($field in ${table.fields})
#if(${field.propertyType.equals("boolean")})
#set($getprefix="is")
#else
#set($getprefix="get")
#end
public ${field.propertyType} ${getprefix}${field.capitalName}() {
return ${field.propertyName};
}
#if(${entityBuilderModel})
public ${entity} set${field.capitalName}(${field.propertyType} ${field.propertyName}) {
#else
public void set${field.capitalName}(${field.propertyType} ${field.propertyName}) {
#end
this.${field.propertyName} = ${field.propertyName};
#if(${entityBuilderModel})
return this;
#end
}
#end
#end
#if(${entityColumnConstant})
#foreach($field in ${table.fields})
public static final String ${field.name.toUpperCase()} = "${field.name}";
#end
#end
#if(!${entityLombokModel})
@Override
public String toString() {
return "${entity}{" +
#foreach($field in ${table.fields})
#if($!{velocityCount}==1)
"${field.propertyName}=" + ${field.propertyName} +
#else
", ${field.propertyName}=" + ${field.propertyName} +
#end
#end
"}";
}
#end
}
mapper.java.vm
package ${package.Mapper}; import ${package.Entity}.${entity}; import ${superMapperClassPackage}; public interface ${table.mapperName} extends ${superMapperClass}<${entity}> { }
mapper.xml.vm
<?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="${package.Mapper}.${table.mapperName}"> #if(${enableCache}) <!-- 开启二级缓存 --> <cache type="org.mybatis.caches.ehcache.LoggingEhcache"/> #end #if(${baseResultMap}) <!-- 通用查询映射结果 --> <resultMap id="BaseResultMap" type="${package.Entity}.${entity}"> #foreach($field in ${table.fields}) #if(${field.keyFlag})##生成主键排在第一位 <id column="${field.name}" property="${field.propertyName}" /> #end #end #foreach($field in ${table.commonFields})##生成公共字段 <result column="${field.name}" property="${field.propertyName}" /> #end #foreach($field in ${table.fields}) #if(!${field.keyFlag})##生成普通字段 <result column="${field.name}" property="${field.propertyName}" /> #end #end </resultMap> #end #if(${baseColumnList}) <!-- 通用查询结果列 --> <sql id="Base_Column_List"> #foreach($field in ${table.commonFields}) #if(${field.name} == ${field.propertyName})${field.name}#else${field.name} AS ${field.propertyName}#end, #end ${table.fieldNames} </sql> #end </mapper>
service.java.vm
package ${package.ServiceImpl}; import ${package.Entity}.${entity}; import ${package.Mapper}.${table.mapperName}; import ${superServiceImplClassPackage}; import org.springframework.stereotype.Service; /** * * @ClassName: ${entity}Service * @Description: TODO(这里用一句话描述这个类的作用) * @author ${author} * @date ${date} */ @Service public class ${entity}Service extends ${superServiceImplClass}<${table.mapperName}, ${entity}> { }
模板存放位置:
4.最后生成文件目录
======================================================补充代码=============================================================
/** * restful请求统一返回响应体生成工具 * */ public class RestResultGenerator { private RestResultGenerator() { } public static <T> RestResult<T> success() { return success(null, "操作成功!"); } public static <T> RestResult<T> success(T data) { return success(data, "操作成功!"); } public static <T> RestResult<T> success(T data, String msg, boolean result) { return generator(result, data, msg, HttpCodeEnum.OK.value()); } public static <T> RestResult<T> success(T data, String msg) { return generator(true, data, msg, HttpCodeEnum.OK.value()); } public static <T> RestResult<T> success(T data, String msg, String ticket) { return generator(true, data, msg, ticket, HttpCodeEnum.OK.value()); } public static <T> RestResult<T> success(T data, String msg, Integer httpCode) { return generator(true, data, msg, httpCode); } public static <T> RestResult<T> fail() { return fail(null, null); } public static <T> RestResult<T> fail(String msg) { return fail(null, msg); } public static <T> RestResult<T> fail(T data, String msg) { return generator(false, data, msg, HttpCodeEnum.SERVER_ERROR.value()); } public static <T> RestResult<T> generator(boolean result, T data, String msg, Integer httpCode) { return generator(result, data, msg, null, httpCode); } public static <T> RestResult<T> generator(boolean result, T data, String msg, String ticket, Integer httpCode) { RestResult<T> restResult = new RestResult<T>(); restResult.setResult(result); restResult.setData(data); restResult.setMsg(msg); restResult.setTicket(ticket); restResult.setHttpCode(httpCode); return restResult; } }
public class RestResult<T> implements Serializable { private static final long serialVersionUID = 5363887016344037411L; private boolean result; private T data; private Integer httpCode; private String msg; private String ticket; public RestResult() { } public RestResult(boolean result) { this.result = result; } public RestResult(String msg) { this.msg = msg; } public RestResult(String msg, boolean result) { this.msg = msg; this.result = result; } public String toJSONString() { return toJSONString("yyyy-MM-dd HH:mm:ss"); } public String toJSONString(String dateFormat) { return JsonUtil.toJSONString(this, dateFormat); } public boolean isResult() { return result; } public void setResult(boolean result) { this.result = result; } public T getData() { return data; } public void setData(T data) { this.data = data; } public Integer getHttpCode() { return httpCode; } public void setHttpCode(Integer httpCode) { this.httpCode = httpCode; } public String getMsg() { return msg; } public void setMsg(String msg) { this.msg = msg; } public String getTicket() { return ticket; } public void setTicket(String ticket) { this.ticket = ticket; } }
public enum HttpCodeEnum { /** * 200请求成功 */ OK(200, "请求成功"), /** * 500控制器出错 */ SERVER_ERROR(500, "服务器内部抛出错误"), private final Integer value; private final String desc; private HttpCodeEnum(Integer value, String desc) { this.value = value; this.desc = desc; } /** * Return the integer value of this status code. */ public Integer value() { return this.value; } public String msg() { //return Resources.getMessage("HTTPCODE_" + this.value); return this.desc; } public String toString() { return this.value.toString(); } }
public interface AddGroup { }
public interface EditGroup { }
public interface PageGroup { }
public class BaseQuery implements Serializable { private static final long serialVersionUID = -6893838165058397865L; private String fromDate; private String toDate; @Min(value = 1, message = "分页记录数不能为空" , groups = {PageGroup.class}) private int rows; @Min(value = 1, message = "分页页码不能为空" , groups = {PageGroup.class}) private int page; public String getFromDate() { return fromDate; } public void setFromDate(String fromDate) { this.fromDate = fromDate; } public String getToDate() { return toDate; } public void setToDate(String toDate) { this.toDate = toDate; } public int getRows() { return rows; } public void setRows(int rows) { this.rows = rows; } public int getPage() { return page; } public void setPage(int page) { this.page = page; } public void convertToLongTime() { if (StrUtil.isNotBlank(fromDate) && StrUtil.isNotBlank(toDate)) { this.fromDate = DateUtil.parse(fromDate, ConfigConsts.NORM_DATE_PATTERN).toDateStr() + " 00:00:00"; this.toDate = DateUtil.parse(toDate, ConfigConsts.NORM_DATE_PATTERN).toDateStr() + " 23:59:59"; ; } } }
public interface ConfigConsts { static final String NORM_DATE_PATTERN = "yyyy-MM-dd"; }