From 91c4253f062be7de9872ff3b52216e81bd7fa428 Mon Sep 17 00:00:00 2001 From: 疯狂的狮子li <15040126243@163.com> Date: 星期二, 12 十月 2021 11:47:46 +0800 Subject: [PATCH] update satoken 1.26.0 => 1.27.0 --- ruoyi-framework/src/main/java/com/ruoyi/framework/mybatisplus/CreateAndUpdateMetaObjectHandler.java | 85 +++++++++++++++++++++--------------------- 1 files changed, 42 insertions(+), 43 deletions(-) diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/mybatisplus/CreateAndUpdateMetaObjectHandler.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/mybatisplus/CreateAndUpdateMetaObjectHandler.java index d4ef144..6d9d56a 100644 --- a/ruoyi-framework/src/main/java/com/ruoyi/framework/mybatisplus/CreateAndUpdateMetaObjectHandler.java +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/mybatisplus/CreateAndUpdateMetaObjectHandler.java @@ -1,67 +1,66 @@ package com.ruoyi.framework.mybatisplus; +import cn.hutool.http.HttpStatus; import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler; +import com.ruoyi.common.core.domain.entity.SysUser; +import com.ruoyi.common.exception.ServiceException; import com.ruoyi.common.utils.SecurityUtils; +import lombok.extern.slf4j.Slf4j; import org.apache.ibatis.reflection.MetaObject; import java.util.Date; /** - * @author woo - * @date 2021/3/11 + * MP娉ㄥ叆澶勭悊鍣� + * + * @author Lion Li + * @date 2021/4/25 */ +@Slf4j public class CreateAndUpdateMetaObjectHandler implements MetaObjectHandler { + @Override public void insertFill(MetaObject metaObject) { - //region 澶勭悊鍒涘缓浜轰俊鎭� - Object createBy = this.getFieldValByName("createBy", metaObject); - Object createTime = this.getFieldValByName("createTime", metaObject); - if (createBy == null) { - createBy = SecurityUtils.getUsername(); - this.setFieldValByName("createBy", createBy, metaObject); + try { + //鏍规嵁灞炴�у悕瀛楄缃濉厖鐨勫�� + if (metaObject.hasGetter("createTime")) { + this.setFieldValByName("createTime", new Date(), metaObject); + } + if (metaObject.hasGetter("createBy")) { + this.setFieldValByName("createBy", getLoginUsername(), metaObject); + } + } catch (Exception e) { + throw new ServiceException("鑷姩娉ㄥ叆寮傚父 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED); } - if (createTime == null) { - createTime = new Date(); - this.setFieldValByName("createTime", createTime, metaObject); - } - //endregion - //region 澶勭悊淇敼浜轰俊鎭� - Object updateBy = this.getFieldValByName("updateBy", metaObject); - Object updateTime = this.getFieldValByName("updateTime", metaObject); - if (updateBy == null) { - updateBy = createBy; - this.setFieldValByName("updateBy", updateBy, metaObject); - } - if (updateTime == null) { - updateTime = createTime; - this.setFieldValByName("updateTime", updateTime, metaObject); - } - //endregion + updateFill(metaObject); } @Override public void updateFill(MetaObject metaObject) { - //region 澶勭悊淇敼浜轰俊鎭� - Object updateBy = this.getFieldValByName("updateBy", metaObject); - Object updateTime = this.getFieldValByName("updateTime", metaObject); - if (updateBy == null) { - updateBy = SecurityUtils.getUsername(); - this.setFieldValByName("updateBy", updateBy, metaObject); + try { + if (metaObject.hasGetter("updateBy")) { + this.setFieldValByName("updateBy", getLoginUsername(), metaObject); + } + if (metaObject.hasGetter("updateTime")) { + this.setFieldValByName("updateTime", new Date(), metaObject); + } + } catch (Exception e) { + throw new ServiceException("鑷姩娉ㄥ叆寮傚父 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED); } - if (updateTime == null) { - updateTime = new Date(); - this.setFieldValByName("updateTime", updateTime, metaObject); - } - //endregion } - @Override - public boolean openInsertFill() { - return true; + /** + * 鑾峰彇鐧诲綍鐢ㄦ埛鍚� + */ + private String getLoginUsername() { + SysUser loginUser; + try { + loginUser = SecurityUtils.getUser(); + } catch (Exception e) { + log.warn("鑷姩娉ㄥ叆璀﹀憡 => 鐢ㄦ埛鏈櫥褰�"); + return null; + } + return loginUser.getUserName(); } - @Override - public boolean openUpdateFill() { - return true; - } } -- Gitblit v1.9.3