|
@@ -7,7 +7,10 @@ import lombok.extern.slf4j.Slf4j;
|
|
|
import org.apache.ibatis.reflection.MetaObject;
|
|
|
import org.dromara.common.core.domain.model.LoginUser;
|
|
|
import org.dromara.common.core.exception.ServiceException;
|
|
|
+import org.dromara.common.mybatis.core.domain.BaseAllPo;
|
|
|
+import org.dromara.common.mybatis.core.domain.BaseCreatePo;
|
|
|
import org.dromara.common.mybatis.core.domain.BaseEntity;
|
|
|
+import org.dromara.common.mybatis.core.domain.BaseTimePo;
|
|
|
import org.dromara.common.satoken.utils.LoginHelper;
|
|
|
|
|
|
import java.util.Date;
|
|
@@ -40,7 +43,29 @@ public class InjectionMetaObjectHandler implements MetaObjectHandler {
|
|
|
baseEntity.setCreateDept(ObjectUtil.isNotNull(baseEntity.getCreateDept())
|
|
|
? baseEntity.getCreateDept() : loginUser.getDeptId());
|
|
|
}
|
|
|
+ } else if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseAllPo po) {
|
|
|
+ Date current = new Date();
|
|
|
+ po.setCreateTime(current);
|
|
|
+ po.setUpdateTime(current);
|
|
|
+
|
|
|
+ LoginUser loginUser = getLoginUser();
|
|
|
+ if (ObjectUtil.isNotNull(loginUser)) {
|
|
|
+ Long userId = loginUser.getUserId();
|
|
|
+ po.setCreateBy(userId);
|
|
|
+ po.setUpdateBy(userId);
|
|
|
+ }
|
|
|
+ } else if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseCreatePo po) {
|
|
|
+ Date current = new Date();
|
|
|
+ po.setCreateTime(current);
|
|
|
+
|
|
|
+ LoginUser loginUser = getLoginUser();
|
|
|
+ if (ObjectUtil.isNotNull(loginUser)) {
|
|
|
+ po.setCreateBy(loginUser.getUserId());
|
|
|
+ }
|
|
|
+ } else if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseTimePo po) {
|
|
|
+ po.setCreateTime(new Date());
|
|
|
}
|
|
|
+
|
|
|
} catch (Exception e) {
|
|
|
throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
|
|
|
}
|
|
@@ -58,6 +83,17 @@ public class InjectionMetaObjectHandler implements MetaObjectHandler {
|
|
|
if (ObjectUtil.isNotNull(loginUser)) {
|
|
|
baseEntity.setUpdateBy(loginUser.getUserId());
|
|
|
}
|
|
|
+ } else if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseAllPo baseEntity) {
|
|
|
+ Date current = new Date();
|
|
|
+ baseEntity.setUpdateTime(current);
|
|
|
+
|
|
|
+ LoginUser loginUser = getLoginUser();
|
|
|
+ if (ObjectUtil.isNotNull(loginUser)) {
|
|
|
+ Long userId = loginUser.getUserId();
|
|
|
+ baseEntity.setUpdateBy(userId);
|
|
|
+ }
|
|
|
+ } else if (ObjectUtil.isNotNull(metaObject) && metaObject.getOriginalObject() instanceof BaseTimePo po) {
|
|
|
+ po.setUpdateTime(new Date());
|
|
|
}
|
|
|
} catch (Exception e) {
|
|
|
throw new ServiceException("自动注入异常 => " + e.getMessage(), HttpStatus.HTTP_UNAUTHORIZED);
|