| | |
| | | import cn.hutool.http.HttpStatus; |
| | | import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler; |
| | | import com.ruoyi.common.core.domain.BaseEntity; |
| | | import com.ruoyi.common.core.domain.entity.SysUser; |
| | | import com.ruoyi.common.core.domain.model.LoginUser; |
| | | import com.ruoyi.common.exception.ServiceException; |
| | | import com.ruoyi.common.utils.SecurityUtils; |
| | | import com.ruoyi.common.utils.LoginUtils; |
| | | import com.ruoyi.common.utils.StringUtils; |
| | | import lombok.extern.slf4j.Slf4j; |
| | | import org.apache.ibatis.reflection.MetaObject; |
| | |
| | | @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 = new Date(); |
| | | // 创建时间为空 则填充 |
| | | if (ObjectUtil.isNull(baseEntity.getCreateTime())) { |
| | | baseEntity.setCreateTime(current); |
| | | } |
| | | // 更新时间为空 则填充 |
| | | if (ObjectUtil.isNull(baseEntity.getUpdateTime())) { |
| | | baseEntity.setUpdateTime(current); |
| | | } |
| | | String username = getLoginUsername(); |
| | | // 当前已登录 且 创建人为空 则填充 |
| | | if (StringUtils.isNotBlank(username) |
| | | && StringUtils.isBlank(baseEntity.getCreateBy())) { |
| | | baseEntity.setCreateBy(username); |
| | | } |
| | | // 当前已登录 且 更新人为空 则填充 |
| | | if (StringUtils.isNotBlank(username) |
| | | && StringUtils.isBlank(baseEntity.getUpdateBy())) { |
| | | baseEntity.setUpdateBy(username); |
| | | } |
| | | } |
| | | } catch (Exception e) { |
| | | throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED); |
| | | } |
| | | } |
| | | @Override |
| | | public void insertFill(MetaObject metaObject) { |
| | | try { |
| | | if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseEntity) { |
| | | BaseEntity baseEntity = (BaseEntity) metaObject.getOriginalObject(); |
| | | Date current = new Date(); |
| | | // 创建时间为空 则填充 |
| | | if (ObjectUtil.isNull(baseEntity.getCreateTime())) { |
| | | baseEntity.setCreateTime(current); |
| | | } |
| | | // 更新时间为空 则填充 |
| | | if (ObjectUtil.isNull(baseEntity.getUpdateTime())) { |
| | | baseEntity.setUpdateTime(current); |
| | | } |
| | | String username = getLoginUsername(); |
| | | // 当前已登录 且 创建人为空 则填充 |
| | | if (StringUtils.isNotBlank(username) |
| | | && StringUtils.isBlank(baseEntity.getCreateBy())) { |
| | | baseEntity.setCreateBy(username); |
| | | } |
| | | // 当前已登录 且 更新人为空 则填充 |
| | | if (StringUtils.isNotBlank(username) |
| | | && StringUtils.isBlank(baseEntity.getUpdateBy())) { |
| | | baseEntity.setUpdateBy(username); |
| | | } |
| | | } |
| | | } 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(); |
| | | @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); |
| | | } |
| | | } |
| | | } catch (Exception e) { |
| | | throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED); |
| | | } |
| | | } |
| | | } |
| | | } catch (Exception e) { |
| | | throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED); |
| | | } |
| | | } |
| | | |
| | | /** |
| | | * 获取登录用户名 |
| | | */ |
| | | private String getLoginUsername() { |
| | | SysUser user; |
| | | try { |
| | | user = SecurityUtils.getUser(); |
| | | } catch (Exception e) { |
| | | log.warn("自动注入警告 => 用户未登录"); |
| | | return null; |
| | | } |
| | | return user.getUserName(); |
| | | } |
| | | /** |
| | | * 获取登录用户名 |
| | | */ |
| | | private String getLoginUsername() { |
| | | LoginUser loginUser; |
| | | try { |
| | | loginUser = LoginUtils.getLoginUser(); |
| | | } catch (Exception e) { |
| | | log.warn("自动注入警告 => 用户未登录"); |
| | | return null; |
| | | } |
| | | return loginUser.getUsername(); |
| | | } |
| | | |
| | | } |