From fa705507ba574c857b1667553737d23b1b7ff495 Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期四, 12 五月 2022 12:06:31 +0800
Subject: [PATCH] 后端模板代码优化

---
 src/main/resources/code/service/app/src/main/java/com/ks/app/aop/AdminApiFilter.java |   27 +++++++++++++++++++--------
 1 files changed, 19 insertions(+), 8 deletions(-)

diff --git a/src/main/resources/code/service/app/src/main/java/com/ks/app/aop/AdminApiFilter.java b/src/main/resources/code/service/app/src/main/java/com/ks/app/aop/AdminApiFilter.java
index 64379ee..e991a30 100644
--- a/src/main/resources/code/service/app/src/main/java/com/ks/app/aop/AdminApiFilter.java
+++ b/src/main/resources/code/service/app/src/main/java/com/ks/app/aop/AdminApiFilter.java
@@ -1,27 +1,23 @@
 package com.ks.app.aop;
 
-import com.ks.app.entity.config.SystemConfigKey;
-import com.ks.app.service.inter.config.SystemConfigService;
+import com.ks.app.entity.AdminUser;
 import com.ks.app.utils.ApiCodeConstant;
 import com.ks.app.utils.SystemInfoUtil;
 import com.ks.app.vo.AcceptAdminData;
-import com.ks.app.vo.AcceptData;
-import net.sf.json.JSONObject;
 import org.aspectj.lang.ProceedingJoinPoint;
 import org.aspectj.lang.annotation.Around;
 import org.aspectj.lang.annotation.Aspect;
 import org.springframework.core.annotation.Order;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContext;
 import org.springframework.stereotype.Component;
 import org.springframework.web.context.request.RequestContextHolder;
 import org.springframework.web.context.request.ServletRequestAttributes;
 import org.yeshi.utils.JsonUtil;
-import org.yeshi.utils.StringUtil;
 
-import javax.annotation.Resource;
-import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpSession;
 import java.io.IOException;
 import java.io.PrintWriter;
-import java.util.*;
 
 //鍚庡彴鎺ュ彛鐩戞帶
 @Component
@@ -46,7 +42,22 @@
         }
 
         if (acceptAdminData != null) {
+            HttpSession session = servletContainer.getRequest().getSession();
+
             acceptAdminData.setSystem(SystemInfoUtil.getAdminSelectedSystem(servletContainer.getRequest().getSession()));
+            if (acceptAdminData.getSystem() == null) {
+                PrintWriter out = servletContainer.getResponse().getWriter();
+                out.print(JsonUtil.loadFalseResult(ApiCodeConstant.CODE_ERROR_IN_SERVER, "鏈�夋嫨绯荤粺"));
+                out.close();
+                return null;
+            }
+            //濡傛灉鐢ㄦ埛鐧诲綍浜嗗皢鐢ㄦ埛淇℃伅鍔犲叆杩涘幓
+            SecurityContext contextSession = (SecurityContext) session.getAttribute("SPRING_SECURITY_CONTEXT");
+            if (contextSession != null) {
+                Authentication authentication = contextSession.getAuthentication();
+                AdminUser adminUser = (AdminUser) authentication.getPrincipal();
+                acceptAdminData.setAdminUser(adminUser);
+            }
         }
 
 

--
Gitblit v1.8.0