package org.ruoyi.handler;
|
|
import cn.hutool.core.util.ObjectUtil;
|
import cn.hutool.http.HttpStatus;
|
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.ibatis.reflection.MetaObject;
|
import org.ruoyi.common.core.domain.model.LoginUser;
|
import org.ruoyi.common.core.exception.ServiceException;
|
import org.ruoyi.common.core.utils.ObjectUtils;
|
import org.ruoyi.common.satoken.utils.LoginHelper;
|
import org.ruoyi.core.domain.BaseEntity;
|
|
import java.util.Date;
|
|
/**
|
* MP注入处理器
|
*
|
* @author Lion Li
|
* @date 2021/4/25
|
*/
|
@Slf4j
|
public class InjectionMetaObjectHandler implements MetaObjectHandler {
|
|
/**
|
* 插入填充方法，用于在插入数据时自动填充实体对象中的创建时间、更新时间、创建人、更新人等信息
|
*
|
* @param metaObject 元对象，用于获取原始对象并进行填充
|
*/
|
@Override
|
public void insertFill(MetaObject metaObject) {
|
try {
|
if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseEntity baseEntity) {
|
// 获取当前时间作为创建时间和更新时间，如果创建时间不为空，则使用创建时间，否则使用当前时间
|
Date current = ObjectUtils.notNull(baseEntity.getCreateTime(), new Date());
|
baseEntity.setCreateTime(current);
|
baseEntity.setUpdateTime(current);
|
|
// 如果创建人为空，则填充当前登录用户的信息
|
if (ObjectUtil.isNull(baseEntity.getCreateBy())) {
|
LoginUser loginUser = getLoginUser();
|
if (ObjectUtil.isNotNull(loginUser)) {
|
Long userId = loginUser.getUserId();
|
// 填充创建人、更新人和创建部门信息
|
baseEntity.setCreateBy(userId);
|
baseEntity.setUpdateBy(userId);
|
baseEntity.setCreateDept(ObjectUtils.notNull(baseEntity.getCreateDept(), loginUser.getDeptId()));
|
}
|
}
|
} else {
|
Date date = new Date();
|
this.strictInsertFill(metaObject, "createTime", Date.class, date);
|
this.strictInsertFill(metaObject, "updateTime", Date.class, date);
|
}
|
} catch (Exception e) {
|
throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
|
}
|
}
|
|
/**
|
* 更新填充方法，用于在更新数据时自动填充实体对象中的更新时间和更新人信息
|
*
|
* @param metaObject 元对象，用于获取原始对象并进行填充
|
*/
|
@Override
|
public void updateFill(MetaObject metaObject) {
|
try {
|
if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseEntity baseEntity) {
|
// 获取当前时间作为更新时间，无论原始对象中的更新时间是否为空都填充
|
Date current = new Date();
|
baseEntity.setUpdateTime(current);
|
|
// 获取当前登录用户的ID，并填充更新人信息
|
Long userId = LoginHelper.getUserId();
|
if (ObjectUtil.isNotNull(userId)) {
|
baseEntity.setUpdateBy(userId);
|
}
|
} else {
|
this.strictUpdateFill(metaObject, "updateTime", Date.class, new Date());
|
}
|
} catch (Exception e) {
|
throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
|
}
|
}
|
|
/**
|
* 获取当前登录用户信息
|
*
|
* @return 当前登录用户的信息，如果用户未登录则返回 null
|
*/
|
private LoginUser getLoginUser() {
|
LoginUser loginUser;
|
try {
|
loginUser = LoginHelper.getLoginUser();
|
} catch (Exception e) {
|
log.warn("自动注入警告 => 用户未登录");
|
return null;
|
}
|
return loginUser;
|
}
|
|
}
|
