package com.dl.framework.handler;
|
|
import cn.hutool.core.util.ObjectUtil;
|
import cn.hutool.http.HttpStatus;
|
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
|
import com.dl.common.core.domain.BaseEntity;
|
import com.dl.common.core.domain.BaseNewEntity;
|
import com.dl.common.core.domain.model.LoginUser;
|
import com.dl.common.exception.ServiceException;
|
import com.dl.common.helper.LoginHelper;
|
import com.dl.common.utils.StringUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.ibatis.reflection.MetaObject;
|
import org.springframework.stereotype.Component;
|
|
import java.util.Date;
|
import java.util.Objects;
|
|
/**
|
* MP注入处理器
|
*
|
* @author Lion Li
|
* @date 2021/4/25
|
*/
|
@Slf4j
|
public class CreateAndUpdateMetaObjectHandler implements MetaObjectHandler {
|
|
@Override
|
public void insertFill(MetaObject metaObject) {
|
try {
|
if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseEntity) {
|
BaseEntity baseEntity = (BaseEntity) metaObject.getOriginalObject();
|
Date current = ObjectUtil.isNotNull(baseEntity.getCreateTime())
|
? baseEntity.getCreateTime() : new Date();
|
baseEntity.setCreateTime(current);
|
baseEntity.setUpdateTime(current);
|
String username = StringUtils.isNotBlank(baseEntity.getCreateBy())
|
? baseEntity.getCreateBy() : getLoginUsername();
|
// 当前已登录 且 创建人为空 则填充
|
baseEntity.setCreateBy(username);
|
// 当前已登录 且 更新人为空 则填充
|
baseEntity.setUpdateBy(username);
|
}
|
if(ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseNewEntity){
|
BaseNewEntity baseNewEntity = (BaseNewEntity) metaObject.getOriginalObject();
|
Date currentDate = ObjectUtil.isNotNull(baseNewEntity.getCreatedDateTime())
|
? baseNewEntity.getCreatedDateTime() : new Date();
|
baseNewEntity.setCreatedDateTime(currentDate);
|
baseNewEntity.setLastUpdatedDateTime(currentDate);
|
String usernameCreate = StringUtils.isNotBlank(baseNewEntity.getCreatedUserName())
|
? baseNewEntity.getCreatedUserName() : getLoginUser().getUsername();
|
baseNewEntity.setCreatedUserName(usernameCreate);
|
baseNewEntity.setCreatedUserId(Objects.requireNonNull(getLoginUser()).getUserId());
|
baseNewEntity.setLastUpdatedUserId(Objects.requireNonNull(getLoginUser()).getUserId());
|
|
}
|
|
|
} catch (Exception e) {
|
throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
|
}
|
}
|
|
@Override
|
public void updateFill(MetaObject metaObject) {
|
try {
|
if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseEntity) {
|
BaseEntity baseEntity = (BaseEntity) metaObject.getOriginalObject();
|
Date current = new Date();
|
// 更新时间填充(不管为不为空)
|
baseEntity.setUpdateTime(current);
|
|
String username = getLoginUsername();
|
// 当前已登录 更新人填充(不管为不为空)
|
if (StringUtils.isNotBlank(username)) {
|
baseEntity.setUpdateBy(username);
|
}
|
}
|
|
if(ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseNewEntity){
|
BaseNewEntity baseNewEntity = (BaseNewEntity) metaObject.getOriginalObject();
|
Date current = new Date();
|
baseNewEntity.setLastUpdatedDateTime(current);
|
String username = getLoginUsername();
|
if (StringUtils.isNotBlank(username)) {
|
baseNewEntity.setLastUpdatedUserName(username);
|
baseNewEntity.setLastUpdatedUserId(getLoginUser().getUserId());
|
}
|
}
|
|
} catch (Exception e) {
|
throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
|
}
|
}
|
|
/**
|
* 获取登录用户名
|
*/
|
private String getLoginUsername() {
|
LoginUser loginUser;
|
try {
|
loginUser = LoginHelper.getLoginUser();
|
} catch (Exception e) {
|
log.warn("自动注入警告 => 用户未登录");
|
return null;
|
}
|
return ObjectUtil.isNotNull(loginUser) ? loginUser.getUsername() : null;
|
}
|
|
/**
|
* 获取登录用户信息
|
*/
|
private LoginUser getLoginUser() {
|
LoginUser loginUser;
|
try {
|
loginUser = LoginHelper.getLoginUser();
|
} catch (Exception e) {
|
log.warn("自动注入警告 => 用户未登录");
|
return null;
|
}
|
return ObjectUtil.isNotNull(loginUser) ? loginUser : new LoginUser();
|
}
|
|
}
|