1.導包
<!--mybatisplus的包-->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>2.2.0</version>
</dependency>
<!--連接數據庫-->
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>5.1.6</version>
</dependency>
<!-- druid連接池的包 -->
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid</artifactId>
<version>1.1.10</version>
</dependency>
<!--模板引擎-->
<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity-engine-core</artifactId>
<version>2.0</version>
</dependency>
2.寫一個properties文件
#代碼生成好之后,輸出路徑 ,基本輸出路徑 OutputDir=F:/IDEAWorkSpace/hrm-parent/hrm-systemmange-parnet/hrm-systemmange-service-2010/src/main/java #mapper.xml SQL映射文件目錄 OutputDirXml=F:/IDEAWorkSpace/hrm-parent/hrm-systemmange-parnet/hrm-systemmange-service-2010/src/main/resources #我們需要的公共包的特殊路徑 OutputDirCommon=F:/IDEAWorkSpace/hrm-parent/hrm-systemmange-parnet/hrm-systemmange-common/src/main/java #設置作者 author=jiedada #自定義包路徑 parent=cn.jiedada.hrm #數據庫連接信息 jdbc.url=jdbc:mysql:///hrm-systemmanage jdbc.user=root jdbc.pwd=123456 jdbc.driver=com.mysql.jdbc.Driver
在resources下面我們還自己創建了一個controller和query
templates
controller.java.vm
package ${package.Controller}; import ${package.Service}.${table.serviceName}; import ${package.Entity}.${entity}; import cn.itsource.hrm.query.${entity}Query; import cn.itsource.hrm.util.AjaxResult; import cn.itsource.hrm.util.PageList; import com.baomidou.mybatisplus.plugins.Page; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import java.util.List; @RestController @RequestMapping("/${table.entityPath}") public class ${entity}Controller { @Autowired public ${table.serviceName} ${table.entityPath}Service; /** * 保存和修改公用的 * @param ${table.entityPath} 傳遞的實體 * @return Ajaxresult轉換結果 */ @RequestMapping(value="/save",method= RequestMethod.POST) public AjaxResult save(@RequestBody ${entity} ${table.entityPath}){ try { if(${table.entityPath}.getId()!=null){ ${table.entityPath}Service.updateById(${table.entityPath}); }else{ ${table.entityPath}Service.insert(${table.entityPath}); } return AjaxResult.me(); } catch (Exception e) { e.printStackTrace(); return AjaxResult.me().setMessage("保存對象失敗！"+e.getMessage()); } } /** * 刪除對象信息 * @param id * @return */ @RequestMapping(value="/{id}",method=RequestMethod.DELETE) public AjaxResult delete(@PathVariable("id") Long id){ try { ${table.entityPath}Service.deleteById(id); return AjaxResult.me(); } catch (Exception e) { e.printStackTrace(); return AjaxResult.me().setMessage("刪除對象失敗！"+e.getMessage()); } } //獲取用戶 @RequestMapping(value = "/{id}",method = RequestMethod.GET) public ${entity} get(@PathVariable("id")Long id) { return ${table.entityPath}Service.selectById(id); } /** * 查看所有的員工信息 * @return */ @RequestMapping(value = "/list",method = RequestMethod.GET) public List<${entity}> list(){ return ${table.entityPath}Service.selectList(null); } /** * 分頁查詢數據 * * @param query 查詢對象 * @return PageList 分頁對象 */ @RequestMapping(value = "/pagelist",method = RequestMethod.POST) public PageList<${entity}> json(@RequestBody ${entity}Query query) { Page<${entity}> page = new Page<${entity}>(query.getPage(),query.getRows()); page = ${table.entityPath}Service.selectPage(page); return new PageList<${entity}>(page.getTotal(),page.getRecords()); } }
query.java.vm
package cn.itsource.hrm.query; /** * * @author ${author} * @since ${date} */ public class ${table.entityName}Query extends BaseQuery{ }
3.使用一個主配置類
通過該類生成我們需要的文件，及文件位置（檢查每個配置類，如果有需要這自己寫入AjaxRsult,PageList,BaseQuery等分頁工具）
package cn.jiedada; import com.baomidou.mybatisplus.generator.AutoGenerator; import com.baomidou.mybatisplus.generator.InjectionConfig; import com.baomidou.mybatisplus.generator.config.*; import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert; import com.baomidou.mybatisplus.generator.config.po.TableInfo; import com.baomidou.mybatisplus.generator.config.rules.DbType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import java.util.*; /** * Created by CDHong on 2018/4/6. * 通過主方法生成我們的mapper，service，controller */ public class GenteratorCode { public static void main(String[] args) throws InterruptedException { //用來獲取Mybatis-Plus.properties文件的配置信息 ResourceBundle rb = ResourceBundle.getBundle("mybatiesplus-config"); //不要加后綴 AutoGenerator mpg = new AutoGenerator(); // 全局配置 GlobalConfig gc = new GlobalConfig(); gc.setOutputDir(rb.getString("OutputDir")); gc.setFileOverride(true); gc.setActiveRecord(true);// 開啟 activeRecord 模式 gc.setEnableCache(false);// XML 二級緩存 gc.setBaseResultMap(true);// XML ResultMap gc.setBaseColumnList(false);// XML columList gc.setAuthor(rb.getString("author")); mpg.setGlobalConfig(gc); // 數據源配置 DataSourceConfig dsc = new DataSourceConfig(); dsc.setDbType(DbType.MYSQL); dsc.setTypeConvert(new MySqlTypeConvert()); dsc.setDriverName(rb.getString("jdbc.driver")); dsc.setUsername(rb.getString("jdbc.user")); dsc.setPassword(rb.getString("jdbc.pwd")); dsc.setUrl(rb.getString("jdbc.url")); mpg.setDataSource(dsc); // 策略配置 StrategyConfig strategy = new StrategyConfig(); strategy.setTablePrefix(new String[] { "t_" });// 此處可以修改為您的表前綴 strategy.setNaming(NamingStrategy.underline_to_camel);// 表名生成策略 strategy.setInclude(new String[]{"t_systemdictionary","t_systemdictionaryitem"}); // 需要生成的表 mpg.setStrategy(strategy); // 包配置 PackageConfig pc = new PackageConfig(); pc.setParent(rb.getString("parent")); pc.setController("web.controller"); //cn.itsource.hrm.web.controller pc.setService("service"); pc.setServiceImpl("service.impl"); pc.setEntity("domain"); pc.setMapper("mapper"); mpg.setPackageInfo(pc); // 注入自定義配置，可以在 VM 中使用 cfg.abc 【可無】 InjectionConfig cfg = new InjectionConfig() { @Override public void initMap() { Map<String, Object> map = new HashMap<String, Object>(); map.put("abc", this.getConfig().getGlobalConfig().getAuthor() + "-rb"); this.setMap(map); } }; List<FileOutConfig> focList = new ArrayList<FileOutConfig>(); // 調整 controller 生成目錄演示 focList.add(new FileOutConfig("/templates/controller.java.vm") { @Override public String outputFile(TableInfo tableInfo) { //輸出的位置 return rb.getString("OutputDir")+ "/cn/jiedada/hrm/web/controller/" + tableInfo.getEntityName() + "Controller.java"; } }); // 調整 query 生成目錄演示 focList.add(new FileOutConfig("/templates/query.java.vm") { @Override public String outputFile(TableInfo tableInfo) { //輸出的位置 return rb.getString("OutputDirCommon")+ "/cn/jiedada/hrm/query/" + tableInfo.getEntityName() + "Query.java"; } }); // 調整 domain 生成目錄演示 focList.add(new FileOutConfig("/templates/entity.java.vm") { @Override public String outputFile(TableInfo tableInfo) { //輸出的位置 return rb.getString("OutputDirCommon")+ "/cn/jiedada/hrm/domain/" + tableInfo.getEntityName() + ".java"; } }); // 調整 xml 生成目錄演示 focList.add(new FileOutConfig("/templates/mapper.xml.vm") { @Override public String outputFile(TableInfo tableInfo) { return rb.getString("OutputDirXml")+ "/cn/jiedada/hrm/mapper/" + tableInfo.getEntityName() + "Mapper.xml"; } }); cfg.setFileOutConfigList(focList); mpg.setCfg(cfg); // 自定義模板配置，可以 copy 源碼 mybatis-plus/src/main/resources/templates 下面內容修改， // 放置自己項目的 src/main/resources/templates 目錄下, 默認名稱一下可以不配置，也可以自定義模板名稱 TemplateConfig tc = new TemplateConfig(); tc.setService("/templates/service.java.vm"); tc.setServiceImpl("/templates/serviceImpl.java.vm"); tc.setEntity(null); //domain tc.setMapper("/templates/mapper.java.vm"); tc.setController(null); tc.setXml(null); // 如上任何一個模塊如果設置 空 OR Null 將不生成該模塊。 mpg.setTemplate(tc); // 執行生成 mpg.execute(); } }