update 优化 增加MP注入异常拦截

ruoyi-framework/src/main/java/com/ruoyi/framework/mybatisplus/CreateAndUpdateMetaObjectHandler.java

@@ -1,6 +1,8 @@
 package com.ruoyi.framework.mybatisplus;
 
+import cn.hutool.http.HttpStatus;
 import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
+import com.ruoyi.common.exception.CustomException;
 import com.ruoyi.common.utils.SecurityUtils;
 import org.apache.ibatis.reflection.MetaObject;
 
@@ -8,6 +10,7 @@ import java.util.Date;
 
 /**
  * MP注入处理器
+ *
  * @author Lion Li
  * @date 2021/4/25
  */
@@ -15,6 +18,7 @@ public class CreateAndUpdateMetaObjectHandler implements MetaObjectHandler {
 
 	@Override
 	public void insertFill(MetaObject metaObject) {
+		try {
 			//根据属性名字设置要填充的值
 			if (metaObject.hasGetter("createTime")) {
 				if (metaObject.getValue("createTime") == null) {
@@ -26,10 +30,14 @@ public class CreateAndUpdateMetaObjectHandler implements MetaObjectHandler {
 					this.setFieldValByName("createBy", SecurityUtils.getUsername(), metaObject);
 				}
 			}
+		} catch (Exception e) {
+			throw new CustomException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
+		}
 	}
 
 	@Override
 	public void updateFill(MetaObject metaObject) {
+		try {
 			if (metaObject.hasGetter("updateBy")) {
 				if (metaObject.getValue("updateBy") == null) {
 					this.setFieldValByName("updateBy", SecurityUtils.getUsername(), metaObject);
@@ -40,6 +48,9 @@ public class CreateAndUpdateMetaObjectHandler implements MetaObjectHandler {
 					this.setFieldValByName("updateTime", new Date(), metaObject);
 				}
 			}
+		} catch (Exception e) {
+			throw new CustomException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
+		}
 	}
 
 }