package com.ruoyi.user.config;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.ruoyi.common.security.service.TokenService;
import com.ruoyi.system.api.model.LoginUserInfo;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.Date;

/**
 * 自动生成时间
 *
 * @author HJL
 */
@Component
public class MyDateObjectHandler implements MetaObjectHandler {
    @Resource
    private TokenService tokenService;

    /**
     * mybatisPlus新增修改自动插入当前时间
     *
     * @author hjl
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        LoginUserInfo loginUser = tokenService.getLoginUserByUser();
        String userName = null == loginUser ? "" : loginUser.getName();
        // 创建时间
        this.setFieldValByName("createTime", new Date(), metaObject);
        // 更新时间
        this.setFieldValByName("updateTime", new Date(), metaObject);
        // 创建人
        this.setFieldValByName("createBy", userName, metaObject);
        this.setFieldValByName("updateBy", userName, metaObject);
        // 更新时间
        this.setFieldValByName("isDelete", 0, metaObject);

    }

    @Override
    public void updateFill(MetaObject metaObject) {
        LoginUserInfo loginUser = tokenService.getLoginUserByUser();
        String userName = null == loginUser ? "" : loginUser.getName();
        // 修改时间为当前时间
        this.setFieldValByName("updateTime", new Date(), metaObject);
        this.setFieldValByName("updateBy", userName, metaObject);
    }
}