这是公众号《Throwable文摘》发布的第25篇原创文章,收录于专辑《SpringBoot2.x入门》。
前提
这篇文章是《SpringBoot2.x入门》专辑的第8篇文章,使用的SpringBoot
版本为2.3.1.RELEASE
,JDK
版本为1.8
。
SpringBoot
项目引入MyBatis
一般的套路是直接引入mybatis-spring-boot-starter
或者使用基于MyBatis
进行二次封装的框架例如MyBatis-Plus
或者tk.mapper
等,但是本文会使用一种更加原始的方式,单纯依赖org.mybatis:mybatis
和org.mybatis:mybatis-spring
把MyBatis
的功能整合到SpringBoot
中,Spring(Boot)
使用的是微内核架构,任何第三方框架或者插件都可以按照本文的思路融合到该微内核中。
引入MyBatis依赖
编写本文的时候(2020-07-18
)org.mybatis:mybatis
的最新版本是3.5.5
,而org.mybatis:mybatis-spring
的最新版本是2.0.5
,在使用BOM
管理SpringBoot
版本的前提下,引入下面的依赖:
org.springframework.boot
spring-boot-starter-web
org.springframework.boot
spring-boot-starter-jdbc
org.mybatis
mybatis
3.5.5
org.mybatis
mybatis-spring
2.0.5
注意的是低版本的MyBatis如果需要使用JDK8的日期时间API,需要额外引入mybatis-typehandlers-jsr310依赖,但是某个版本之后mybatis-typehandlers-jsr310中的类已经移植到org.mybatis:mybatis中作为内建类,可以放心使用JDK8的日期时间API。
添加MyBatis配置
MyBatis
的核心模块是SqlSessionFactory
与MapperScannerConfigurer
。前者可以使用SqlSessionFactoryBean
,功能是为每个SQL
的执行提供SqlSession
和加载全局配置或者SQL
实现的XML
文件,后者是一个BeanDefinitionRegistryPostProcessor
实现,主要功能是主动通过配置的基础包(Base Package
)中递归搜索Mapper
接口(这个算是MyBatis
独有的扫描阶段,务必指定明确的扫描包,否则会因为效率太低导致启动阶段耗时增加),并且把它们注册成MapperFactoryBean
(简单理解为接口动态代理实现添加到方法缓存中,并且委托到IOC
容器,此后可以直接注入Mapper
接口),注意这个BeanFactoryPostProcessor
的回调优先级极高,在自动装配@Autowired
族注解或者@ConfigurationProperties
属性绑定处理之前已经回调,因此在处理MapperScannerConfigurer
的属性配置时候绝对不能使用@Value
或者自定义前缀属性Bean
进行自动装配,但是可以从Environment
中直接获取。
这里添加一个自定义属性前缀mybatis
,用于绑定配置文件中的属性到MyBatisProperties
类中:
@ConfigurationProperties(prefix = "mybatis")
@Data
public class MyBatisProperties {
private String configLocation;
private String mapperLocations;
private String mapperPackages;
private static final ResourcePatternResolver RESOLVER = new PathMatchingResourcePatternResolver();
/**
* 转化Mapper映射文件为Resource
*/
public Resource[] getMapperResourceArray() {
if (!StringUtils.hasLength(mapperLocations)) {
return new Resource[0];
}
List resources = new ArrayList<>();
String[] locations = StringUtils.commaDelimitedListToStringArray(mapperLocations);
for (String location : locations) {
try {
resources.addAll(Arrays.asList(RESOLVER.getResources(location)));
} catch (IOException e) {
throw new IllegalArgumentException(e);
}
}
return resources.toArray(new Resource[0]);
}
}
接着添加一个MybatisAutoConfiguration
用于配置SqlSessionFactory
:
@Configuration
@EnableConfigurationProperties(value = {MyBatisProperties.class})
@ConditionalOnClass({SqlSessionFactory.class, SqlSessionFactoryBean.class})
@RequiredArgsConstructor
public class MybatisAutoConfiguration {
private final MyBatisProperties myBatisProperties;
@Bean
public SqlSessionFactoryBean sqlSessionFactoryBean(DataSource dataSource) {
SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
bean.setDataSource(dataSource);
// 其实核心配置就是这两项,其他TypeHandlersPackage、TypeAliasesPackage等等自行斟酌是否需要添加
bean.setConfigLocation(new ClassPathResource(myBatisProperties.getConfigLocation()));
bean.setMapperLocations(myBatisProperties.getMapperResourceArray());
return bean;
}
/**
* 事务模板,用于编程式事务 - 可选配置
*/
@Bean
@ConditionalOnMissingBean
public TransactionTemplate transactionTemplate(PlatformTransactionManager platformTransactionManager) {
return new TransactionTemplate(platformTransactionManager);
}
/**
* 数据源事务管理器 - 可选配置
*/
@Bean
@ConditionalOnMissingBean
public PlatformTransactionManager platformTransactionManager(DataSource dataSource) {
return new DataSourceTransactionManager(dataSource);
}
}
一般情况下,启用事务需要定义PlatformTransactionManager
的实现,而TransactionTemplate
适用于编程式事务(和声明式事务@Transactional
区别,编程式更加灵活)。上面的配置类中只使用了两个属性,而mybatis.mapperPackages
将用于MapperScannerConfigurer
的加载上。添加MapperScannerRegistrarConfiguration
如下:
@Configuration
public class MapperScannerRegistrarConfiguration {
public static class AutoConfiguredMapperScannerRegistrar implements
BeanFactoryAware, EnvironmentAware, ImportBeanDefinitionRegistrar {
private Environment environment;
private BeanFactory beanFactory;
@Override
public void setBeanFactory(@NonNull BeanFactory beanFactory) throws BeansException {
this.beanFactory = beanFactory;
}
@Override
public void setEnvironment(@NonNull Environment environment) {
this.environment = environment;
}
@Override
public void registerBeanDefinitions(@NonNull AnnotationMetadata importingClassMetadata,
@NonNull BeanDefinitionRegistry registry) {
BeanDefinitionBuilder builder = BeanDefinitionBuilder.genericBeanDefinition(MapperScannerConfigurer.class);
builder.addPropertyValue("processPropertyPlaceHolders", true);
StringJoiner joiner = new StringJoiner(ConfigurableApplicationContext.CONFIG_LOCATION_DELIMITERS);
// 这里使用了${mybatis.mapperPackages},否则会使用AutoConfigurationPackages.get(this.beanFactory)获取项目中自定义配置的包
String mapperPackages = environment.getProperty("mybatis.mapperPackages");
if (null != mapperPackages) {
String[] stringArray = StringUtils.commaDelimitedListToStringArray(mapperPackages);
for (String pkg : stringArray) {
joiner.add(pkg);
}
} else {
List packages = AutoConfigurationPackages.get(this.beanFactory);
for (String pkg : packages) {
joiner.add(pkg);
}
}
builder.addPropertyValue("basePackage", joiner.toString());
BeanWrapper beanWrapper = new BeanWrapperImpl(MapperScannerConfigurer.class);
Stream.of(beanWrapper.getPropertyDescriptors())
.filter(x -> "lazyInitialization".equals(x.getName())).findAny()
.ifPresent(x -> builder.addPropertyValue("lazyInitialization",
"${mybatis.lazyInitialization:false}"));
registry.registerBeanDefinition(MapperScannerConfigurer.class.getName(), builder.getBeanDefinition());
}
}
@Configuration
@Import(AutoConfiguredMapperScannerRegistrar.class)
@ConditionalOnMissingBean({MapperFactoryBean.class, MapperScannerConfigurer.class})
public static class MapperScannerRegistrarNotFoundConfiguration {
}
}
到此基本的配置Bean
已经定义完毕,接着需要添加配置项。一般一个项目的MyBatis
配置是相对固定的,可以直接添加在主配置文件application.properties
中:
server.port=9098
spring.application.name=ch8-mybatis
mybatis.configLocation=mybatis-config.xml
mybatis.mapperLocations=classpath:mappings/base,classpath:mappings/ext
mybatis.mapperPackages=club.throwable.ch8.repository.mapper,club.throwable.ch8.repository
个人喜欢在resource/mappings
目录下定义base
和ext
两个目录,base
目录用于存在MyBatis
生成器生成的XML
文件,这样就能在后续添加了表字段之后直接重新生成和覆盖base
目录下对应的XML
文件即可。同理,在项目的源码包下建repository/mapper
,然后Mapper
类直接存放在repository/mapper
目录,DAO
类存放在repository
目录,MyBatis
生成器生成的Mapper
类可以直接覆盖repository/mapper
目录中对应的类。
resources
目录下添加一个MyBatis
的全局配置文件mybatis-config.xml
:
项目目前的基本结构如下:
使用Mybatis
为了简单起见,这里使用h2
内存数据库进行演示。添加h2
的依赖:
com.h2database
h2
1.4.200
resources
目录下添加一个schema.sql
和data.sql
:
// resources/schema.sql
drop table if exists customer;
create table customer
(
id bigint generated by default as identity,
customer_name varchar(32),
age int,
create_time timestamp default current_timestamp,
edit_time timestamp default current_timestamp,
primary key (id)
);
// resources/data.sql
INSERT INTO customer(customer_name,age) VALUES ('doge', 22);
INSERT INTO customer(customer_name,age) VALUES ('throwable', 23);
添加对应的实体类club.throwable.ch8.entity.Customer
:
@Data
public class Customer {
private Long id;
private String customerName;
private Integer age;
private LocalDateTime createTime;
private LocalDateTime editTime;
}
添加Mapper
和DAO
类:
// club.throwable.ch8.repository.mapper.CustomerMapper
public interface CustomerMapper {
}
// club.throwable.ch8.repository.CustomerDao
public interface CustomerDao extends CustomerMapper {
Customer queryByName(@Param("customerName") String customerName);
}
添加XML
文件resource/mappings/base/BaseCustomerMapper.xml
和resource/mappings/base/ExtCustomerMapper.xml
:
// BaseCustomerMapper.xml
// ExtCustomerMapper.xml
细心的伙伴会发现,DAO和Mapper类是继承关系,而ext和base下对应的Mapper文件中的BaseResultMap也是继承关系
配置文件中增加h2
数据源的配置:
// application.properties
spring.datasource.url=jdbc:h2:mem:db_customer;MODE=MYSQL;DB_CLOSE_DELAY=-1;DATABASE_TO_UPPER=FALSE
spring.datasource.username=root
spring.datasource.password=123456
spring.datasource.driver-class-name=org.h2.Driver
spring.h2.console.enabled=true
spring.h2.console.path=/h2-console
spring.h2.console.settings.web-allow-others=true
spring.datasource.schema=classpath:schema.sql
spring.datasource.data=classpath:data.sql
添加一个启动类进行验证:
public class Ch8Application implements CommandLineRunner {
@Autowired
private CustomerDao customerDao;
@Autowired
private ObjectMapper objectMapper;
public static void main(String[] args) {
SpringApplication.run(Ch8Application.class, args);
}
@Override
public void run(String... args) throws Exception {
Customer customer = customerDao.queryByName("doge");
log.info("Query [name=doge],result:{}", objectMapper.writeValueAsString(customer));
customer = customerDao.queryByName("throwable");
log.info("Query [name=throwable],result:{}", objectMapper.writeValueAsString(customer));
}
}
执行结果如下:
使用Mybatis生成器生成Mapper文件
有些时候为了提高开发效率,更倾向于使用生成器去预生成一些已经具备简单CRUD
方法的Mapper
文件,这个时候可以使用mybatis-generator-core
。编写本文的时候(2020-07-18
)mybatis-generator-core
的最新版本为1.4.0
,mybatis-generator-core
可以通过编程式使用或者Maven
插件形式使用。
这里仅仅简单演示一下Maven
插件形式下使用mybatis-generator-core
的方式,关于mybatis-generator-core
后面会有一篇数万字的文章详细介绍此生成器的使用方式和配置项的细节。在项目的resources
目录下添加一个generatorConfig.xml
:
然后再项目的POM
文件添加一个Maven
插件:
org.mybatis.generator
mybatis-generator-maven-plugin
1.4.0
Generate MyBatis Artifacts
generate
jdbc:h2:mem:db_customer;MODE=MYSQL;DB_CLOSE_DELAY=-1;DATABASE_TO_UPPER=FALSE
org.h2.Driver
root
123456
${basedir}/src/main/resources/schema.sql
com.h2database
h2
1.4.200
笔者发现这里必须要在插件的配置中重新定义数据库连接属性和schema.sql,因为插件跑的时候无法使用项目中已经启动的h2实例,具体原因未知。
配置完毕之后,执行Maven
命令:
mvn -Dmybatis.generator.overwrite=true mybatis-generator:generate -X
然后resource/mappings/base
目录下新增了一个带有基本CRUD
方法实现的CustomerMapper.xml
,同时CustoemrMapper
接口和Customer
实体也被重新覆盖生成了。
这里把前面手动编写的BaseCustomerMapper.xml注释掉,预防冲突。另外,CustomerMapper.xml的insertSelective标签需要加上keyColumn="id" keyProperty="id" useGeneratedKeys="true"属性,用于实体insert后的主键回写。
最后,修改并重启启动一下Ch8Application
验证结果:
小结
这篇文章相对详细地介绍了SpringBoot
项目如何使用MyBatis
,如果需要连接MySQL
或者其他数据库,只需要修改数据源配置和MyBatis
生成器的配置文件即可,其他配置类和项目骨架可以直接复用。
本文demo
仓库:
Github
:https://github.com/zjcscut/spring-boot-guide/tree/master/ch8-mybatis
(本文完 c-2-d e-a-20200719 封面来自《秒速五厘米》)
公众号《Throwable文摘》(id:throwable-doge),不定期推送架构设计、并发、源码探究相关的原创文章: