公共字段自动填充

1.在实体类中定义(对需要自动填充的字段加注释@TableField)

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.TableField;
import lombok.Data;
import java.io.Serializable;
import java.time.LocalDateTime;

/**
 * 员工实体类
 */

@Data
public class Employee implements Serializable {

    private static final long serialVersionUID = 1L;

    private Long id;

    private String username;

    private String name;

    private String password;

    private String phone;

    private String sex;

    private String idNumber; //身份证号码

    private Integer status;

    @TableField(fill = FieldFill.INSERT) //插入时填充字段
    private LocalDateTime createTime;

    @TableField(fill = FieldFill.INSERT_UPDATE) //插入和更新时填充字段
    private LocalDateTime updateTime;

    @TableField(fill = FieldFill.INSERT)  //插入时填充字段
    private Long createUser;

    @TableField(fill = FieldFill.INSERT_UPDATE) //插入和更新时填充字段
    private Long updateUser;

}

2.编写BaseContext工具类,基于ThreadLocal封装的工具类

/**
 * 基于ThreadLocal封装的工具类,用于保存和获取当前登录用户id
 */


public class BaseContext {
    private static ThreadLocal threadLocal = new ThreadLocal<>();

    /**
     * 设置值
     * @param id
     */
    public static void setCurrentId(Long id){
        threadLocal.set(id);
    }

    /**
     * 获取值
     * @return
     */
    public static  Long getCurrentId(){
        return threadLocal.get();
    }
}

3.在LoginCheckFilter的doFilter方法中调用BaseContext来获取当前登录用户的id

import com.alibaba.fastjson.JSON;
import com.itheima.reggie.common.BaseContext;
import com.itheima.reggie.common.R;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.AntPathMatcher;

import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

/**
 * 检查用户是否已经完成登录
 */
@WebFilter(filterName = "loginCheckFilter", urlPatterns = "/*")  //filterName(拦截器的名字),urlPatterns(拦截路径)
@Slf4j
public class LoginCheckFilter implements Filter {

    //路径匹配器,支持通配符(比较两个路径是否相同)
    public static final AntPathMatcher PATH_MATCHER = new AntPathMatcher();

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        HttpServletRequest request = (HttpServletRequest) servletRequest;
        HttpServletResponse response = (HttpServletResponse) servletResponse;

        //1。获取本次请求的URI
        String requestURI = request.getRequestURI();     //  /backend/index.html

        log.info("拦截到请求:{}", requestURI);

        //定义不需要处理的请求路径
        String[] urls = new String[]{
                "/employee/login",
                "/employee/logout",
                "/backend/**",
                "/front/**"
        };

        //2.判断本次请求是否需要处理
        boolean check = check(urls,requestURI);

        //3.如果不需要处理,则直接放行
        if(check){
            log.info("本次请求{}不需要处理", requestURI);
            filterChain.doFilter(request,response); // 传递到下一个Filter,如果没有下一个过滤器,则返回当前请求的资源(也就是你在地址栏输入的地址)
            return;
        }

        //4.判断登录状态,如果已登录,则直接放行
        if(request.getSession().getAttribute("employee") != null){
            log.info("用户已登录,用户id为:{}",request.getSession().getAttribute("employee"));

            /**
             * 将id存入BaseContext中
             */
            Long empId = (Long) request.getSession().getAttribute("employee");
            BaseContext.setCurrentId(empId);

            filterChain.doFilter(request,response);
            return;
        }

        log.info("用户未登录");
        //5.如果未登录,则返回未登录结果,通过输出流方式向客户端页面相应数据
        response.getWriter().write(JSON.toJSONString(R.error("NOTLOGIN")));
        return;

    }

    /**
     * 路径匹配,检查本次请求是否需要放行
     * @param urls
     * @param requestURI
     * @return
     */
    public boolean check(String[] urls, String requestURI){
        for (String url : urls) {
            boolean match = PATH_MATCHER.match(url, requestURI);
            if(match){
                return true;
            }
        }
        return false;
    }

}

4.在MyMetaObjectHandler的方法中调用BaseContext获取登录用户的id

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;

//公共字段自动填充

/**
 * 实现步骤:
 * 1、在实体类属性上加入@TableField注解,指定自动填充的策略
 * 2、按照框架要求编写元数据对象处理器,在此类中统一为公共字段赋值,此类需要实现MetaObjectHandler接口
 */

/**
 * 在学习ThreadLocal之前,客户端发送的每次http请求,对应的在服务端都会分配一个新的线程来处理,在处理过程中,以本
 * 项目为例,下面类中的方法都属于相同的一个线程(即三个方法会因为一个请求而在一个线程上执行):
 * 1.LoginCheckFilter的doFilter方法(拦截用户是否登录的拦截器,对任何资源请求时都会调用此方法)
 * 2.EmployeeController的update方法
 * 3.MyMetaObjectHandler的updateFill方法(此方法在执行update数据时,会执行)
 */

/**
 * ThreadLocal并不是一个Thread,而是Thread的局部变量。当使用ThreadLocal维护变量时,ThreadLocal为每个使用该
 * 变量的线程提供了独立的变量副本,所以每一个线程都可以独立的改变自己的副本,而不会影响其他线程对应的副本。
 * ThreadLocal为每个线程提供单独一份存储空间,具有线程隔离的效果,只有在线程内才能获取到对应的值,线程外不能访问
 *
 * ThreadLocal常用方法:
 * public void set(T value)设置当前线程的线程局部变量的值
 * public T get()          返回当前线程所对应的线程局部变量的值
 */

/**
 * 具体操作:
 * 在LoginCheckFilter的doFilter方法中获取当前登录的用户id,并调用ThreadLocal的set方法来设置当前线程的线程局部
 * 变量的值(用户id),然后在MyMetaObjectHandler的updateFill方法中调用ThreadLocal的get方法来获取当前线程所对应的
 * 线程局部变量的值(用户id)
 */
@Component
@Slf4j
public class MyMetaObjecthandler implements MetaObjectHandler {
    /**
     * 插入操作,自动填充
     * @param metaObject
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        log.info("公共字段自动填充【insert】。。。");
        metaObject.setValue("createTime", LocalDateTime.now());
        metaObject.setValue("updateTime",LocalDateTime.now());
        metaObject.setValue("createUser",BaseContext.getCurrentId());
        metaObject.setValue("updateUser",BaseContext.getCurrentId());
    }

    /**
     * 更新操作,自动填充
     * @param metaObject
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("公共字段自动填充【update】。。。");
        metaObject.setValue("updateTime",LocalDateTime.now());
        metaObject.setValue("updateUser",BaseContext.getCurrentId());
    }
}

你可能感兴趣的:(数据库,mybatis,java)