关于后端:MyBatis-如何实现流式查询

基本概念

流式查问指的是查问胜利后不是返回一个汇合而是返回一个迭代器，利用每次从迭代器取一条查问后果。流式查问的益处是可能升高内存应用。

如果没有流式查问，咱们想要从数据库取 1000 万条记录而又没有足够的内存时，就不得不分页查问，而分页查问效率取决于表设计，如果设计的不好，就无奈执行高效的分页查问。因而流式查问是一个数据库拜访框架必须具备的性能。

流式查问的过程当中，数据库连贯是放弃关上状态的，因而要留神的是：执行一个流式查问后，数据库拜访框架就不负责敞开数据库连贯了，须要利用在取完数据后本人敞开。

MyBatis 流式查问接口

MyBatis 提供了一个叫 org.apache.ibatis.cursor.Cursor 的接口类用于流式查问，这个接口继承了 java.io.Closeable 和 java.lang.Iterable 接口，由此可知：

1. Cursor 是可敞开的；
2. Cursor 是可遍历的。

除此之外，Cursor 还提供了三个办法：

1. isOpen()：用于在取数据之前判断 Cursor 对象是否是关上状态。只有当关上时 Cursor 能力取数据；
2. isConsumed()：用于判断查问后果是否全副取完。
3. getCurrentIndex()：返回曾经获取了多少条数据

因为 Cursor 实现了迭代器接口，因而在理论应用当中，从 Cursor 取数据非常简单：

cursor.forEach(rowObject -> {...});

但构建 Cursor 的过程不简略

咱们举个理论例子。上面是一个 Mapper 类：

@Mapper
public interface FooMapper {@Select("select * from foo limit #{limit}")
    Cursor<Foo> scan(@Param("limit") int limit);
}

办法 scan() 是一个非常简单的查问。通过指定 Mapper 办法的返回值为 Cursor 类型，MyBatis 就晓得这个查询方法是一个流式查问。
而后咱们再写一个 SpringMVC Controller 办法来调用 Mapper（无关的代码曾经省略）：

@GetMapping("foo/scan/0/{limit}")
public void scanFoo0(@PathVariable("limit") int limit) throws Exception {try (Cursor<Foo> cursor = fooMapper.scan(limit)) {  // 1
        cursor.forEach(foo -> {});                      // 2
    }
}

下面的代码中，fooMapper 是 @Autowired 进来的。正文 1 处调用 scan 办法，失去 Cursor 对象并保障它能最初敞开；2 处则是从 cursor 中取数据。
下面的代码看上去没什么问题，然而执行 scanFoo0() 时会报错：

java.lang.IllegalStateException: A Cursor is already closed.

这是因为咱们后面说了在取数据的过程中须要放弃数据库连贯，而 Mapper 办法通常在执行完后连贯就敞开了，因而 Cusor 也一并敞开了。

所以，解决这个问题的思路不简单，放弃数据库连贯关上即可。咱们至多有三种计划可选。

计划一：SqlSessionFactory

咱们能够用 SqlSessionFactory 来手工关上数据库连贯，将 Controller 办法批改如下：

@GetMapping("foo/scan/1/{limit}")
public void scanFoo1(@PathVariable("limit") int limit) throws Exception {
    try (SqlSession sqlSession = sqlSessionFactory.openSession();  // 1
        Cursor<Foo> cursor = 
              sqlSession.getMapper(FooMapper.class).scan(limit)   // 2
    ) {cursor.forEach(foo -> {});
    }
}

下面的代码中，1 处咱们开启了一个 SqlSession（实际上也代表了一个数据库连贯），并保障它最初能敞开；2 处咱们应用 SqlSession 来取得 Mapper 对象。这样能力保障失去的 Cursor 对象是关上状态的。

计划二：TransactionTemplate

在 Spring 中，咱们能够用 TransactionTemplate 来执行一个数据库事务，这个过程中数据库连贯同样是关上的。代码如下：

@GetMapping("foo/scan/2/{limit}")
public void scanFoo2(@PathVariable("limit") int limit) throws Exception {
    TransactionTemplate transactionTemplate = 
            new TransactionTemplate(transactionManager);  // 1

    transactionTemplate.execute(status -> {               // 2
        try (Cursor<Foo> cursor = fooMapper.scan(limit)) {cursor.forEach(foo -> {});
        } catch (IOException e) {e.printStackTrace();
        }
        return null;
    });
}

下面的代码中，1 处咱们创立了一个 TransactionTemplate 对象（此处 transactionManager 是怎么来的不必多解释，本文假如读者对 Spring 数据库事务的应用比拟相熟了），2 处执行数据库事务，而数据库事务的内容则是调用 Mapper 对象的流式查问。留神这里的 Mapper 对象无需通过 SqlSession 创立。

计划三：@Transactional 注解

这个实质上和计划二一样，代码如下：

@GetMapping("foo/scan/3/{limit}")
@Transactional
public void scanFoo3(@PathVariable("limit") int limit) throws Exception {try (Cursor<Foo> cursor = fooMapper.scan(limit)) {cursor.forEach(foo -> {});
    }
}

它仅仅是在原来办法下面加了个 @Transactional 注解。这个计划看上去最简洁，但请留神 Spring 框架当中注解应用的坑：只在内部调用时失效。在以后类中调用这个办法，依旧会报错。

以上是三种实现 MyBatis 流式查问的办法。