MyBatis代碼生成器MyBatisPlus-generator(自定義模板引擎)

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文件

建立配置文件 resources/mybatiesplus-config.propertiesjava

#代碼生成好以後,輸出路徑 ,基本輸出路徑 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和querymysql

templatesweb

controller.java.vmspring

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.vmsql

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(); } }
View Code
相關文章
相關標籤/搜索