项目背景

今天小刀和各位小伙伴们一起来深入探讨下模板使用的问题,我们来研究下怎么去封装一个通用的 Mybatis 模板,让简单的增删改查操作,直接通过我们的代码生成工具就能解决,让工具从demo级别上升到生产可用的项目

mybatis简单介绍

相信很多小伙伴都已经用过Mybatis了,很简单方便, 在springboot的集成环境里面,写个 DAO 接口,加上注解,然后写个xml和DAO关联起来,然后xml里面方法和DAO里面的方法关联起来,然后就可以通过调用DAO方法的形式来调用这个sql,并获得相应的返回结果,代码结构如图下所示:

这是以往的方式,在新版Mybatis中,我们有了一个新的选择,用java类和注解的方式去完成一个sql ,官方文档如下:

写法如下:

只建议简单的sql用注解的方式去处理,复杂的还是要手写sql , 可以重新建一个DAO , 或者在provider里面用字符串拼接的方式去完成

具体用法,大家可以参考上面的官方文档中的地址

设计封装的方法

封装规则要求

我们总说,我们一直都在做重复的增删改查工作,现在我们就可以把这些基本的操作都封装起来,把更多的精力放在应用高可用,高响应,业务逻辑的梳理上面

因此 ,我们的封装也是以增删改查四大方向为主.

在封装之前,我们先约定以下规定:

1.数据库对应的实体对象,我们用DO(domain object)来表示

2.数据库对应的查询对应,我们用QC(query object)来表示

3.各种操作方法的命名，如，getModel,listModels,saveModels等等

这样把数据库实体对象和查询对象分开，因为我们的在加查询条件的时候，比如通过idList去查询，如果在DO里面新加字段就不太好，所以抽象了一个QC的概念，专门做数据库查询对象。

从前到后的对应关系

为什么我们项目可以用模板生成工具进行生成呢,大家仔细研究下手中的项目可以发现我们的项目从controller开始,到最后的DAO其实都是有名字对应起来的,我们以查询为例: listItems,通过这个名字我们可以直观的看到,是获取商品列表的.

Controller中的代码:

@GetMapping("/item/query")
 @ResponseBody
 public BaseResponse<PageData<ItemDO>> queryItem(ItemDO item, Integer pageIndex , Integer pageSize){
 List<ItemDO> items = itemManager.queryItem(item, pageIndex, pageSize);
 Long totalCount = itemManager.countItem(item);
 pageIndex = pageIndex == null?1:pageIndex;
 pageSize = pageSize == null?20:pageSize;
 return BaseResponse.assemblePageResponse(items,totalCount,pageIndex,pageSize);
 }
 

模板:

@GetMapping("/[(${table.javaCamelName})]/query")
 @ResponseBody
 public BaseResponse<PageData<[(${table.javaTableName})]DO>> query[(${table.javaTableName})]([(${table.javaTableName})]DO [(${table.javaCamelName})], Integer pageIndex , Integer pageSize){
 List<[(${table.javaTableName})]DO> [(${table.javaCamelName})]s = [(${table.javaCamelName})]Manager.query[(${table.javaTableName})]([(${table.javaCamelName})], pageIndex, pageSize);
 Long totalCount = [(${table.javaCamelName})]Manager.count[(${table.javaTableName})]([(${table.javaCamelName})]);
 pageIndex = pageIndex == null?1:pageIndex;
 pageSize = pageSize == null?20:pageSize;
 return BaseResponse.assemblePageResponse([(${table.javaCamelName})]s,totalCount,pageIndex,pageSize);
 }
 

Manager中的代码:

public List<ItemDO> queryItem(ItemDO item, int pageIndex , int pageSize);
 

模板:

public List<[(${table.javaTableName})]DO> query[(${table.javaTableName})]([(${table.javaTableName})]DO [(${table.javaCamelName})], int pageIndex , int pageSize);

ManagerImpl中的代码:

@Override
 public List<ItemDO> queryItem(ItemDO item, int pageIndex , int pageSize){
 Page page = new Page(pageIndex,pageSize);
 ItemQC qc = new ItemQC();
 qc.setPage(page);
 BeanUtils.copyProperties(item,qc);
 List<ItemDO> items = itemMapper.listItems(qc);
 return items;
 }
 

模板:

@Override
 public List<[(${table.javaTableName})]DO> query[(${table.javaTableName})]([(${table.javaTableName})]DO [(${table.javaCamelName})], int pageIndex , int pageSize){
 Page page = new Page(pageIndex,pageSize);
 [(${table.javaTableName})]QC qc = new [(${table.javaTableName})]QC();
 qc.setPage(page);
 BeanUtils.copyProperties([(${table.javaCamelName})],qc);
 List<[(${table.javaTableName})]> [(${table.javaCamelName})]s = [(${table.javaCamelName})]Mapper.list[(${table.javaTableName})]s(qc);
 return [(${table.javaCamelName})]s;
 }
 

Mapper中的代码:

@SelectProvider(type = ItemProvider.class)
 List<ItemDO> listItems(ItemQC itemQC);
 

模板:

@SelectProvider(type = [(${table.javaTableName})]Provider.class)

List<[(${table.javaTableName})]DO> list[(${table.javaTableName})]s([(${table.javaTableName})]QC [(${table.javaCamelName})]QC);

Provider中的代码:

public String listItems(ItemQC itemQC){
 SQL sql = new SQL() {{
 SELECT(TABLE_FIELDS);
 FROM("item");
 }};
 MapperUtils.richWhereSql(sql, itemQC);
 return sql. toString ();
 }
 

模板:

public String list[(${table.javaTableName})]s([(${table.javaTableName})]QC [(${table.javaCamelName})]QC){
 SQL sql = new SQL() {{
 SELECT(TABLE_FIELDS);
 FROM("[(${table.javaCamelName})]");
 }};
 MapperUtils.richWhereSql(sql, [(${table.javaCamelName})]QC);
 return sql.toString();
 }
 

这样我们从前到后看过来,发其实都是围绕着一个关键词在走: Item , 有时候把他变成大写的,有时候把他变成驼峰命名,有时候把他加个DO后缀,有时候把他加个Mapper后缀.

生成文件到对应目录中

在上篇文章中,我们只是生成了对应的字符串,但是在实际开发中,我们是需要生成文件的,最好是能在对应文件夹中.

所以这就有两步,1是生成文件.2是找到对应文件夹

生成对应文件

生成文件这个应该是很简单的,直接一句代码搞定:

 Files.write(Paths.get(pathMap.get("model")+tableInfo.getJavaTableName() + "DO.java"),model.getBytes());
 Files.write(Paths.get(pathMap.get("mapper")+tableInfo.getJavaTableName() + "Mapper.java"),dao.getBytes());
 Files.write(Paths.get(pathMap.get("provider")+tableInfo.getJavaTableName() + "Provider.java"),provider.getBytes());
 Files.write(Paths.get(pathMap.get("manager")+tableInfo.getJavaTableName() + "Manager.java"),manager.getBytes());
 Files.write(Paths.get(pathMap.get("managerImpl")+tableInfo.getJavaTableName() + "ManagerImpl.java"),managerImpl.getBytes());
 Files.write(Paths.get(pathMap.get("controller")+tableInfo.getJavaTableName() + "Controller.java"),controller.getBytes());
 
 

输出到对应文件夹中

这一步,就要我们明确,我们的每个文件是放在什么地方在,如Mapper/Provider是查询数据库用的, 我们把他放在business-impl模块中的dao文件夹中,下面小刀把自己项目中放的位置列出来和大家分享下,大家有好的想法可以一起交流:

XXXDO放在 business 模块中 models.dos文件夹中

XXXController 放在admin 模块中 controller 文件夹中

XXXManager 放在business模块中 business文件夹中

XXXManagerImpl 放在business-impl模块中 business.impl文件夹中

XXXMapper/XXXProvider 放在 business-impl模块中 dao 文件夹中

这样我们就可以用代码去定位到对应的文件夹中了:

 // 获得当前项目的文件夹
 String currentDir = System.getProperty("user.dir");
 currentDir=currentDir.substring(0,currentDir.lastIndexOf("/"));
 pathMap.put("model",currentDir);
 pathMap.put("mapper",currentDir);
 pathMap.put("provider",currentDir);
 pathMap.put("manager",currentDir);
 pathMap.put("managerImpl",currentDir);
 pathMap.put("controller",currentDir);
 //如果设置了自动路径,就YY一下路径
 if(autoFilePath){
 Files.list(Paths.get(currentDir)).forEach(file->{
 if(Files.isDirectory(file)){
 String fileName = file.toString();
 if(fileName.endsWith("admin")){
 pathMap.put("controller",fileName+"/src/main/java/"+packageName.replaceAll("\\.","/")+"/admin/controller/");
 }
 if(fileName.endsWith("business")){
 pathMap.put("model",fileName+"/src/main/java/"+packageName.replaceAll("\\.","/")+"/models/dos/");
 pathMap.put("manager",fileName+"/src/main/java/"+packageName.replaceAll("\\.","/")+"/business/manager/");
 }
 if(fileName.endsWith("business-impl")){
 pathMap.put("managerImpl",fileName+"/src/main/java/"+packageName.replaceAll("\\.","/")+"/business/manager/impl/"+domainName+"/");
 pathMap.put("mapper",fileName+"/src/main/java/"+packageName.replaceAll("\\.","/")+"/dao/"+domainName+"/");
 pathMap.put("provider",fileName+"/src/main/java/"+packageName.replaceAll("\\.","/")+"/dao/"+domainName+"/");
 }
 }
 });
 }
 

这样我们就可以做到,把增删改查从controller到dao直接一键生成到对应文件夹啦,只要把表建好,剩下的事就一步搞定啦

最后说两句

这个工具一般是做后台管理页面,增删改查的时候用, 业务逻辑不建议用工具类生成,业务逻辑一般是提供dubbo接口出来,不要直接把manager中的增删改查提供出去,提供出去的一定是要先有业务,再有接口,然后在serviceImpl中调用manager去完成业务逻辑

大家有什么想法,欢迎留言和小刀一起交流