MybatisPlus 从入门到精通-基础篇

空~2022年11月30日
  • mybatis
大约 15 分钟

MybatisPlus 从入门到精通-基础篇

1.概述

MybatisPlus 是一款 Mybatis 增强工具,用于简化开发,提高效率。 它在 MyBatis 的基础上只做增强不做改变,为简化开发、提高效率而生。

官网: https://mp.baomidou.com/open in new window

2.快速入门

2.0 准备工作

① 准备数据

CREATE TABLE `user` (
  `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'id',
  `user_name` varchar(20) NOT NULL COMMENT '用户名',
  `password` varchar(20) NOT NULL COMMENT '密码',
  `name` varchar(30) DEFAULT NULL COMMENT '姓名',
  `age` int(11) DEFAULT NULL COMMENT '年龄',
  `address` varchar(100) DEFAULT NULL COMMENT '地址',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8;

insert  into `user`(`id`,`user_name`,`password`,`name`,`age`,`address`) values (1,'ruiwen','123','瑞文',12,'山东'),(2,'gailun','1332','盖伦',13,'平顶山'),(3,'timu','123','提姆',22,'蘑菇石'),(4,'daji','1222','妲己',221,'狐山');

② 创建 SpringBoot 工程

添加依赖
<parent>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-parent</artifactId>
    <version>2.5.0</version>
</parent>
<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.projectlombok</groupId>
        <artifactId>lombok</artifactId>
        <optional>true</optional>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-test</artifactId>
        <scope>test</scope>
    </dependency>
</dependencies>
创建启动类
@SpringBootApplication
public class SGApplication {

    public static void main(String[] args) {
        SpringApplication.run(SGApplication.class);
    }
}

③ 准备实体类

@Data
@NoArgsConstructor
@AllArgsConstructor
public class User {
    private Long id;
    private String userName;
    private String password;
    private String name;
    private Integer age;
    private String address;
}

2.1 使用 MybatisPlus

① 添加依赖

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>3.4.3</version>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
</dependency>

② 配置数据库信息

spring:
  datasource:
    url: jdbc:mysql://localhost:3306/mp_db?characterEncoding=utf-8&serverTimezone=UTC
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver

③ 创建 Mapper 接口

创建 Mapper 接口继承 BaseMapper 接口

public interface UserMapper extends BaseMapper<User> {
}

BaseMapper 接口中已经提供了很多常用方法。所以我们只需要直接从容器中获取 Mapper 就可以进行操作了,不需要自己去编写 Sql 语句。

④ 配置 Mapper 扫描

在启动类上配置我们的 Mapper 在哪个包。

@SpringBootApplication
@MapperScan("com.sangeng.mapper")
public class SGApplication {
    public static void main(String[] args) {
        SpringApplication.run(SGApplication.class,args);
    }
}

⑤ 获取 Mapper 进行测试

@SpringBootTest
public class MPTest {

    @Autowired
    private UserMapper userMapper;

    @Test
    public void testQueryList(){
        System.out.println(userMapper.selectList(null));
    }
}

3.常用设置

3.1 设置表映射规则

默认情况下 MP 操作的表名就是实体类的类名,但是如果表名和类名不一致就需要我们自己设置映射规则。

3.1.1 单独设置

可以在实体类的类名上加上@TableName 注解进行标识。

例如:

如果表名是 tb_user,而实体类名是 User 则可以使用以下写法。

@TableName("tb_user")
public class User {
    //....
}

3.1.2 全局设置表名前缀

一般一个项目表名的前缀都是统一风格的,这个时候如果一个个设置就太麻烦了。我们可以通过配置来设置全局的表名前缀。

例如:

如果一个项目中所有的表名相比于类名都是多了个前缀: tb_ 。这可以使用如下方式配置

mybatis-plus:
  global-config:
    db-config:
      #表名前缀
      table-prefix: tb_

3.2 设置主键生成策略

3.2.0 测试代码

@Test
public void testInsert(){
    User user = new User();
    user.setUserName("三更草堂222");
    user.setPassword("7777");
    int r = userMapper.insert(user);
    System.out.println(r);
}

3.2.1 单独设置

默认情况下使用 MP 插入数据时,如果在我们没有设置主键生成策略的情况下默认的策略是基于雪花算法的自增 id。

如果我们需要使用别的策略可以在定义实体类时,在代表主键的字段上加上@TableId注解,使用其type属性指定主键生成策略。

例如:

我们要设置主键自动增长则可以设置如下

@Data
@NoArgsConstructor
@AllArgsConstructor
public class User {
    @TableId(type = IdType.AUTO)
    private Long id;
    //.....
}

全部主键策略定义在了枚举类IdType中,IdType有如下的取值

  • AUTO

    数据库 ID 自增,依赖于数据库。该类型请确保数据库设置了 ID 自增 否则无效

  • NONE

    未设置主键类型。若在代码中没有手动设置主键,则会根据主键的全局策略自动生成(默认的主键全局策略是基于雪花算法的自增 ID)

  • INPUT

    需要手动设置主键,若不设置。插入操作生成 SQL 语句时,主键这一列的值会是null

  • ASSIGN_ID

    当没有手动设置主键,即实体类中的主键属性为空时,才会自动填充,使用雪花算法

  • ASSIGN_UUID

    当实体类的主键属性为空时,才会自动填充,使用 UUID

3.2.2 全局设置

mybatis-plus:
  global-config:
    db-config:
      # id生成策略 auto为数据库自增
      id-type: auto

3.3 设置字段映射关系

默认情况下 MP 会根据实体类的属性名去映射表的列名。

如果数据库的列表和实体类的属性名不一致了我们可以使用@TableField注解的value属性去设置映射关系。

例如:

如果表中一个列名叫 address 而 实体类中的属性名为 addressStr 则可以使用如下方式进行配置。

@TableField("address")
private String addressStr;

3.4 设置字段和列名的驼峰映射

默认情况下 MP 会开启字段名列名的驼峰映射, 即从经典数据库列名 A_COLUMN(下划线命名) 到经典 Java 属性名 aColumn(驼峰命名) 的类似映射

如果需要关闭我们可以使用如下配置进行关闭。

mybatis-plus:
  configuration:
    #是否开启自动驼峰命名规则(camel case)映射,即从经典数据库列名 A_COLUMN(下划线命名) 到经典 Java 属性名 aColumn(驼峰命名) 的类似映射
    map-underscore-to-camel-case: false

3.5 日志

如果需要打印 MP 操作对应的 SQL 语句等,可以配置日志输出。

配置方式如下:

mybatis-plus:
  configuration:
    # 日志
    log-impl: org.apache.ibatis.logging.stdout.StdOutImpl

4.基本使用

4.1 插入数据

我们可以使用 insert 方法来实现数据的插入。

示例:

@Test
public void testInsert(){
    User user = new User();
    user.setUserName("三更草堂333");
    user.setPassword("7777888");
    int r = userMapper.insert(user);
    System.out.println(r);
}

4.2 删除操作

我们可以使用deleteXXX方法来实现数据的删除。

示例:

@Test
public void testDelete(){
    List<Integer> ids = new ArrayList<>();
    ids.add(5);
    ids.add(6);
    ids.add(7);
    int i = userMapper.deleteBatchIds(ids);
    System.out.println(i);
}
@Test
public void testDeleteById(){
    int i = userMapper.deleteById(8);
    System.out.println(i);
}
@Test
public void testDeleteByMap(){
    Map<String, Object> map = new HashMap<>();
    map.put("name","提姆");
    map.put("age",22);
    int i = userMapper.deleteByMap(map);
    System.out.println(i);
}

4.3 更新操作

我们可以使用updateXXX方法来实现数据的删除。

示例:

@Test
public void testUpdate(){
    //把id为2的用户的年龄改为14
    User user = new User();
    user.setId(2L);
    user.setAge(14);
    int i = userMapper.updateById(user);
    System.out.println(i);
}

5.条件构造器 Wrapper

5.1 概述

我们在实际操作数据库的时候会涉及到很多的条件。所以 MP 为我们提供了一个功能强大的条件构造器 Wrapper 。使用它可以让我们非常方便的构造条件。

其继承体系如下:

image-20210823105447044

在其子类AbstractWrapper中提供了很多用于构造 Where 条件的方法。

AbstractWrapper的子类QueryWrapper则额外提供了用于针对 Select 语法的select方法。可以用来设置查询哪些列。

AbstractWrapper的子类UpdateWrapper则额外提供了用于针对 SET 语法的set方法。可以用来设置对哪些列进行更新。

完整的 AbstractWrapper 方法可以参照:https://baomidou.com/guide/wrapper.html#abstractwrapperopen in new window 介绍是用来干什么的。它的实现类有哪些

QueryWrapper,UpdateWrapper,【LambdaQueryWrapper】

5.2 常用 AbstractWrapper 方法

eq:equals,等于 gt:greater than ,大于 > ge:greater than or equals,大于等于 ≥ lt:less than,小于< le:less than or equals,小于等于 ≤ between:相当于 SQL 中的 BETWEEN like:模糊匹配。like("name","黄"),相当于 SQL 的 name like '%黄%' likeRight:模糊匹配右半边。likeRight("name","黄"),相当于 SQL 的 name like '黄%' likeLeft:模糊匹配左半边。likeLeft("name","黄"),相当于 SQL 的 name like '%黄' notLike:notLike("name","黄"),相当于 SQL 的 name not like '%黄%' isNull isNotNull and:SQL 连接符 AND or:SQL 连接符 OR

in: in(“age",{1,2,3})相当于 age in(1,2,3)

groupBy: groupBy("id","name")相当于 group by id,name

orderByAsc :orderByAsc("id","name")相当于 order by id ASC,name ASC

orderByDesc :orderByDesc ("id","name")相当于 order by id DESC,name DESC

示例一

SQL 语句如下:

SELECT
    id,user_name,PASSWORD,NAME,age,address
FROM
    USER
WHERE
    age > 18 AND address = '狐山'

如果用 Wrapper 写法如下:

@Test
public void testWrapper01(){
    QueryWrapper wrapper = new QueryWrapper();
    wrapper.gt("age",18);
    wrapper.eq("address","狐山");
    List<User> users = userMapper.selectList(wrapper);
    System.out.println(users);
}

示例二

SQL 语句如下:

SELECT
    id,user_name,PASSWORD,NAME,age,address
FROM
    USER
WHERE
    id IN(1,2,3) AND
    age BETWEEN 12 AND 29 AND
    address LIKE '%山%'

如果用 Wrapper 写法如下:

@Test
public void testWrapper02(){
    QueryWrapper<User> wrapper = new QueryWrapper<>();
    wrapper.in("id",1,2,3);
    wrapper.between("age",12,29);
    wrapper.like("address","山");
    List<User> users = userMapper.selectList(wrapper);
    System.out.println(users);
}

示例三

SQL 语句如下:

SELECT
    id,user_name,PASSWORD,NAME,age,address
FROM
    USER
WHERE
    id IN(1,2,3) AND
    age > 10
ORDER BY
    age DESC

如果用 Wrapper 写法如下:

@Test
public void testWrapper03(){
    QueryWrapper<User> queryWrapper = new QueryWrapper<>();
    queryWrapper.in("id",1,2,3);
    queryWrapper.gt("age",10);
    queryWrapper.orderByDesc("age");
    List<User> users = userMapper.selectList(queryWrapper);
    System.out.println(users);
}

5.3 常用 QueryWrapper 方法

QueryWrapper 的 select 可以设置要查询的列。

示例一:

select(String... sqlSelect) 方法的测试为要查询的列名

SQL 语句如下:

SELECT
    id,user_name
FROM
    USER

MP 写法如下:

@Test
public void testSelect01(){
    QueryWrapper<User> queryWrapper = new QueryWrapper<>();
    queryWrapper.select("id","user_name");
    List<User> users = userMapper.selectList(queryWrapper);
    System.out.println(users);
}

示例二:

select(Class<T> entityClass, Predicate<TableFieldInfo> predicate)

方法的第一个参数为实体类的字节码对象,第二个参数为 Predicate 类型,可以使用 lambda 的写法,过滤要查询的字段 (主键除外) 。

SQL 语句如下:

SELECT
    id,user_name
FROM
    USER

MP 写法如下:

@Test
public void testSelect02(){
    QueryWrapper<User> queryWrapper = new QueryWrapper<>();
    queryWrapper.select(User.class, new Predicate<TableFieldInfo>() {
        @Override
        public boolean test(TableFieldInfo tableFieldInfo) {
            return "user_name".equals(tableFieldInfo.getColumn());
        }
    });
    List<User> users = userMapper.selectList(queryWrapper);
    System.out.println(users);
}

示例三:

select(Predicate<TableFieldInfo> predicate)

方法第一个参数为 Predicate 类型,可以使用 lambda 的写法,过滤要查询的字段 (主键除外) 。

SQL 语句如下:

SELECT
    id,user_name,PASSWORD,NAME,age
FROM
    USER

就是不想查询 address 这列,其他列都查询了

MP 写法如下:

@Test
public void testSelect03(){
    QueryWrapper<User> queryWrapper = new QueryWrapper<>(new User());
    queryWrapper.select(new Predicate<TableFieldInfo>() {
        @Override
        public boolean test(TableFieldInfo tableFieldInfo) {
            return !"address".equals(tableFieldInfo.getColumn());
        }
    });
    List<User> users = userMapper.selectList(queryWrapper);
    System.out.println(users);
}

5.4 常用 UpdateWrapper 方法

我们前面在使用 update 方法时需要创建一个实体类对象传入,用来指定要更新的列及对应的值。但是如果需要更新的列比较少时,创建这么一个对象显的有点麻烦和复杂。

我们可以使用 UpdateWrapper 的 set 方法来设置要更新的列及其值。同时这种方式也可以使用 Wrapper 去指定更复杂的更新条件。

示例

SQL 语句如下:

UPDATE
    USER
SET
    age = 99
where
    id > 1

我们想把 id 大于 1 的用户的年龄修改为 99,则可以使用如下写法:

@Test
public void testUpdateWrapper(){
    UpdateWrapper<User> updateWrapper = new UpdateWrapper<>();
    updateWrapper.gt("id",1);
    updateWrapper.set("age",99);
    userMapper.update(null,updateWrapper);
}

5.5 Lambda 条件构造器

我们前面在使用条件构造器时列名都是用字符串的形式去指定。这种方式无法在编译期确定列名的合法性。

所以 MP 提供了一个 Lambda 条件构造器可以让我们直接以实体类的方法引用的形式来指定列名。这样就可以弥补上述缺陷。

示例:

要执行的查询对应的 SQL 如下

SELECT
    id,user_name,PASSWORD,NAME,age,address
FROM
    USER
WHERE
    age > 18 AND address = '狐山'

如果使用之前的条件构造器写法如下

@Test
public void testLambdaWrapper(){
    QueryWrapper<User> queryWrapper = new QueryWrapper();
    queryWrapper.gt("age",18);
    queryWrapper.eq("address","狐山");
    List<User> users = userMapper.selectList(queryWrapper);
}

如果使用 Lambda 条件构造器写法如下

@Test
public void testLambdaWrapper2(){
    LambdaQueryWrapper<User> queryWrapper = new LambdaQueryWrapper<>();
    queryWrapper.gt(User::getAge,18);
    queryWrapper.eq(User::getAddress,"狐山");
    List<User> users = userMapper.selectList(queryWrapper);
}

6.自定义 SQL

虽然 MP 为我们提供了很多常用的方法,并且也提供了条件构造器。但是如果真的遇到了复制的 SQL 时,我们还是需要自己去定义方法,自己去写对应的 SQL,这样 SQL 也更有利于后期维护。

因为 MP 是对 mybatis 做了增强,所以还是支持之前 Mybatis 的方式去自定义方法。

同时也支持在使用 Mybatis 的自定义方法时使用 MP 的条件构造器帮助我们进行条件构造。

接下去我们分别来讲讲。

6.0 准备工作

① 准备数据

CREATE TABLE `orders` (
  `id` bigint(20) NOT NULL AUTO_INCREMENT,
  `price` int(11) DEFAULT NULL COMMENT '价格',
  `remark` varchar(100) DEFAULT NULL COMMENT '备注',
  `user_id` int(11) DEFAULT NULL COMMENT '用户id',
  `update_time` timestamp NULL DEFAULT NULL COMMENT '更新时间',
  `create_time` timestamp NULL DEFAULT NULL COMMENT '创建时间',
  `version` int(11) DEFAULT '1' COMMENT '版本',
  `del_flag` int(1) DEFAULT '0' COMMENT '逻辑删除标识,0-未删除,1-已删除',
  `create_by` varchar(100) DEFAULT NULL COMMENT '创建人',
  `update_by` varchar(100) DEFAULT NULL COMMENT '更新人',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=6 DEFAULT CHARSET=utf8;

/*Data for the table `orders` */

insert  into `orders`(`id`,`price`,`remark`,`user_id`,`update_time`,`create_time`,`version`,`del_flag`,`create_by`,`update_by`) values (1,2000,'无',2,'2021-08-24 21:02:43','2021-08-24 21:02:46',1,0,NULL,NULL),(2,3000,'无',3,'2021-08-24 21:03:32','2021-08-24 21:03:35',1,0,NULL,NULL),(3,4000,'无',2,'2021-08-24 21:03:39','2021-08-24 21:03:41',1,0,NULL,NULL);

② 创建实体类

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Orders  {

    private Long id;

    /**
     * 价格
     */
    private Integer price;

    /**
     * 备注
     */
    private String remark;

    /**
     * 用户id
     */
    private Integer userId;

    /**
     * 更新时间
     */
    private LocalDateTime updateTime;

    /**
     * 创建时间
     */
    private LocalDateTime createTime;

    /**
     * 版本
     */
    private Integer version;

    /**
     * 逻辑删除标识,0-未删除,1-已删除
     */
    private Integer delFlag;


}

6.1 Mybatis 方式

① 定义方法

在 Mapper 接口中定义方法

public interface UserMapper extends BaseMapper<User> {

    User findMyUser(Long id);
}

② 创建 xml

先配置 xml 文件的存放目录

mybatis-plus:
  mapper-locations: classpath*:/mapper/**/*.xml

创建对应的 xml 映射文件

③ 在 xml 映射文件中编写 SQL

创建对应的标签,编写对应的 SQL 语句

<?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.sangeng.mapper.UserMapper">

    <select id="findMyUser" resultType="com.sangeng.domian.User">
       select * from user where id = #{id}
    </select>
</mapper>

6.2 Mybatis 方式结合条件构造器

我们在使用上述方式自定义方法时。如果也希望我们的自定义方法能像 MP 自带方法一样使用条件构造器来进行条件构造的话只需要使用如下方式即可。

① 方法定义中添加 Warpper 类型的参数

添加 Warpper 类型的参数,并且要注意给其指定参数名。

public interface UserMapper extends BaseMapper<User> {

    User findMyUserByWrapper(@Param(Constants.WRAPPER) Wrapper<User> wrapper);
}

② 在 SQL 语句中获取 Warpper 拼接的 SQL 片段进行拼接。

<select id="findMyUserByWrapper" resultType="com.sangeng.domian.User">
    select * from user ${ew.customSqlSegment
</select>

注意:不能使用#{}应该用${}

7.分页查询

7.1 基本分页查询

① 配置分页查询拦截器

@Configuration
public class PageConfig {

    /**
     * 3.4.0之前的版本
     * @return
     */
   /* @Bean
    public PaginationInterceptor paginationInterceptor(){
        return  new PaginationInterceptor();
    }*/

    /**
     * 3.4.0之后版本
     * @return
     */
   @Bean
   public MybatisPlusInterceptor mybatisPlusInterceptor(){
       MybatisPlusInterceptor mybatisPlusInterceptor = new MybatisPlusInterceptor();
       mybatisPlusInterceptor.addInnerInterceptor(new PaginationInnerInterceptor());
       return mybatisPlusInterceptor;
   }
}

② 进行分页查询

@Test
public void testPage(){
    IPage<User> page = new Page<>();
    //设置每页条数
    page.setSize(2);
    //设置查询第几页
    page.setCurrent(1);
    userMapper.selectPage(page, null);
    System.out.println(page.getRecords());//获取当前页的数据
    System.out.println(page.getTotal());//获取总记录数
    System.out.println(page.getCurrent());//当前页码
}

7.2 多表分页查询

如果需要在多表查询时进行分页查询的话,就可以在 mapper 接口中自定义方法,然后让方法接收 Page 对象。

示例:

需求

我们需要去查询 Orders 表,并且要求查询的时候除了要获取到 Orders 表中的字段,还要获取到每个订单的下单用户的用户名。

SQL

SELECT
    o.*,u.`user_name`
FROM
    USER u,orders o
WHERE
    o.`user_id` = u.`id`

实体类修改, 增加一个 userName 属性

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Orders  {
    //省略无关代码
    private String userName;
}

实现

① 定义接口,定义方法

方法第一个参数定义成 Page 类型

public interface OrdersMapp er extends BaseMapper<Orders> {

    IPage<Orders> findAllOrders(Page<Orders> page);
}

在 xml 中不需要关心分页操作,MP 会帮我们完成。

<?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.sangeng.mapper.OrdersMapper">
    <select id="findAllOrders" resultType="com.sangeng.domian.Orders">
        SELECT
            o.*,u.`user_name`
        FROM
            USER u,orders o
        WHERE
            o.`user_id` = u.`id`
    </select>
</mapper>

然后调用方法测试即可

@Autowired
private OrdersMapper ordersMapper;
@Test
public void testOrdersPage(){
    Page<Orders> page = new Page<>();
    //设置每页大小
    page.setSize(2);
    //设置当前页码
    page.setCurrent(2);
    ordersMapper.findAllOrders(page);
    System.out.println(page.getRecords());
    System.out.println(page.getTotal());
}

8.Service 层接口

MP 也为我们提供了 Service 层的实现。我们只需要编写一个接口,继承IService,并创建一个接口实现类继承ServiceImpl,即可使用。

相比于 Mapper 接口,Service 层主要是支持了更多批量操作的方法。

8.1 基本使用

8.1.1 改造前

定义接口

public interface UserService {
    List<User> list();
}

定义实现类

@Service
public class UserServiceImpl implements UserService {
    @Autowired
    private UserMapper userMapper;

    @Override
    public List<User> list() {
        return userMapper.selectList(null);
    }
}

8.1.2 改造后

接口

public interface UserService extends IService<User> {

}

实现类

@Service
public class UserServiceImpl extends ServiceImpl<UserMapper,User> implements UserService {
}

测试

@Autowired
private UserService userService;


@Test
public void testSeervice(){
    List<User> list = userService.list();
    System.out.println(list);
}

8.2 自定义方法

public interface UserService extends IService<User> {


    User test();
}
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper,User> implements UserService {

    @Autowired
    private OrdersMapper ordersMapper;

    @Override
    public User test() {
        UserMapper userMapper = getBaseMapper();
        List<Orders> orders = ordersMapper.selectList(null);
        User user = userMapper.selectById(3);
        //查询用户对于的订单
        QueryWrapper<Orders> wrapper = new QueryWrapper<>();
        wrapper.eq("user_id",3);
        List<Orders> ordersList = ordersMapper.selectList(wrapper);
        return user;
    }
}

9.代码生成器

MP 提供了一个代码生成器,可以让我们一键生成实体类,Mapper 接口,Service,Controller 等全套代码 。使用方式如下

① 依赖

<!--mybatisplus代码生成器-->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>3.4.1</version>
</dependency>
<!--模板引擎-->
<dependency>
    <groupId>org.freemarker</groupId>
    <artifactId>freemarker</artifactId>
</dependency>

② 生成

修改相应配置后执行以下代码即可

public class GeneratorTest {
    @Test
    public void generate() {
        AutoGenerator generator = new AutoGenerator();

        // 全局配置
        GlobalConfig config = new GlobalConfig();
        String projectPath = System.getProperty("user.dir");
        // 设置输出到的目录
        config.setOutputDir(projectPath + "/src/main/java");
        config.setAuthor("sangeng");
        // 生成结束后是否打开文件夹
        config.setOpen(false);

        // 全局配置添加到 generator 上
        generator.setGlobalConfig(config);

        // 数据源配置
        DataSourceConfig dataSourceConfig = new DataSourceConfig();
        dataSourceConfig.setUrl("jdbc:mysql://localhost:3306/mp_db?characterEncoding=utf-8&serverTimezone=UTC");
        dataSourceConfig.setDriverName("com.mysql.cj.jdbc.Driver");
        dataSourceConfig.setUsername("root");
        dataSourceConfig.setPassword("root");

        // 数据源配置添加到 generator
        generator.setDataSource(dataSourceConfig);

        // 包配置, 生成的代码放在哪个包下
        PackageConfig packageConfig = new PackageConfig();
        packageConfig.setParent("com.sangeng.mp.generator");

        // 包配置添加到 generator
        generator.setPackageInfo(packageConfig);

        // 策略配置
        StrategyConfig strategyConfig = new StrategyConfig();
        // 下划线驼峰命名转换
        strategyConfig.setNaming(NamingStrategy.underline_to_camel);
        strategyConfig.setColumnNaming(NamingStrategy.underline_to_camel);
        // 开启lombok
        strategyConfig.setEntityLombokModel(true);
        // 开启RestController
        strategyConfig.setRestControllerStyle(true);
        generator.setStrategy(strategyConfig);
        generator.setTemplateEngine(new FreemarkerTemplateEngine());

        // 开始生成
        generator.execute();
    }
}