fix 修复 多数据源aop语法错误

疯狂的狮子li

2021-09-28 6e67e1a8496e2a00ebbb55fd1b01e25f1d2c8713

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

@@ -2,8 +2,10 @@

import cn.hutool.http.HttpStatus;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.ruoyi.common.exception.CustomException;
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;
@@ -14,6 +16,7 @@
 * @author Lion Li
 * @date 2021/4/25
 */
@Slf4j
public class CreateAndUpdateMetaObjectHandler implements MetaObjectHandler {

   @Override
@@ -21,36 +24,43 @@
      try {
         //根据属性名字设置要填充的值
         if (metaObject.hasGetter("createTime")) {
            if (metaObject.getValue("createTime") == null) {
               this.setFieldValByName("createTime", new Date(), metaObject);
            }
            this.setFieldValByName("createTime", new Date(), metaObject);
         }
         if (metaObject.hasGetter("createBy")) {
            if (metaObject.getValue("createBy") == null) {
               this.setFieldValByName("createBy", SecurityUtils.getUsername(), metaObject);
            }
            this.setFieldValByName("createBy", getLoginUsername(), metaObject);
         }
      } catch (Exception e) {
         throw new CustomException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
         throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
      }
      updateFill(metaObject);
   }

   @Override
   public void updateFill(MetaObject metaObject) {
      try {
         if (metaObject.hasGetter("updateBy")) {
            if (metaObject.getValue("updateBy") == null) {
               this.setFieldValByName("updateBy", SecurityUtils.getUsername(), metaObject);
            }
            this.setFieldValByName("updateBy", getLoginUsername(), metaObject);
         }
         if (metaObject.hasGetter("updateTime")) {
            if (metaObject.getValue("updateTime") == null) {
               this.setFieldValByName("updateTime", new Date(), metaObject);
            }
            this.setFieldValByName("updateTime", new Date(), metaObject);
         }
      } catch (Exception e) {
         throw new CustomException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
         throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
      }
   }

   /**
    * 获取登录用户名
    */
   private String getLoginUsername() {
      SysUser loginUser;
      try {
         loginUser = SecurityUtils.getUser();
      } catch (Exception e) {
         log.warn("自动注入警告 => 用户未登录");
         return null;
      }
      return loginUser.getUserName();
   }

}