Mybatis-plus,真香!
前言
mybatis相信都不陌生,目前互联网公司大部分都使用mybatis作为持久层框架,无他,因为可以直接在xml文件中编写SQL语句操作数据库,灵活。但是我们在使用的时候,也会发现有很多增删改查的SQL是每个表都会有的基本操作,如果每个表都写一套增删改查的SQL显然是非常耗时耗力的。
于是乎,就有了mybatis-plus这个框架。正如官网所说,mybatis-plus是为简化开发而生
。
mybatis-plus有以下特点:
只做增强不做改变,引入它不会对现有工程产生影响,如丝般顺滑。 只需简单配置,即可快速进行单表CRUD操作,节省大量时间。 代码生成,物理分页,性能分析等功能一应俱全。
一、整合mybatis-plus
这里用的是SpringBoot2.5.2做演示。首先导入依赖:
<dependency>
<groupId>org.mybatis.spring.boot</groupId>
<artifactId>mybatis-spring-boot-starter</artifactId>
<version>2.2.0</version>
</dependency>
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<scope>runtime</scope>
</dependency>
<!-- 引入 mybatis-plus -->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.3.0</version>
</dependency>
然后在application.properties文件配置数据库信息:
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.url=jdbc:mysql://localhost:3306/user?createDatabaseIfNotExist=true
spring.datasource.username=root
spring.datasource.password=root123456
#mapper.xml文件路径地址
mybatis-plus.mapper-locations=classpath:mapper/*Mapper.xml
在启动类加上扫描注解:
@SpringBootApplication
@MapperScan(basePackages = "com.yehongzhi.mydemo.mapper")
public class MydemoApplication {
public static void main(String[] args) {
SpringApplication.run(MydemoApplication.class, args);
}
}
其实这样就完成了,但是我们要建个表进行测试。
CREATE TABLE `user` (
`id` char(36) NOT NULL DEFAULT '' COMMENT 'ID',
`name` varchar(255) DEFAULT '' COMMENT '姓名',
`age` int(3) DEFAULT NULL COMMENT '年龄',
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;
-- 初始化4条数据
INSERT INTO `user`.`user` (`id`, `name`, `age`) VALUES ('1345fc0985b111eba0e488d7f66fdab8', '观辰', '20');
INSERT INTO `user`.`user` (`id`, `name`, `age`) VALUES ('d47561e885b011eba0e488d7f66fdab8', '姚大秋', '30');
INSERT INTO `user`.`user` (`id`, `name`, `age`) VALUES ('ef2741fe87f011eba0e488d7f66fdab8', '周星驰', '60');
INSERT INTO `user`.`user` (`id`, `name`, `age`) VALUES ('ff784f6b85b011eba0e488d7f66fdab8', '李嘉晟', '33');
建了表之后,再创建一个User实体类对应:
//表名
@TableName("user")
public class User {
//主键
@TableId(type = IdType.UUID)
private String id;
//姓名
private String name;
//年龄
private Integer age;
//getter、setter方法
}
接着创建UserMapper接口类,,然后继承BaseMapper:
@Repository
public interface UserMapper extends BaseMapper<User> {
}
创建一个UserMapper.xml与UserMapper对应:
<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.yehongzhi.mydemo.mapper.UserMapper">
</mapper>
最后我们写一个UserService接口,查询user表:
@Service
public class UserServiceImpl implements UserService {
@Resource
private UserMapper userMapper;
@Override
public List<User> getList() {
return userMapper.selectList(null);
}
}
启动项目,测试是没问题的:
二、CRUD操作
整合完了之后,按照mybatis-plus的官方说明,是有简单的单表CRUD操作功能。
这些单表的CRUD操作其实都放在BaseMapper里面了,所以当我们继承了BaseMapper类之后,就会获得mybatis-plus的增强特性,其中就包括单表的CRUD操作。
1、insert操作
BaseMapper直接提供一个insert()方法,传一个实体类。
@Service
public class UserServiceImpl implements UserService {
@Resource
private UserMapper userMapper;
@Override
public int insertOne(User user) {
return userMapper.insert(user);
}
}
接着在Controller调用,因为在User类上面贴了注解@TableId
,所以会自动生成ID。
@TableName("user")
public class User {
@TableId(type = IdType.UUID)
private String id;
//省略...
}
Controller层代码:
@RequestMapping("/insert")
public String insertUser(@RequestParam(name = "name") String name,
@RequestParam(name = "age") Integer age) {
User user = new User();
user.setName(name);
user.setAge(age);
//在实体类使用了@TableId注解,ID会自动生成
int i = userService.insertOne(user);
return i == 1 ? "success" : "fail";
}
2、update操作
BaseMapper直接提供一个updateById()方法,传一个实体类。
@Override
public int updateOne(User user) {
return userMapper.updateById(user);
}
Controller层代码:
@RequestMapping("/update")
public String updateUser(@RequestParam(name = "id") String id,
@RequestParam(name = "name",required = false) String name,
@RequestParam(name = "age",required = false) Integer age) {
User user = new User();
user.setId(id);
user.setName(name);
user.setAge(age);
int i = userService.updateOne(user);
return i == 1 ? "success" : "fail";
}
3、delete操作
BaseMapper直接提供一个deleteById()方法,传主键值。
@Override
public int deleteOne(String id) {
return userMapper.deleteById(id);
}
Controller层代码:
@RequestMapping("/delete")
public String deleteUser(@RequestParam(name = "id") String id) {
int i = userService.deleteOne(id);
return i == 1 ? "success" : "fail";
}
除此之外,还有批量删除deleteBatchIds方法,传主键值的集合:
@Override
public int deleteBatch(List<String> ids){
return userMapper.deleteBatchIds(ids);
}
Controller层代码:
@RequestMapping("/deleteBatch")
public String deleteBatchUser(@RequestParam(name = "ids") String ids) {
List<String> list = Arrays.asList(ids.split(","));
int i = userService.deleteBatch(list);
return i == list.size() ? "success" : "fail";
}
三、条件构造器(Wrapper)
条件构造器在Mybatis-plus各种方法中都有出现,比如:
Integer selectCount(@Param(Constants.WRAPPER) Wrapper<T> queryWrapper);
List<T> selectList(@Param(Constants.WRAPPER) Wrapper<T> queryWrapper);
int update(@Param(Constants.ENTITY) T entity, @Param(Constants.WRAPPER) Wrapper<T> updateWrapper);
int delete(@Param(Constants.WRAPPER) Wrapper<T> queryWrapper);
Wrapper通俗点理解就是定义where语句后面的查询条件,是Mybatis-Plus里功能比较强大的工具。Wrapper是一个抽象类,下面有很多子类,我们先看个类图混个眼熟。
常用的子类实现有四个,分别是:
QueryWrapper UpdateWrapper LambdaQueryWrapper LambdaUpdateWrapper
QueryWrapper
主要用于生成where条件,举个例子,我们用name查询user表:
public List<User> queryUserByName(String name) {
//相当于:SELECT * FROM user WHERE name = ?
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
//eq()表示相等
queryWrapper.eq("name", name);
return userMapper.selectList(queryWrapper);
}
我们看日志打印:
==> Preparing: SELECT id,name,age FROM user WHERE name = ?
==> Parameters: 姚大秋(String)
<== Columns: id, name, age
<== Row: d47561e885b011eba0e488d7f66fdab8, 姚大秋, 30
<== Total: 1
假如我们要like查询,可以这样写:
public List<User> queryUserLikeName(String name) {
//相当于:SELECT * FROM user WHERE name like %#{name}%
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
queryWrapper.like("name",name);
return userMapper.selectList(queryWrapper);
}
假如要查询年龄大于30岁,可以这样:
public List<User> queryUserGtByAge(int age) {
//相当于:SELECT * FROM user WHERE age > ?
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
queryWrapper.gt("age", age);
//小于是lt()
//大于等于是ge()
//小于等于是le()
//范围的话,则使用between()
return userMapper.selectList(queryWrapper);
}
如果要查询某个字段不为空,可以这样:
public List<User> queryUserByNameNotNull() {
//相当于:SELECT * FROM user WHERE name IS NOT NULL
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
queryWrapper.isNotNull("name");
//查询某个字段为空,则使用isNull()
return userMapper.selectList(queryWrapper);
}
如果使用IN查询,可以这样:
public List<User> queryUserByIds(List<String> ids) {
//相当于:SELECT * FROM user WHERE name IN ('id1','id2');
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
queryWrapper.in("id", ids);
//相反也提供了notIn()方法
return userMapper.selectList(queryWrapper);
}
如果需要排序,可以这样写:
public List<User> queryUserOrderByAge() {
//相当于:SELECT * FROM user ORDER BY age ASC
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
queryWrapper.orderByAsc("age");
//相反的,如果降序则使用orderByDesc()方法
return userMapper.selectList(queryWrapper);
}
如果需要子查询,可以这样写:
public List<User> queryUserInSql() {
//相当于:SELECT * FROM user WHERE id IN (SELECT id FROM user WHERE age > 30)
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
queryWrapper.inSql("id","select id from user where age > 30");
return userMapper.selectList(queryWrapper);
}
大部分的方法都是源自AbstractWrapper抽象类,除此之外还有很多功能,这里就不一一介绍下去了,有兴趣的可以到官网的条件构造器慢慢探索。
UpdateWrapper
UpdateWrapper也是AbstractWrapper抽象类的子类实现,所以上述的设置条件的方法都有,不同的是,UpdateWrapper会有set()和setSql()设置更新的值。举个例子:
public int updateUserNameById(String id, String name) {
//相当于:UPDATE user SET name = ? where id = ?
UpdateWrapper<User> userUpdateWrapper = new UpdateWrapper<>();
userUpdateWrapper.eq("id", id);
//设置set关键字后面的语句,相当于set name = #{name}
userUpdateWrapper.set("name", name);
return userMapper.update(new User(), userUpdateWrapper);
}
setSql()就是设置SET关键字后面拼接的部分SQL语句,举个例子:
public int updateUserBySql(String id, String name) {
//相当于:UPDATE user SET name = ? where id = ?
UpdateWrapper<User> userUpdateWrapper = new UpdateWrapper<>();
userUpdateWrapper.setSql("name = '" + name + "'");
userUpdateWrapper.eq("id", id);
return userMapper.update(new User(), userUpdateWrapper);
}
setSql()就是纯粹的SQL语句拼接,我们可以看到SET后面接的是name='大D',而不是占位符会有SQL注入的风险。
==> Preparing: UPDATE user SET name = '大D' WHERE id = ?
==> Parameters: d47561e885b011eba0e488d7f66fdab8(String)
<== Updates: 1
LambdaQueryWrapper
用LambdaQueryWrapper的好处在于消除硬编码,比如QueryWrapper在查询name=?
时,需要这样写:
QueryWrapper<User> queryWrapper = new QueryWrapper<>();
//使用"name"字符串,就是硬编码
queryWrapper.eq("name", name);
如果换成LambdaQueryWrapper就可以这样写:
public List<User> lambdaQueryUserByName(String name) {
//相当于:SELECT id,name,age FROM user WHERE name = ?
LambdaQueryWrapper<User> lambdaQueryWrapper = new QueryWrapper<User>().lambda();
lambdaQueryWrapper.eq(User::getName, name);
return userMapper.selectList(lambdaQueryWrapper);
}
再比如使用模糊查询,可以这样写:
public List<User> lambdaQueryUserLikeName(String name) {
LambdaQueryWrapper<User> lambdaQueryWrapper = new QueryWrapper<User>().lambda();
lambdaQueryWrapper.like(User::getName, name);
return userMapper.selectList(lambdaQueryWrapper);
}
LambdaUpdateWrapper
跟上面差不多,LambdaUpdateWrapper也可以消除硬编码:
public int lambdaUpdateUserNameById(String id, String name) {
//相当于:UPDATE user SET name=? WHERE id = ?
LambdaUpdateWrapper<User> lambdaUpdateWrapper = new UpdateWrapper<User>().lambda();
lambdaUpdateWrapper.set(User::getName, name);
lambdaUpdateWrapper.eq(User::getId, id);
return userMapper.update(new User(), lambdaUpdateWrapper);
}
分页查询
Mybatis-plus提供了分页插件支持分页查询,只需要几个步骤即可实现。
首先添加一个配置类:
@Configuration
public class MybatisPlusConfig {
@Bean
public PaginationInterceptor paginationInterceptor() {
return new PaginationInterceptor();
}
}
第二步,在UserMapper类里定义分页查询方法:
@Repository
public interface UserMapper extends BaseMapper<User> {
//分页查询方法
IPage<User> selectPageByName(Page<User> page, @Param("name") String name);
}
对应的UserMapper.xml方法:
<select id="selectPageByName" parameterType="java.lang.String" resultType="com.yehongzhi.mydemo.model.User">
select * from `user` where name = #{name}
</select>
第三步,在userService里调用:
public IPage<User> selectPageByName(long pageNo, long pageSize, String name) {
Page<User> page = new Page<>();
//设置当前页码
page.setCurrent(pageNo);
//设置每页显示数
page.setSize(pageSize);
return userMapper.selectPageByName(page, name);
}
最后写个Controller接口测试:
@RequestMapping("/queryPage/ByName")
public IPage<User> selectPageByName(@RequestParam("pageNo") long pageNo,
@RequestParam("pageSize") long pageSize,
@RequestParam("name") String name) {
return userService.selectPageByName(pageNo, pageSize, name);
}
查看控制台日志:
==> Preparing: SELECT COUNT(1) FROM `user` WHERE name = ?
==> Parameters: 杜琪峰(String)
<== Columns: COUNT(1)
<== Row: 1
==> Preparing: select * from `user` where name = ? LIMIT ?,?
==> Parameters: 杜琪峰(String), 0(Long), 10(Long)
<== Columns: id, name, age
<== Row: d47561e885b011eba0e488d7f66fdab8, 杜琪峰, 30
<== Total: 1
分页查询成功!
自定义主键生成器
有时在实际开发中,可能会遇到,Mybatis-plus提供的主键生成策略并不能满足,需要自定义主键ID生成策略,怎么设置呢?
很简单,根据官网的说明,我们先定义一个主键生成器:
@Component
public class SnowflakeKeyGenerator implements IdentifierGenerator {
//自己实现的一个雪花ID生成工具类
@Resource
private SnowflakeIdWorker snowflakeIdWorker;
@Override
public Number nextId(Object entity) {
//使用雪花ID生成器,生成一个雪花ID
long nextId = snowflakeIdWorker.nextId();
System.out.println(String.format("使用自定义ID生成器,生成雪花ID:%s", nextId));
return nextId;
}
}
然后我们在需要使用该自定义ID生成器的实体类上面加上注解属性:
@TableName("user")
public class User {
//属性设置为:IdType.ASSIGN_ID
@TableId(type = IdType.ASSIGN_ID)
private String id;
//省略...
}
接着测试一下,我们可以看到控制台有打印日志:
总结
除了上面介绍的功能之外,Mybatis-plus还有很多功能,比如:代码生成器、扩展等等。这里就不再一一介绍了,实际上我们掌握上面介绍的CRUD、条件构造器、分页查询、自定义主键策略,基本上已经足够日常的开发。
当然如果你觉得会用还不够,还想要看懂框架的源码实现,那没问题。下一篇文章,我就讲讲框架的源码分析,敬请期待,
非常感谢你的阅读,希望这篇文章能给到你帮助和启发。
觉得有用就点个赞吧,你的点赞是我创作的最大动力~
我是一个努力让大家记住的程序员。我们下期再见!!!
能力有限,如果有什么错误或者不当之处,请大家批评指正,一起学习交流!