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(); } }