|
@@ -1,5 +1,6 @@
|
|
|
package com.middle.platform.mybatis.core.handler;
|
|
package com.middle.platform.mybatis.core.handler;
|
|
|
|
|
|
|
|
|
|
+import cn.dev33.satoken.stp.StpUtil;
|
|
|
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
|
|
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
|
|
|
import com.middle.platform.mybatis.core.dataobject.BaseDO;
|
|
import com.middle.platform.mybatis.core.dataobject.BaseDO;
|
|
|
import org.apache.ibatis.reflection.MetaObject;
|
|
import org.apache.ibatis.reflection.MetaObject;
|
|
@@ -25,16 +26,16 @@ public class DefaultDbFieldHandler implements MetaObjectHandler {
|
|
|
if (Objects.isNull(baseDO.getUpdateTime())) {
|
|
if (Objects.isNull(baseDO.getUpdateTime())) {
|
|
|
baseDO.setUpdateTime(current);
|
|
baseDO.setUpdateTime(current);
|
|
|
}
|
|
}
|
|
|
- //todo 添加用户id获取
|
|
|
|
|
-// Long userId = WebFrameworkUtils.getLoginUserId();
|
|
|
|
|
- Long userId = null;
|
|
|
|
|
- // 当前登录用户不为空,创建人为空,则当前登录用户为创建人
|
|
|
|
|
- if (Objects.nonNull(userId) && Objects.isNull(baseDO.getCreateBy())) {
|
|
|
|
|
- baseDO.setCreateBy(userId);
|
|
|
|
|
- }
|
|
|
|
|
- // 当前登录用户不为空,更新人为空,则当前登录用户为更新人
|
|
|
|
|
- if (Objects.nonNull(userId) && Objects.isNull(baseDO.getUpdateBy())) {
|
|
|
|
|
- baseDO.setUpdateBy(userId);
|
|
|
|
|
|
|
+ if (StpUtil.isLogin()) {
|
|
|
|
|
+ Long userId = StpUtil.getLoginIdAsLong();
|
|
|
|
|
+ // 当前登录用户不为空,创建人为空,则当前登录用户为创建人
|
|
|
|
|
+ if (Objects.isNull(baseDO.getCreateBy())) {
|
|
|
|
|
+ baseDO.setCreateBy(userId);
|
|
|
|
|
+ }
|
|
|
|
|
+ // 当前登录用户不为空,更新人为空,则当前登录用户为更新人
|
|
|
|
|
+ if (Objects.isNull(baseDO.getUpdateBy())) {
|
|
|
|
|
+ baseDO.setUpdateBy(userId);
|
|
|
|
|
+ }
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
@@ -47,13 +48,12 @@ public class DefaultDbFieldHandler implements MetaObjectHandler {
|
|
|
setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
|
|
setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- // 当前登录用户不为空,更新人为空,则当前登录用户为更新人
|
|
|
|
|
- Object modifier = getFieldValByName("updater", metaObject);
|
|
|
|
|
- //todo
|
|
|
|
|
-// Long userId = WebFrameworkUtils.getLoginUserId();
|
|
|
|
|
- Long userId = 0L;
|
|
|
|
|
- if (Objects.nonNull(userId) && Objects.isNull(modifier)) {
|
|
|
|
|
- setFieldValByName("updater", userId.toString(), metaObject);
|
|
|
|
|
|
|
+ if (StpUtil.isLogin()) {
|
|
|
|
|
+ Long userId = StpUtil.getLoginIdAsLong();
|
|
|
|
|
+ Object updateBy = getFieldValByName("updateBy", metaObject);
|
|
|
|
|
+ if (Objects.isNull(updateBy)) {
|
|
|
|
|
+ setFieldValByName("updateBy", userId, metaObject);
|
|
|
|
|
+ }
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|