From e405dcb06aaf338416f64e2d41acbb7043aa396e Mon Sep 17 00:00:00 2001
From: yanyang <rainieyanic@outlook.com>
Date: Thu, 2 Mar 2023 17:02:10 +0800
Subject: [PATCH] commonfield

---
 .../java/com/cnbm/admin/base/BaseSupport.java | 21 ++++++++++---------
 .../java/com/cnbm/admin/entity/LoginUser.java |  2 ++
 .../admin/service/impl/LoginServiceImpl.java  | 10 +++++++++
 3 files changed, 23 insertions(+), 10 deletions(-)

diff --git a/ym-admin/src/main/java/com/cnbm/admin/base/BaseSupport.java b/ym-admin/src/main/java/com/cnbm/admin/base/BaseSupport.java
index 6779778..4211d71 100644
--- a/ym-admin/src/main/java/com/cnbm/admin/base/BaseSupport.java
+++ b/ym-admin/src/main/java/com/cnbm/admin/base/BaseSupport.java
@@ -7,6 +7,7 @@ package com.cnbm.admin.base;
 
 import com.alibaba.fastjson.JSON;
 import com.cnbm.admin.dto.LoginUserDTO;
+import com.cnbm.admin.entity.LoginUser;
 import com.cnbm.admin.enums.WhetherEnum;
 import lombok.Builder;
 import lombok.Data;
@@ -32,18 +33,18 @@ import java.time.LocalDateTime;
 @Service
 public class BaseSupport {
 
-    protected LoginUserDTO getLoginUser() {
+    protected LoginUser getLoginUser() {
         // 后续完善拦截器再使用该方式
 //        LoginUser loginUser = loginUserHolder.get();
 //        if (loginUser != null) {
 //            return loginUser;
 //        }
         HttpSession session = getHttpServletRequest().getSession(false);
-        LoginUserDTO loginUser = null;
+        LoginUser loginUser = null;
         if (session != null) {
-            String loginUserJson = (String) session.getAttribute(LoginUserDTO.HTTP_HEADER_NAME);
+            String loginUserJson = (String) session.getAttribute(LoginUser.HTTP_HEADER_NAME);
             if (StringUtils.isNotBlank(loginUserJson)) {
-                loginUser = JSON.parseObject(loginUserJson, LoginUserDTO.class);
+                loginUser = JSON.parseObject(loginUserJson, LoginUser.class);
                 return loginUser;
             }
         }
@@ -93,11 +94,11 @@ public class BaseSupport {
                 .enabled(WhetherEnum.YES.getValue())
                 .valid(WhetherEnum.YES.getValue())
                 .createTime(LocalDateTime.now())
-                .creatorId(getLoginUser().getUserId())
-                .creatorName(getLoginUser().getUserName())
+                .creatorId(getLoginUser().getSysUserEntity().getId())
+                .creatorName(getLoginUser().getUsername())
                 .updateTime(LocalDateTime.now())
-                .updaterId(getLoginUser().getUserId())
-                .updaterName(getLoginUser().getUserName())
+                .updaterId(getLoginUser().getSysUserEntity().getId())
+                .updaterName(getLoginUser().getUsername())
                 .build();
         BeanUtils.copyProperties(commonField, t, ignoreProperties);
         return t;
@@ -117,8 +118,8 @@ public class BaseSupport {
      */
     public <T extends Serializable> T setUpdateCommonField(T t) {
         CommonField commonField = CommonField.builder()
-                .updaterId(getLoginUser().getUserId())
-                .updaterName(getLoginUser().getUserName())
+                .updaterId(getLoginUser().getSysUserEntity().getId())
+                .updaterName(getLoginUser().getUsername())
                 .updateTime(LocalDateTime.now())
                 .build();
         BeanUtils.copyProperties(commonField, t, "enabled", "valid");
diff --git a/ym-admin/src/main/java/com/cnbm/admin/entity/LoginUser.java b/ym-admin/src/main/java/com/cnbm/admin/entity/LoginUser.java
index 6d1b4f4..7812267 100644
--- a/ym-admin/src/main/java/com/cnbm/admin/entity/LoginUser.java
+++ b/ym-admin/src/main/java/com/cnbm/admin/entity/LoginUser.java
@@ -24,6 +24,8 @@ public class LoginUser implements UserDetails{
 
     private static final long serialVersionUID = 1L;
 
+    public static final String HTTP_HEADER_NAME = "loginUser";
+
     private SysUserEntity sysUserEntity;
 
     private List<String> permissions;
diff --git a/ym-admin/src/main/java/com/cnbm/admin/service/impl/LoginServiceImpl.java b/ym-admin/src/main/java/com/cnbm/admin/service/impl/LoginServiceImpl.java
index cdf977c..3e5fbda 100644
--- a/ym-admin/src/main/java/com/cnbm/admin/service/impl/LoginServiceImpl.java
+++ b/ym-admin/src/main/java/com/cnbm/admin/service/impl/LoginServiceImpl.java
@@ -1,5 +1,7 @@
 package com.cnbm.admin.service.impl;
 
+import com.alibaba.fastjson.JSON;
+import com.cnbm.admin.dto.LoginUserDTO;
 import com.cnbm.admin.entity.LoginUser;
 import com.cnbm.admin.entity.SysLogLoginEntity;
 import com.cnbm.admin.enums.LoginOperationEnum;
@@ -22,8 +24,11 @@ import org.springframework.security.authentication.UsernamePasswordAuthenticatio
 import org.springframework.security.core.Authentication;
 import org.springframework.security.core.context.SecurityContextHolder;
 import org.springframework.stereotype.Service;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
 
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpSession;
 import java.util.Date;
 import java.util.HashMap;
 import java.util.Map;
@@ -81,6 +86,11 @@ public class LoginServiceImpl implements LoginService {
         map.put("token",jwt);
         //把完整的用户信息存入redis  userid作为key
         redisTemplate.opsForValue().set("login:"+userid,loginUser,1, TimeUnit.DAYS);
+
+        //当前登录用户信息存入session中
+        HttpSession session = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest().getSession();
+        session.setAttribute(loginUser.HTTP_HEADER_NAME, JSON.toJSONString(loginUser));
+
         //登录成功
         log.setStatus(LoginStatusEnum.SUCCESS.value());
         log.setCreator(loginUser.getSysUserEntity().getId());