Spring Boot 实践系列,Spring Boot + MyBatis 。

目的

将 MyBatis 与 Spring Boot 应用程序一起使用来访问数据库。

本次使用的Library

  • spring-boot-starter:2.2.0.M4
  • mybatis-spring-boot-starter:2.0.1
  • mybatis-spring-boot-starter-test:2.0.1
  • h2:1.4.199
  • lombok:1.18.8

数据库访问

在实现 MyBatis 之前,首先进行 DB 访问的设置。

数据库访问定义

DataSourceSpring Boot自动定义了访问 DB等 bean。
默认情况下,它访问内存中的 H2 数据库,因此您需要将 h2 添加到依赖项中。

pom.xml

 
    <dependencies>
        <dependency>
            <groupId>com.h2database</groupId>
            <artifactId>h2</artifactId>
            <scope>runtime</scope>
        </dependency>
    </dependencies>

如果直接在 src/main/resources 下创建如下 SQL 文件,也可以在应用启动时自动初始化数据库。
默认识别的 SQL 文件如下。

  • schema.sql
  • schema-${platform}.sql
  • data.sql
  • data-${platform}.sql

如您所见,我们在 schema.sql 中定义了 DDL,在 data.sql 中定义了 DML。

这一次,创建 schema.sql 并创建一个表。

src/main/resources/schema.sql

 
create table if not exists todo (
    todo_id identity,
    todo_title varchar(30),
    finished boolean,
    created_at timestamp
);

mybatis-spring-boot-starter

它是使用 MyBatis 和 Spring Boot 的入门者。
通过使用 Spring Boot 的 Auto Configuration 机制,自动完成在 Spring Boot 应用中使用 MyBatis 的 Bean 定义。开发者只需要将 mybatis-spring-boot-starter 添加到他们的依赖项中。

pom.xml

 
    <dependencies>
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>2.0.1</version>
        </dependency>
    </dependencies>

src/main/java/*/Application.java

 
@SpringBootApplication
public class Application {
    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }
}

主类没有改变 Spring Boot 应用程序中的默认值。

领域类

映射 DB 数据的域类对于普通的 Java Bean 是可以的。
在下文中,Lombok 用于省略 Getter 和 Setter 的实现@Data

src/main/java/*/domain/Todo.java

 
@Data
public class Todo {
    private String todoId;
    private String todoTitle;
    private boolean finished;
    private LocalDateTime createdAt;
}

存储库接口

MyBatis 的 Mapper 对应 Spring 的 Repository 接口。
MyBatis 的 Mapper 可以通过以下任意一种方式实现。

  • 创建Repository接口对应的Mapper XML
  • @Mapper授予存储库接口


这次我将根据注解库 Spring Boot 用 MyBatis 的 Mapper 来实现@Mapper

src/main/java/*/repository/TodoRepository.java

 
@Mapper // (1)
public interface TodoRepository {

    // (2)
    @Select("SELECT todo_id, todo_title, finished, created_at FROM todo WHERE todo_id = #{todoId}")
    Optional<Todo> findById(String todoId);

    @Select("SELECT todo_id, todo_title, finished, created_at FROM todo")
    Collection<Todo> findAll();

    @Insert("INSERT INTO todo (todo_title, finished, created_at) VALUES (#{todoTitle}, #{finished}, #{createdAt})")
    @Options(useGeneratedKeys = true, keyProperty = "todoId") // (3)
    void create(Todo todo);

    @Update("UPDATE todo SET todo_title = #{todoTitle}, finished = #{finished}, created_at = #{createdAt} WHERE todo_id = #{todoId}")
    boolean updateById(Todo todo);

    @Delete("DELETE FROM todo WHERE todo_id = #{todoId}")
    void deleteById(Todo todo);

    @Select("SELECT COUNT(*) FROM todo WHERE finished = #{finished}")
    long countByFinished(boolean finished);
}

(1)如果添加到Repository界面@Mapper,MyBatis会自动扫描并注册到Mapper中。让我们将Repository接口放在主类下的包中。

(2)在给方法的...中实现要执行的SQL 。由于参数在 SQL 中使用,因此比 XML 更容易理解,因为它在同一个文件中进行了描述。@Select@Insert@Update@Delete#{}

(3) @Options在需要执行异常设置的 SQL 时给出。这里,由于表的关键项是IDENTITY列,所以在DB端是自动编号的,但是@Options你可以通过using来使用自动编号的ID。

多行 SQL

虽然上面的 SQL 是用一行描述的,但还是建议多行描述,因为它不可读。
它比 XML 更难看,因为它是一种像这样的字符串连接形式。

 
    @Select("SELECT"
            + " todo_id,"
            + " todo_title,"
            + " finished,"
            + " created_at"
            + " FROM todo"
            + " WHERE"
            + " todo_id = #{todoId}")
    Optional<Todo> findById(String todoId);

选择结果的自动和手动映射

Select 结果映射失败,因为图示的 Select 语句的结果列名称和 Todo 类的属性名称不同。(todo_title等等todoTitle

有以下方法之一可以解决此问题。

  • MyBatis 命名规则自动映射
  • @Results@ResultMap手动映射和

就个人而言,为属性命名以便自动映射可用,并且仅在它偏离规则时才进行手动映射是一个好主意。

自动映射

如果下划线分隔的列名与驼峰式的属性名匹配,则可以通过 MyBatis 的命名规则进行自动映射。

将以下设置添加到 Spring Boot 配置文件中。

src/main/resources/application.yml

 
mybatis:
  configuration:
    map-underscore-to-camel-case: true
手动映射

如果列名和属性名不匹配,则@Results需要@ResultMap为每个 SQL 定义手动映射。

  • @Results在中定义手动映射规则
  • @Results如果要使用中定义的规则@ResultMap@Results请指定中的 ID。
 
    @Select("SELECT todo_id, todo_title, finished, created_at FROM todo WHERE todo_id = #{todoId}")
    @Results(id = "todo", value = {
            @Result(column = "todo_id", property = "todoId"),
            @Result(column = "todo_title", property = "todoTitle"),
            @Result(column = "finished", property = "finished"),
            @Result(column = "created_at", property = "createdAt") })
    Optional<Todo> findById(String todoId);

    @Select("SELECT todo_id, todo_title, finished, created_at FROM todo")
    @ResultMap("todo")
    Collection<Todo> findAll();

与动态 SQL 通用

使用 Mapper XML 可能实现的动态 SQL ( <if>) 和通用性()<sql>现在通过 SqlProvider 实现。

 
    // (2)
    @SelectProvider(type = TodoSqlProvider.class, method = "find")
    Optional<Todo> findById(String todoId);

    @SelectProvider(type = TodoSqlProvider.class, method = "find")
    Collection<Todo> findAll();

    // (1)
    public class TodoSqlProvider {
        public String find(String todoId) {
            return new SQL() {{
                SELECT("todo_id", "todo_title", "finished", "created_at");
                FROM("todo");
                if (todoId != null) {
                    WHERE("todo_id = #{todoId}");
                }
            }}.toString();
        }
    }

(1) 实现SqlProvider类,并new SQL()在方法中使用组装SQL。例子中使用了实例初始化器( new SQL() {{ココ}}),当然你也可以正常写在方法链中。

(2)通过赋值@Select而不是@SelectProvider方法来指定实现的SqlProvider类和方法。

mybatis-spring-boot-starter-test

使用 Spring Boot 测试 MyBatis 的入门者。

通过使用 Spring Boot Auto Configuration 机制,Spring Boot 应用会自动定义一个 bean 用于测试 MyBatis Mapper。开发者只需要在自己的依赖中加入 mybatis-spring-boot-starter-test 并做一些设置即可。

pom.xml

 
    <dependencies>
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter-test</artifactId>
            <version>2.0.1</version>
        </dependency>
    </dependencies>

JUnit 测试用例

在 JUnit 测试用例中,您所要做的就是提供类并@MyBatisTest制作要测试的存储库@Autowired

src/test/java/*/repository/TodoRepositoryTest.java

 
@MybatisTest
class TodoRepositoryTest {

    @Autowired
    TodoRepository todoRepository;

    @Test
    @Sql(statements = "INSERT INTO todo (todo_title, finished, created_at) VALUES ('sample todo', false, '2019-01-01')")
    void testFindAll() {
        // execute
        Collection<Todo> todos = todoRepository.findAll();

        // assert
        assertThat(todos)
                .hasSize(1)
                .extracting(Todo::getTodoTitle, Todo::isFinished, Todo::getCreatedAt)
                .containsExactly(tuple("sample todo", false, LocalDate.of(2019, 1, 1).atStartOfDay()));
    }

@MyBatisTestDataSource会自动定义使用 MyBatis、访问 DB 等的 bean ,但@Transactional也会添加更多。

@Transactional@Sql测试完成后会回滚测试中执行的SQL。这是安全的,因为即使在访问和测试不在内存中的实际数据库时,也可以保持测试的独立性。

06-17 19:31