Mapper 接口

基于 mybatis-mapper/provider 核心部分实现的基础的增删改查操作,提供了一个核心的 io.mybatis.mapper.Mapper 接口,接口定义如下:

  1. /**
  2. * 基础 Mapper 方法,可以在此基础上继承覆盖已有方法
  3. *
  4. * @param <T> 实体类类型
  5. * @param <I> 主键类型
  6. * @author liuzh
  7. */
  8. public interface Mapper<T, I extends Serializable>
  9. extends EntityMapper<T, I>, ExampleMapper<T, Example<T>>, CursorMapper<T, Example<T>> {
  10. /**
  11. * 保存实体,默认主键自增,并且名称为 id
  12. * <p>
  13. * 这个方法是个示例,你可以在自己的接口中使用相同的方式覆盖父接口中的配置
  14. *
  15. * @param entity 实体类
  16. * @return 1成功,0失败
  17. */
  18. @Override
  19. @Lang(Caching.class)
  20. @Options(useGeneratedKeys = true, keyProperty = "id")
  21. @InsertProvider(type = EntityProvider.class, method = "insert")
  22. int insert(T entity);
  23. /**
  24. * 保存实体中不为空的字段,默认主键自增,并且名称为 id
  25. * <p>
  26. * 这个方法是个示例,你可以在自己的接口中使用相同的方式覆盖父接口中的配置
  27. *
  28. * @param entity 实体类
  29. * @return 1成功,0失败
  30. */
  31. @Override
  32. @Lang(Caching.class)
  33. @Options(useGeneratedKeys = true, keyProperty = "id")
  34. @InsertProvider(type = EntityProvider.class, method = "insertSelective")
  35. int insertSelective(T entity);
  36. /**
  37. * 根据主键更新实体中不为空的字段,强制字段不区分是否 null,都更新
  38. * <p>
  39. * 当前方法来自 {@link io.mybatis.mapper.fn.FnMapper},该接口中的其他方法用 {@link ExampleMapper} 也能实现
  40. *
  41. * @param entity 实体类
  42. * @param forceUpdateFields 强制更新的字段,不区分字段是否为 null,通过 {@link Fn#of(Fn...)} 创建 {@link Fn.Fns}
  43. * @return 1成功,0失败
  44. */
  45. @Lang(Caching.class)
  46. @UpdateProvider(type = FnProvider.class, method = "updateByPrimaryKeySelectiveWithForceFields")
  47. int updateByPrimaryKeySelectiveWithForceFields(@Param("entity") T entity, @Param("fns") Fn.Fns<T> forceUpdateFields);
  48. /**
  49. * 根据指定字段集合查询:field in (fieldValueList)
  50. * <p>
  51. * 这个方法是个示例,你也可以使用 Java8 的默认方法实现一些通用方法
  52. *
  53. * @param field 字段
  54. * @param fieldValueList 字段值集合
  55. * @param <F> 字段类型
  56. * @return 实体列表
  57. */
  58. default <F> List<T> selectByFieldList(Fn<T, F> field, List<F> fieldValueList) {
  59. Example<T> example = new Example<>();
  60. example.createCriteria().andIn((Fn<T, Object>) field, fieldValueList);
  61. return selectByExample(example);
  62. }
  63. /**
  64. * 根据指定字段集合删除:field in (fieldValueList)
  65. * <p>
  66. * 这个方法是个示例,你也可以使用 Java8 的默认方法实现一些通用方法
  67. *
  68. * @param field 字段
  69. * @param fieldValueList 字段值集合
  70. * @param <F> 字段类型
  71. * @return 实体列表
  72. */
  73. default <F> int deleteByFieldList(Fn<T, F> field, List<F> fieldValueList) {
  74. Example<T> example = new Example<>();
  75. example.createCriteria().andIn((Fn<T, Object>) field, fieldValueList);
  76. return deleteByExample(example);
  77. }
  78. }

这个接口展示了好几个通用方法的特点:

  1. 可以继承其他通用接口
  2. 可以重写继承接口的定义
  3. 可以直接复制其他接口中的通用方法定义
  4. 可以使用 Java8 默认方法灵活实现通用方法

在下面内容中,还能看到一个特点,“5. 那就是一个 provider 实现,通过修改接口方法的返回值和入参,就能变身无数个通用方法”,通用方法的实现极其容易。

下面开始详细介绍这些特性。

2.1 继承其他通用接口

上面接口定义中,继承了 EntityMapper, ExampleMapperCursorMapper 接口。这些接口中定义了大量的通用方法, 通过继承使得 Mapper 接口获得了大量的通用方法,通过继承可以组合不同类别的方法。

除了这里继承的 3 个接口外,还有其他几个接口,一并简单介绍如下。

2.1.1 io.mybatis.mapper.base.EntityMapper<T, I>

实体类最基础的通用方法:

  • int insert(T entity): 保存实体
  • int insertSelective(T entity): 保存实体中不为空的字段
  • int deleteByPrimaryKey(I id): 根据主键删除
  • int delete(T entity): 保存实体信息批量删除
  • int updateByPrimaryKey(T entity): 根据主键更新实体
  • int updateByPrimaryKeySelective(T entity): 根据主键更新实体中不为空的字段
  • Optional<T> selectByPrimaryKey(I id): 根据主键查询实体
  • Optional<T> selectOne(T entity): 根据实体字段条件查询唯一的实体
  • List<T> selectList(T entity): 根据实体字段条件批量查询
  • int selectCount(T entity): 根据实体字段条件查询总数

这个接口中返回值特殊的有两个 Optional<T>,使用的 Java8 中的 Optional 类型,表明接口返回值可能为空, 使用时应该判断或使用 Optional 提供的 orXX 方法,比如下面几种情况:

  1. 返回值不能为空,为空则抛出异常:
    1. Optional<User> userOptional = entityMapper.selectByPrimaryKey(user.getId());
    2. return userOptional.orElseThrow(() -> new RuntimeException("数据不存在"));
  2. 如果不存在,新建一个:
    1. Optional<User> userOptional = entityMapper.selectByPrimaryKey(user.getId());
    2. return userOptional.orElseGet(User::new)
  3. 如果不存在,直接返回 null:
    1. Optional<User> userOptional = entityMapper.selectByPrimaryKey(user.getId());
    2. return userOptional.orElse(null);

真正使用 Optional 的时候,要尽可能避免以前 obj != null 这种判断思维,避免使用 Optional.isPresent() 判断,才能真正掌握 Optional 的用法。

如果你不喜欢上面返回值为 Optional,就想返回 T,后面第 2.5 会介绍实现方式。

2.1.2 io.mybatis.mapper.example.ExampleMapper<T, E>

Example 类相关的查询方法,Example 中扩展了一些新的方法,因此默认的 Example 查询已经不支持 MBG 生成的 XXExample 对象作为参数。

先看接口中包含的方法:

  • default Example<T> example(): 返回 Example<T> 对象,自己也可以直接 new
  • int deleteByExample(E example): 根据 Example 删除
  • int updateByExample(@Param("entity") T entity, @Param("example") E example): 根据 Example 条件批量更新实体信息
  • int updateByExampleSelective(@Param("entity") T entity, @Param("example") E example);: 根据 Example 条件批量更新实体不为空的字段
  • List<T> selectByExample(E example): 根据 Example 条件批量查询
  • Optional<T> selectOneByExample(E example): 根据 Example 条件查询单个实体
  • int countByExample(E example): 根据 Example 条件查询总数

接口源码中还注释了几个方法,这些方式是用来说明 2.4 的特点,这里先不介绍。

看完接口方法,在看看 Example 对象中增加了那些特殊的字段和方法:

  1. public class Example<T> {
  2. /**
  3. * 排序字段
  4. */
  5. protected String orderByClause;
  6. /**
  7. * 是否使用 distinct
  8. */
  9. protected boolean distinct;
  10. /**
  11. * 指定查询列
  12. */
  13. protected String selectColumns;
  14. /**
  15. * 起始 SQL,添加到 SQL 前,注意防止 SQL 注入
  16. */
  17. protected String startSql;
  18. /**
  19. * 结尾 SQL,添加到 SQL 最后,注意防止 SQL 注入
  20. */
  21. protected String endSql;
  22. /**
  23. * 多组条件通过 OR 连接
  24. */
  25. protected List<Criteria<T>> oredCriteria;
  26. /**
  27. * 允许Example查询条件为空
  28. */
  29. protected boolean allowCriteriaEmpty;
  30. //省略其他
  31. }

2.1.2.1 orderByClause

排序字段是默认就有的,但是为了字段使用的安全,增加了额外的赋值方法:

  1. /**
  2. * 通过方法引用方式设置排序字段
  3. *
  4. * @param fn 排序列的方法引用
  5. * @param order 排序方式
  6. * @return Example
  7. */
  8. public Example<T> orderBy(Fn<T, Object> fn, Order order) {
  9. if (orderByClause == null) {
  10. orderByClause = "";
  11. }
  12. orderByClause += fn.toColumn() + " " + order;
  13. return this;
  14. }

示例:

  1. example.orderBy(User::getId, Example.Order.DESC);

这会生成 ORDER BY id desc 排序。

2.1.2.2 distinct

去重 distinct 也是默认的。如果设置为 true,查询时就会使用 SELECT DISTINCT ...

2.1.2.3 selectColumns

selectColumns 是新增的,可以用来指定查询列,对应的方法如下:

  1. /**
  2. * 指定查询列
  3. *
  4. * @param fns 方法引用
  5. */
  6. public Example<T> selectColumns(Fn<T, Object>... fns) {
  7. if (selectColumns == null) {
  8. selectColumns = "";
  9. }
  10. for (Fn<T, Object> fn : fns) {
  11. String column = fn.toColumn();
  12. String field = fn.toField();
  13. if (selectColumns.length() != 0) {
  14. selectColumns += ",";
  15. }
  16. if (column.equals(field)) {
  17. selectColumns += column;
  18. } else {
  19. selectColumns += column + " AS " + field;
  20. }
  21. }
  22. return this;
  23. }

用法示例:

  1. <User> example = exampleMapper.example();
  2. example.selectColumns(User::getUserName, User::getSex);
  3. //可以多次调用追加查询列
  4. example.selectColumns(User::getId);

指定查询列后,在 SQL 中会变成 select name as userName, sex, id from ...userName 映射的 name)。

2.1.2.4 startSql

新增加的字段,起始 SQL,添加到 SQL 前,注意防止 SQL 注入。

直接通过 set 方法设置值:

  1. /**
  2. * 设置起始 SQL
  3. *
  4. * @param startSql 起始 SQL,添加到 SQL 前,注意防止 SQL 注入
  5. */
  6. public void setStartSql(String startSql) {
  7. this.startSql = startSql;
  8. }

这个字段基本上要配合下面的 endSql 字段一起用,下面示例:

  1. example.setStartSql("select * from (");
  2. example.setEndSql(") tmp limit 1");

示例没太大意义,你可以想个更好的例子来更新此处文档。

2.1.2.5 endSql

和上面新增加的字段类似:结尾 SQL,添加到 SQL 最后,注意防止 SQL 注入

也是直接通过 set 方法设置:

  1. /**
  2. * 设置结尾 SQL
  3. *
  4. * @param endSql 结尾 SQL,添加到 SQL 最后,注意防止 SQL 注入
  5. */
  6. public void setEndSql(String endSql) {
  7. this.endSql = endSql;
  8. }

示例看前一个字段的。

2.1.2.6 oredCriteria

默认就有的字段,记录查询条件块。

2.1.2.7 allowCriteriaEmpty

这个字段是新增的,这个字段破坏了 Example 的整体结构,主要目的是在使用时决定:是否允许查询条件为空

当 select 查询时,是可以为空的,这种情况就是查询了全表,由于这个字段默认值为 false 不允许,因此你需要 new Example(true) 才行。

当 delete, update 时,默认不允许为空,这可以避免清库和更新全库。这个参数只在下面方法用到:

  1. /**
  2. * 获取所有条件,当前方法会校验所有查询条件,如果不存在查询条件就抛出异常。
  3. * <p>
  4. * 不允许通过 Example 相关方法操作全表!!!
  5. *
  6. * @return 条件
  7. */
  8. public List<Criteria<T>> getOredCriteria() {
  9. if (!allowCriteriaEmpty) {
  10. if (oredCriteria.size() == 0) {
  11. throw new IllegalArgumentException("Example 条件不能为空");
  12. }
  13. boolean noCriteria = true;
  14. for (Criteria<T> criteria : oredCriteria) {
  15. if (!criteria.getCriteria().isEmpty()) {
  16. noCriteria = false;
  17. break;
  18. }
  19. }
  20. if (noCriteria) {
  21. throw new IllegalArgumentException("Example 条件不能为空");
  22. }
  23. }
  24. return oredCriteria;
  25. }

当获取查询条件时判断,如果没有有效的查询条件,就抛出异常,避免潜在的风险。

2.1.2.7 clear() 方法

最后说一个方法,同一个 Example 对象经过 clear() 后是可以反复使用的:

  1. /**
  2. * 清除所有设置
  3. */
  4. public void clear() {
  5. oredCriteria.clear();
  6. orderByClause = null;
  7. distinct = false;
  8. selectColumns = null;
  9. startSql = null;
  10. endSql = null;
  11. allowCriteriaEmpty = false;
  12. }

特别注意的是 allowCriteriaEmpty,即使你用 Example(true) 创建的,这里也会设置为 false,想要设置 true 可以调用下面方法:

  1. /**
  2. * 设置是否允许查询条件为空
  3. *
  4. * @param allowCriteriaEmpty true允许空,一般用于查询,false不允许空,一般用于修改和删除
  5. */
  6. public void allowCriteriaEmpty(boolean allowCriteriaEmpty) {
  7. this.allowCriteriaEmpty = allowCriteriaEmpty;
  8. }

2.1.2.8 简单示例

介绍完所有字段和一个特殊方法,下面看几个 Example 示例:

  1. Example<User> example = new Example();
  2. example.createCriteria().andGreaterThan(User::getId, 10L).andLike(User::getUserName, "殷%");
  3. Assert.assertEquals(3, exampleMapper.deleteByExample(example));
  4. example.clear();
  5. example.createCriteria().andEqualTo(User::getId, 1L);
  6. User user = new User();
  7. user.setId(1L);
  8. user.setUserName("男主角");
  9. Assert.assertEquals(1, exampleMapper.updateByExample(user, example));

剩下的,你需要掌握的就是 andGreaterThanandEqualTo 这类的方法,这些方法通过名字可以直接理解。

2.1.3 io.mybatis.mapper.cursor.CursorMapper<T, E>

游标相关的功能,是在 2016年初的 MyBatis 3.3.1 中增加的,当你想逐个从数据库取值时可以使用。

通用方法中,仅提供了两个方法示例:

  • Cursor<T> selectCursor(T entity): 根据实体字段条件查询
  • Cursor<T> selectCursorByExample(E example):根据 Example 条件查询

这两个方法就提现了 “5. 那就是一个 provider 实现,通过修改接口方法的返回值和入参,就能变身无数个通用方法”, 这两个接口使用了 EntityProviderExampleProvider 中的现成实现,只是把返回值从 List<T> 改成了 Cursor<T>

一个简单的示例如下:

  1. @Test
  2. public void testSelectCursor() {
  3. SqlSession sqlSession = getSqlSession();
  4. try {
  5. CursorMapper<User, Example<User>> mapper = sqlSession.getMapper(UserMapper.class);
  6. User user = new User();
  7. user.setSex("女");
  8. Cursor<User> userCursor = mapper.selectCursor(user);
  9. Iterator<User> userIterator = userCursor.iterator();
  10. int count = 0;
  11. while (userIterator.hasNext()) {
  12. count++;
  13. User u = userIterator.next();
  14. System.out.println(u.getUserName());
  15. Assert.assertEquals(count, userCursor.getCurrentIndex() + 1);
  16. }
  17. Assert.assertEquals(16, count);
  18. Assert.assertTrue(userCursor.isConsumed());
  19. } finally {
  20. //不要忘记关闭sqlSession
  21. sqlSession.close();
  22. }
  23. }

2.1.4 io.mybatis.mapper.fn.FnMapper<T>

这个接口主要目的是为了演示如果通过传入字段改变查询的逻辑。接口包含下面几个方法:

  1. public interface FnMapper<T> {
  2. /**
  3. * 根据主键更新实体中不为空的字段,强制字段不区分是否 null,都更新
  4. *
  5. * @param entity 实体类
  6. * @param forceUpdateFields 强制更新的字段,不区分字段是否为 null,通过 {@link Fn#of(Fn...)} 创建 {@link Fn.Fns}
  7. * @return 1成功,0失败
  8. */
  9. @Lang(Caching.class)
  10. @UpdateProvider(type = FnProvider.class, method = "updateByPrimaryKeySelectiveWithForceFields")
  11. int updateByPrimaryKeySelectiveWithForceFields(@Param("entity") T entity, @Param("fns") Fn.Fns<T> forceUpdateFields);
  12. /**
  13. * 根据实体字段条件查询唯一的实体({@link io.mybatis.mapper.example.ExampleMapper} 可以实现一样的功能,当前方法只是示例)
  14. *
  15. * @param entity 实体类
  16. * @param selectFileds 查询的字段,不区分字段是否为 null,通过 {@link Fn#of(Fn...)} 创建 {@link Fn.Fns}
  17. * @return 单个实体,查询结果由多条时报错
  18. */
  19. @Lang(Caching.class)
  20. @SelectProvider(type = FnProvider.class, method = "selectColumns")
  21. Optional<T> selectColumnsOne(@Param("entity") T entity, @Param("fns") Fn.Fns<T> selectFileds);
  22. /**
  23. * 根据实体字段条件批量查询({@link io.mybatis.mapper.example.ExampleMapper} 可以实现一样的功能,当前方法只是示例)
  24. *
  25. * @param entity 实体类
  26. * @param selectFileds 查询的字段,不区分字段是否为 null,通过 {@link Fn#of(Fn...)} 创建 {@link Fn.Fns}
  27. * @return 实体列表
  28. */
  29. @Lang(Caching.class)
  30. @SelectProvider(type = FnProvider.class, method = "selectColumns")
  31. List<T> selectColumns(@Param("entity") T entity, @Param("fns") Fn.Fns<T> selectFileds);
  32. }

第一个方法名有点长,用列表展示不太合适了。

这几个方法,都可以通过传入字段改变行为,并且第一个名字最长的方法,直接被复制到了 Mapper 接口的定义中,这就是第 3 点要说明的内容, 实际上你可以复制粘贴的方式构造一个自己的基类 Mapper。2.3 在介绍。

名字最长这个方法是许多人想要的一个方法,在选择更新的情况下还能指定必须更新的字段,这个方法的示例如下:

  1. User user = mapper.selectByPrimaryKey(1L).get();
  2. user.setUserName(null);
  3. int count = mapper.updateByPrimaryKeySelectiveWithForceFields(user, Fn.of(User::getUserName));

单纯从 Mapper 接口调用来看,这个方法名字太长,写法也有点别扭,等看到后续 Service 层封装时, 会掩盖这些不舒服的地方,会变得更简洁,例如 Service 中的接口定义:

  1. /**
  2. * 更新(非空字段),指定的强制更新字段不区分是否为空
  3. *
  4. * @param entity 实体类
  5. * @param forceUpdateFields 强制更新的字段,不区分字段是否为 null
  6. * @return 返回更新成功后的实体,远程服务调用时,由于序列化和反序列化,入参和返回值不是同一个对象
  7. */
  8. T updateSelective(T entity, Fn<T, Object>... forceUpdateFields);

调用的示例:

  1. userService.updateSelective(user, User::getName, User::getRoleId);

2.1.5 io.mybatis.mapper.list.ListMapper<T>

这个接口算示例,没有包含在 Mapper 中,提供下面一个方法:

  • int insertList(@Param("entityList") List<? extends T> entityList);: 批量插入

需要数据库支持 INSERT TABLE(C1,C2...) VALUES(...), (....) 语法才可以。

2.2 重写继承接口的定义

EntityMapper 中有 insert 方法定义如下:

  1. /**
  2. * 保存实体
  3. *
  4. * @param entity 实体类
  5. * @return 1成功,0失败
  6. */
  7. @Lang(Caching.class)
  8. @InsertProvider(type = EntityProvider.class, method = "insert")
  9. int insert(T entity);

这个定义没有处理主键,需要自己设置好主键后调用该方法新增数据。

如果我使用的 MySql 自增怎么办?主键null也能直接保存,但是不回写。

如果使用 Oracle 序列怎么办?直接用这个方法是没有办法的。

因为可以 重写继承接口的定义,所以可以支持所有 MyBatis 本身能支持的所有主键方式。

Mapper 中,覆盖定义如下:

  1. /**
  2. * 保存实体,默认主键自增,并且名称为 id
  3. * <p>
  4. * 这个方法是个示例,你可以在自己的接口中使用相同的方式覆盖父接口中的配置
  5. *
  6. * @param entity 实体类
  7. * @return 1成功,0失败
  8. */
  9. @Override
  10. @Lang(Caching.class)
  11. @Options(useGeneratedKeys = true, keyProperty = "id")
  12. @InsertProvider(type = EntityProvider.class, method = "insert")
  13. int insert(T entity);

首先 @Override 是重写父接口定义,然后和原来相比增加了下面的注解:

  1. @Options(useGeneratedKeys = true, keyProperty = "id")

这个注解对应 xml 中的配置如下:

  1. <insert id="insert" useGeneratedKeys="true" keyProperty="id">

useGeneratedKeys 意思是要用JDBC接口方式取回主键,主键字段对应的属性名为 id,就是要回写到 id 字段。

上面的配置对 MySQL 这类自增数据库是可行的,如果你自己的主键不叫 id,甚至如果每个表的主键都不统一(如 {tableName}_id), 你需要在每个具体实现的接口中重写。例如:

  1. public interface UserMapper extends Mapper<User, Long> {
  2. /**
  3. * 保存实体,默认主键自增,并且名称为 id
  4. * <p>
  5. * 这个方法是个示例,你可以在自己的接口中使用相同的方式覆盖父接口中的配置
  6. *
  7. * @param entity 实体类
  8. * @return 1成功,0失败
  9. */
  10. @Override
  11. @Lang(Caching.class)
  12. @Options(useGeneratedKeys = true, keyProperty = "userId")
  13. @InsertProvider(type = EntityProvider.class, method = "insert")
  14. int insert(User entity);
  15. }

如果是Oracle序列或者需要执行SQL生成主键或者取回主键时,可以配置 @SelectKey 注解,示例如下:

  1. @Override
  2. @Lang(Caching.class)
  3. @SelectKey(statement = "CALL IDENTITY()", keyProperty = "id", resultType = Long.class, before = false)
  4. @InsertProvider(type = EntityProvider.class, method = "insert")
  5. int insert(User entity);

上面还只是通过增加注解重新定义了接口方法。实际上你还可以更换 @InsertProvider(type = EntityProvider.class, method = "insert"), 将其中的实现换成其他的也可以,如果对默认的方法和逻辑不满意,就可以改成别的。

通过 重写继承接口的定义,应该能感觉出有多强大,多么灵活。

2.3 复制其他接口中的通用方法定义

这是最灵活的一点,在 Mapper 中直接复制了 FnMapper 的一个方法:

  1. /**
  2. * 根据主键更新实体中不为空的字段,强制字段不区分是否 null,都更新
  3. * <p>
  4. * 当前方法来自 {@link io.mybatis.mapper.fn.FnMapper},该接口中的其他方法用 {@link ExampleMapper} 也能实现
  5. *
  6. * @param entity 实体类
  7. * @param forceUpdateFields 强制更新的字段,不区分字段是否为 null,通过 {@link Fn#of(Fn...)} 创建 {@link Fn.Fns}
  8. * @return 1成功,0失败
  9. */
  10. @Lang(Caching.class)
  11. @UpdateProvider(type = FnProvider.class, method = "updateByPrimaryKeySelectiveWithForceFields")
  12. int updateByPrimaryKeySelectiveWithForceFields(@Param("entity") T entity, @Param("fns") Fn.Fns<T> forceUpdateFields);

这就是完全的复制粘贴,利用这一点,你可以不用 Mapper 接口作为自己的基类接口,你可以定义一个自己的接口,复制粘贴自己的需要的通用方法作为基础接口, 例如一个 GuozilanMapper 示例如下:

  1. public interface GuozilanMapper<T> {
  2. /**
  3. * 保存实体
  4. *
  5. * @param entity 实体类
  6. * @return 1成功,0失败
  7. */
  8. @Lang(Caching.class)
  9. @InsertProvider(type = EntityProvider.class, method = "insert")
  10. int insert(T entity);
  11. /**
  12. * 根据主键查询实体
  13. *
  14. * @param id 主键
  15. * @return 实体
  16. */
  17. @Lang(Caching.class)
  18. @SelectProvider(type = EntityProvider.class, method = "selectByPrimaryKey")
  19. Optional<T> selectByPrimaryKey(Long id);
  20. }

只要继承了上面的接口,你就直接拥有了这两个基础方法。

使用这种方式可以自定义一些自己项目需要用到的不同类别的通用接口,例如,如果你有大量实体都没有主键,默认的 Mapper<T, I> 就不太适合, 此时你可以自己创建一个 NoIdMapper<T>,把除了主键操作方法外的其他方法(有选择的)都拷过来,就形成了符合自己实际需要的通用 Mapper。

推而广之之后,还有更绝的用法,不继承接口,或者基础接口没有某个方法,直接复制注解过来,不需要自己写 XML:

  1. public interface UserMapper {
  2. /**
  3. * 保存实体
  4. *
  5. * @param entity 实体类
  6. * @return 1成功,0失败
  7. */
  8. @Lang(Caching.class)
  9. @InsertProvider(type = EntityProvider.class, method = "insert")
  10. int insert(User entity);
  11. }

你不需要任何具体的 SQL,上面的 insert 方法就可以直接使用了。

后期长远规划中,会利用 3.5.6 中的一些功能,直接配置一个 @InsertProvider 就能使用,不需要配置 typemethod。 在更长远的规划中,可以直接支持 JPA 中的方法名查询,例如 getByNameEqualAndAgeLessThen(@Param("name") String name, @Param("age") Integer age) 直接根据方法名生成一个 SQL。

2.4 使用 Java8 默认方法灵活实现通用方法

Mapper 接口中,利用现有的 Example 方法,实现了两个非常常用的通用方法:

  1. /**
  2. * 根据指定字段集合查询:field in (fieldValueList)
  3. * <p>
  4. * 这个方法是个示例,你也可以使用 Java8 的默认方法实现一些通用方法
  5. *
  6. * @param field 字段
  7. * @param fieldValueList 字段值集合
  8. * @param <F> 字段类型
  9. * @return 实体列表
  10. */
  11. default <F> List<T> selectByFieldList(Fn<T, F> field, List<F> fieldValueList) {
  12. Example<T> example = new Example<>();
  13. example.createCriteria().andIn((Fn<T, Object>) field, fieldValueList);
  14. return selectByExample(example);
  15. }
  16. /**
  17. * 根据指定字段集合删除:field in (fieldValueList)
  18. * <p>
  19. * 这个方法是个示例,你也可以使用 Java8 的默认方法实现一些通用方法
  20. *
  21. * @param field 字段
  22. * @param fieldValueList 字段值集合
  23. * @param <F> 字段类型
  24. * @return 实体列表
  25. */
  26. default <F> int deleteByFieldList(Fn<T, F> field, List<F> fieldValueList) {
  27. Example<T> example = new Example<>();
  28. example.createCriteria().andIn((Fn<T, Object>) field, fieldValueList);
  29. return deleteByExample(example);
  30. }

这两个方法可以直接根据某个字段值的集合进行批量查询或者删除,用法示例如下:

  1. List<User> users = mapper.selectByFieldList(User::getUserName, Arrays.asList("张无忌", "赵敏", "周芷若"));
  2. mapper.deleteByFieldList(User::getUserName, Arrays.asList("张无忌", "赵敏", "周芷若"));

除了这个例子外,还有一段 EntityMapper 被注释的示例:

  1. /**
  2. * 根据实体字段条件分页查询
  3. *
  4. * @param entity 实体类
  5. * @param rowBounds 分页信息
  6. * @return 实体列表
  7. *//*
  8. List<T> selectList(T entity, RowBounds rowBounds);
  9. *//**
  10. * 根据查询条件获取第一个结果
  11. *
  12. * @param entity 实体类
  13. * @return 实体
  14. *//*
  15. default Optional<T> selectFirst(T entity) {
  16. List<T> entities = selectList(entity, new RowBounds(0, 1));
  17. if (entities.size() == 1) {
  18. return Optional.of(entities.get(0));
  19. }
  20. return Optional.empty();
  21. }
  22. *//**
  23. * 根据查询条件获取指定的前几个对象
  24. *
  25. * @param entity 实体类
  26. * @param n 指定的个数
  27. * @return 实体
  28. *//*
  29. default List<T> selectTopN(T entity, int n) {
  30. return selectList(entity, new RowBounds(0, n));
  31. }

合理的通过 Java8 的默认方法,能够实现海量的通用方法。至于那些是真正需要用到的通用方法,就需要根据自己的需要来选择,因此虽然上面的方法能通用, 但是在缺乏频繁使用场景的情况下,Mapper 接口并没有接纳这几个方法。

2.5 通过修改接口方法的返回值和入参,就能变身无数个通用方法

EntityProvider 中的 select 方法为例,方法的具体实现如下:

  1. /**
  2. * 根据实体字段条件查询唯一的实体,根据实体字段条件批量查询,查询结果的数量由方法定义
  3. *
  4. * @param providerContext 上下文
  5. * @return cacheKey
  6. */
  7. public static String select(ProviderContext providerContext) {
  8. return SqlScript.caching(providerContext, new SqlScript() {
  9. @Override
  10. public String getSql(EntityTable entity) {
  11. return "SELECT " + entity.baseColumnAsPropertyList()
  12. + " FROM " + entity.table()
  13. + ifParameterNotNull(() ->
  14. where(() ->
  15. entity.whereColumns().stream().map(column ->
  16. ifTest(column.notNullTest(), () -> "AND " + column.columnEqualsProperty())
  17. ).collect(Collectors.joining(LF)))
  18. )
  19. + entity.groupByColumn().orElse("")
  20. + entity.havingColumn().orElse("")
  21. + entity.orderByColumn().orElse("");
  22. }
  23. });
  24. }

最终会生成一个 SELECT .. FROM .. WHERE ... 的 SQL,在 MyBatis 中,SQL 只定义了如何在数据库执行, 执行后的结果和取值方式是通过接口方法定义决定的,因此就这样一个 SELECT 查询,能够实现很多个方法,举例如下:

  1. @Lang(Caching.class)
  2. @SelectProvider(type = EntityProvider.class, method = "select")
  3. Optional<T> selectOne(T entity);
  4. @Lang(Caching.class)
  5. @SelectProvider(type = EntityProvider.class, method = "select")
  6. List<T> selectList(T entity);
  7. @Lang(Caching.class)
  8. @SelectProvider(type = EntityProvider.class, method = "select")
  9. List<T> selectAll();
  10. @Lang(Caching.class)
  11. @SelectProvider(type = EntityProvider.class, method = "select")
  12. Cursor<T> selectCursor(T entity);

利用这一特点,通过修改接口方法的返回值和入参,就能变身无数个通用方法。

如果在加个 RowBounds 分页参数,直接翻倍。