|
@@ -1,6 +1,7 @@
|
|
|
package com.zjugis.framework.mybatis.core.handler;
|
|
|
|
|
|
import com.zjugis.framework.mybatis.core.dataobject.BaseDO;
|
|
|
+import com.zjugis.framework.mybatis.core.dataobject.BaseRmDO;
|
|
|
import com.zjugis.framework.web.core.util.WebFrameworkUtils;
|
|
|
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
|
|
|
import org.apache.ibatis.reflection.MetaObject;
|
|
@@ -19,7 +20,7 @@ public class DefaultDBFieldHandler implements MetaObjectHandler {
|
|
|
|
|
|
@Override
|
|
|
public void insertFill(MetaObject metaObject) {
|
|
|
- if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseDO) {
|
|
|
+ if (Objects.nonNull(metaObject) && (metaObject.getOriginalObject() instanceof BaseDO)) {
|
|
|
BaseDO baseDO = (BaseDO) metaObject.getOriginalObject();
|
|
|
|
|
|
LocalDateTime current = LocalDateTime.now();
|
|
@@ -42,6 +43,30 @@ public class DefaultDBFieldHandler implements MetaObjectHandler {
|
|
|
baseDO.setUpdater(userId.toString());
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+ if (Objects.nonNull(metaObject) && (metaObject.getOriginalObject() instanceof BaseRmDO)) {
|
|
|
+ BaseRmDO baseDO = (BaseRmDO) metaObject.getOriginalObject();
|
|
|
+
|
|
|
+ LocalDateTime current = LocalDateTime.now();
|
|
|
+ // 创建时间为空,则以当前时间为插入时间
|
|
|
+ if (Objects.isNull(baseDO.getCreateTime())) {
|
|
|
+ baseDO.setCreateTime(current);
|
|
|
+ }
|
|
|
+ // 更新时间为空,则以当前时间为更新时间
|
|
|
+ if (Objects.isNull(baseDO.getUpdateTime())) {
|
|
|
+ baseDO.setUpdateTime(current);
|
|
|
+ }
|
|
|
+
|
|
|
+ String userId = WebFrameworkUtils.getLoginUserId();
|
|
|
+ // 当前登录用户不为空,创建人为空,则当前登录用户为创建人
|
|
|
+ if (Objects.nonNull(userId) && Objects.isNull(baseDO.getCreator())) {
|
|
|
+ baseDO.setCreator(userId.toString());
|
|
|
+ }
|
|
|
+ // 当前登录用户不为空,更新人为空,则当前登录用户为更新人
|
|
|
+ if (Objects.nonNull(userId) && Objects.isNull(baseDO.getUpdater())) {
|
|
|
+ baseDO.setUpdater(userId.toString());
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Override
|