SpringBoot系列之MyBatis Plus自動填充實現

系列博客專欄：SpringBoot2.0系列博客專欄

開發環境

• JDK 1.8

• SpringBoot2.2.1

• Maven 3.2+

• Mysql5.7.36

• 開發工具

  • IntelliJ IDEA

  • smartGit

項目場景

在項目中經常會遇到需要自動填充數據的情況，比如新增一個DO類，裏面可能會有id、create_time、modify_time、create_user等等這些通用的字段，每一個接口我們都要去設置值，不僅代碼冗餘，而且不能統一管理

---

解決方案：

針對這種情況，可以使用Mybatis Plus來對通用數據的自動填充，在我之前的博客中，已經對怎麼在Springboot2.0中集成Mybatis Plus做了比較詳細的描述，下面給一個例子，看看自動填充的實現

創建數據庫表

CREATE TABLE `user` (
  `id` bigint NOT NULL,
  `name` varchar(255) DEFAULT NULL,
  `age` int DEFAULT NULL,
  `email` varchar(255) DEFAULT NULL,
  `create_time` datetime DEFAULT NULL,
  `modify_time` datetime DEFAULT NULL,
  `is_deleted` int DEFAULT '0',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;

通用的BO類，裏面有id、createTime、modifyTime、is_deleted等等，對於主鍵ID有如下的主鍵策略

• ASSIGN_ID（雪花算法）

• ASSIGN_UUID（排除中劃線的UUID）

• AUTO（數據庫ID自增）

• INPUT（插入前自行設置主鍵值）

• NONE（無狀態）

package com.example.mybatisplus.model;


import com.baomidou.mybatisplus.annotation.*;
import lombok.Data;

import java.time.LocalDateTime;

@Data
public class BaseDO {


    @TableId(type = IdType.ASSIGN_ID)
    private Long id;

    @TableField(fill = FieldFill.INSERT)
    private LocalDateTime createTime;

    @TableField(fill = FieldFill.INSERT_UPDATE)
    private LocalDateTime modifyTime;

    @TableLogic(value = "0", delval = "1")
    @TableField(value = "is_deleted", select = false)
    private Boolean deleted;

}

用戶信息的DO類繼承BaseDO

package com.example.mybatisplus.model;


import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import lombok.EqualsAndHashCode;

import java.io.Serializable;

@TableName("user")
@EqualsAndHashCode(callSuper=true)
@Data
public class UserDO extends BaseDO implements Serializable {


    private static final long serialVersionUID =  -2767372720145830784L;

    private String name;
    private Integer age;
    private String email;

}

在application.yml配置mybatis plus

mybatis-plus:
  type-aliases-package: com.example.mybatisplus.*.*.model
  mapper-locations: classpath*:mapper/*/*.xml
  global-config:
    db-config:
      logic-not-delete-value: 1
      logic-delete-value: 0
  configuration:
    map-underscore-to-camel-case: true
    default-statement-timeout: 60
    cache-enabled: true
    log-impl: org.apache.ibatis.logging.slf4j.Slf4jImpl

對於創建日期等字段的自動填充，我們寫一個handler類，實現MetaObjectHandler接口，注意加上@Component，才能被Spring容器管理

package com.example.mybatisplus.common.handlers;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;

@Component
@Slf4j
public class FillMetaObjectHandler implements MetaObjectHandler {


    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("insertFill");
        this.setFieldValByName("createTime" , LocalDateTime.now(), metaObject);
        this.setFieldValByName("modifyTime" , LocalDateTime.now() , metaObject);
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("updateFill");
        this.setFieldValByName("modifyTime" , LocalDateTime.now(), metaObject);
    }
}

Mapper接口

package com.example.mybatisplus.mapper;

import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.example.mybatisplus.common.cache.MyBatisRedisCache;
import com.example.mybatisplus.model.UserDO;
import org.apache.ibatis.annotations.CacheNamespace;

@DS(value = "testDB")
@CacheNamespace(implementation = MyBatisRedisCache.class ,eviction = MyBatisRedisCache.class)
public interface UserMapper extends BaseMapper<UserDO>{

}

業務接口api

package com.example.mybatisplus.service;


import com.baomidou.mybatisplus.extension.service.IService;
import com.example.mybatisplus.model.UserDO;

public interface IUserService extends IService<UserDO> {

}

業務實現類

package com.example.mybatisplus.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.example.mybatisplus.mapper.UserMapper;
import com.example.mybatisplus.model.UserDO;
import com.example.mybatisplus.service.IUserService;
import org.springframework.stereotype.Service;

@Service
public class UserServiceImpl extends ServiceImpl<UserMapper , UserDO> implements IUserService {

}

新增用戶的api接口

  @PostMapping(value = "/user")
    public ResultBean<UserDO> save(@RequestBody UserDto userDto) {
        UserDO user = BeanUtil.copyProperties(userDto , UserDO.class);
        boolean flag = userService.save(user);
        if (flag) return ResultBean.ok(user);
        return ResultBean.badRequest("新增失敗");
    }