diff --git a/pear-common/src/main/java/com/pearadmin/common/tools/secure/SecurityUtil.java b/pear-common/src/main/java/com/pearadmin/common/tools/secure/SecurityUtil.java index 0cad31ebe11534036d35492c774f6846e9b0b486..ec9b25d7919b24ccbf9f05a82ecc45ea5e9fcfdb 100644 --- a/pear-common/src/main/java/com/pearadmin/common/tools/secure/SecurityUtil.java +++ b/pear-common/src/main/java/com/pearadmin/common/tools/secure/SecurityUtil.java @@ -1,5 +1,6 @@ package com.pearadmin.common.tools.secure; + import org.springframework.security.authentication.AnonymousAuthenticationToken; import org.springframework.security.core.Authentication; import org.springframework.security.core.context.SecurityContextHolder; @@ -32,5 +33,12 @@ public class SecurityUtil { Authentication auth = SecurityContextHolder.getContext().getAuthentication(); return !(auth instanceof AnonymousAuthenticationToken); } + /** + * 获取当前登录用户对象 + * retrun SysUser + * */ + public static Object currentUserObj(){ + return SecurityUtil.currentUser().getPrincipal(); + } } diff --git a/pear-modules/pear-generate/src/main/resources/vm/java/controller.java.vm b/pear-modules/pear-generate/src/main/resources/vm/java/controller.java.vm index 6676080ab1912df2b9ad946f43b609f0cf3c6a10..1d1456f84bdbeea58d9ae6108030eba835a2fcd2 100644 --- a/pear-modules/pear-generate/src/main/resources/vm/java/controller.java.vm +++ b/pear-modules/pear-generate/src/main/resources/vm/java/controller.java.vm @@ -16,7 +16,7 @@ import ${packageName}.service.I${ClassName}Service; /** * ${functionName}Controller - * + * * @author ${author} * @date ${datetime} */ @@ -66,6 +66,10 @@ public class ${ClassName}Controller extends BaseController @PreAuthorize("hasPermission('/${moduleName}/${businessName}/add','${permissionPrefix}:add')") public Result save(@RequestBody ${ClassName} ${className}) { + SysUser sysUser = (SysUser)SecurityUtil.currentUserObj(); + ${className}.setCreateTime(LocalDateTime.now()); + ${className}.setCreateBy(sysUser.getUserId()); + ${className}.setCreateName(sysUser.getUserName()); return decide(${className}Service.insert${ClassName}(${className})); } @@ -89,6 +93,10 @@ public class ${ClassName}Controller extends BaseController @PreAuthorize("hasPermission('/${moduleName}/${businessName}/edit','${permissionPrefix}:edit')") public Result update(@RequestBody ${ClassName} ${className}) { + SysUser sysUser = (SysUser)SecurityUtil.currentUserObj(); + ${className}.setUpdateTime(LocalDateTime.now()); + ${className}.setUpdateBy(sysUser.getUserId()); + ${className}.setUpdateName(sysUser.getUserName()); return decide(${className}Service.update${ClassName}(${className})); } diff --git a/pear-modules/pear-system/src/main/java/com/pearadmin/system/controller/SysNoticeController.java b/pear-modules/pear-system/src/main/java/com/pearadmin/system/controller/SysNoticeController.java index c8d429b622c6dc26eeca67f90b20062e78a1d7ce..819529973c29e25b17fb5f654b5b28d6276bd085 100644 --- a/pear-modules/pear-system/src/main/java/com/pearadmin/system/controller/SysNoticeController.java +++ b/pear-modules/pear-system/src/main/java/com/pearadmin/system/controller/SysNoticeController.java @@ -27,7 +27,7 @@ import java.util.Map; /** * 消息控制器 - * + * * @author 就眠仪式 * @date 2021-03-13 */ @@ -76,7 +76,7 @@ public class SysNoticeController extends BaseController SysNotice privateParam = new SysNotice(); privateParam.setType("private"); - privateParam.setAccept(((SysUser) SecurityUtil.currentUser().getPrincipal()).getUserId()); + privateParam.setAccept(((SysUser)SecurityUtil.currentUserObj()).getUserId()); SysNotice noticeParam = new SysNotice(); noticeParam.setType("notice"); diff --git a/pear-modules/pear-system/src/main/java/com/pearadmin/system/controller/SysUserController.java b/pear-modules/pear-system/src/main/java/com/pearadmin/system/controller/SysUserController.java index 12f21b84d0bfabc19d1c6ca6d49dd8063e32bcb0..3b87f519271104629e689645fb656932f1d6cdec 100644 --- a/pear-modules/pear-system/src/main/java/com/pearadmin/system/controller/SysUserController.java +++ b/pear-modules/pear-system/src/main/java/com/pearadmin/system/controller/SysUserController.java @@ -204,7 +204,7 @@ public class SysUserController extends BaseController { @PreAuthorize("hasPermission('/system/user/edit','sys:user:edit')") @Logging(title = "修改头像", describe = "修改头像", type = BusinessType.EDIT) public Result updateAvatar(@RequestBody SysUser sysUser) { - sysUser.setUserId(((SysUser)SecurityUtil.currentUser().getPrincipal()).getUserId()); + sysUser.setUserId(((SysUser)SecurityUtil.currentUserObj()).getUserId()); boolean result = sysUserService.update(sysUser); return decide(result); } @@ -247,7 +247,7 @@ public class SysUserController extends BaseController { @GetMapping("getUserMenu") @ApiOperation(value = "获取用户菜单数据") public List getUserMenu() { - SysUser sysUser = (SysUser) SecurityUtil.currentUser().getPrincipal(); + SysUser sysUser = (SysUser)SecurityUtil.currentUserObj(); List menus = sysUserService.getUserMenu(sysUser.getUsername()); return sysUserService.toUserMenu(menus, "0"); } @@ -286,7 +286,7 @@ public class SysUserController extends BaseController { @GetMapping("center") @ApiOperation(value = "个人资料") public ModelAndView center(Model model) { - SysUser sysUser = (SysUser) SecurityUtil.currentUser().getPrincipal(); + SysUser sysUser = (SysUser) SecurityUtil.currentUserObj(); model.addAttribute("userInfo", sysUserService.getById(sysUser.getUserId())); model.addAttribute("logs", sysLogService.selectTopLoginLog(sysUser.getUsername())); return jumpPage(MODULE_PATH + "center"); diff --git a/pear-modules/pear-system/src/main/java/com/pearadmin/system/service/impl/ISysMailServiceImpl.java b/pear-modules/pear-system/src/main/java/com/pearadmin/system/service/impl/ISysMailServiceImpl.java index c96fff4bf271d5a382ec71aa793e59a3eada5408..fa64d780ae7b9d9654dcabafb10704d27b922e17 100644 --- a/pear-modules/pear-system/src/main/java/com/pearadmin/system/service/impl/ISysMailServiceImpl.java +++ b/pear-modules/pear-system/src/main/java/com/pearadmin/system/service/impl/ISysMailServiceImpl.java @@ -35,7 +35,7 @@ public class ISysMailServiceImpl implements ISysMailService { public Integer save(SysMail sysMail) { if (sendMail(sysMail)) { sysMail.setMailId(SequenceUtil.makeStringId()); - sysMail.setCreateBy(((SysUser) SecurityUtil.currentUser().getPrincipal()).getUsername()); + sysMail.setCreateBy(((SysUser)SecurityUtil.currentUserObj()).getUsername()); return sysMailMapper.insert(sysMail); } else { return 0; diff --git a/pear-modules/pear-system/src/main/java/com/pearadmin/system/service/impl/SysUserServiceImpl.java b/pear-modules/pear-system/src/main/java/com/pearadmin/system/service/impl/SysUserServiceImpl.java index 2f4e8268e657a152ba8dce9bac3ea985fa624183..79455778f035a4c775bb6ff076032f786a9a8b0b 100644 --- a/pear-modules/pear-system/src/main/java/com/pearadmin/system/service/impl/SysUserServiceImpl.java +++ b/pear-modules/pear-system/src/main/java/com/pearadmin/system/service/impl/SysUserServiceImpl.java @@ -165,6 +165,7 @@ public class SysUserServiceImpl implements ISysUserService { * Param: SysUser * Return: 操作结果 * */ + @Override public List getUserRole(String userId){ List allRole = sysRoleMapper.selectList(null); List myRole = sysUserRoleMapper.selectByUserId(userId);