From 50a30986640f55200656e6a35659c6efae08f583 Mon Sep 17 00:00:00 2001 From: shenym3 Date: Fri, 17 Jan 2025 16:52:44 +0800 Subject: [PATCH] =?UTF-8?q?[fix]:[][=E4=BF=AE=E6=94=B9=E7=99=BB=E9=99=86?= =?UTF-8?q?=E7=9A=84jwk=E9=85=8D=E7=BD=AE=E5=9C=B0=E5=9D=80]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../jeesite/modules/util/vir/AiFileUtil.java | 4 +- .../modules/util/vir/SecurityUtils.java | 41 +++++++++++++++++++ 2 files changed, 42 insertions(+), 3 deletions(-) create mode 100644 src/main/java/com/jeesite/modules/util/vir/SecurityUtils.java diff --git a/src/main/java/com/jeesite/modules/util/vir/AiFileUtil.java b/src/main/java/com/jeesite/modules/util/vir/AiFileUtil.java index 32d5e1d..b2a4122 100644 --- a/src/main/java/com/jeesite/modules/util/vir/AiFileUtil.java +++ b/src/main/java/com/jeesite/modules/util/vir/AiFileUtil.java @@ -11,9 +11,7 @@ import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.io.*; -/** - * @author adward - */ + @Slf4j public class AiFileUtil { private AiFileUtil() { diff --git a/src/main/java/com/jeesite/modules/util/vir/SecurityUtils.java b/src/main/java/com/jeesite/modules/util/vir/SecurityUtils.java new file mode 100644 index 0000000..b6741ac --- /dev/null +++ b/src/main/java/com/jeesite/modules/util/vir/SecurityUtils.java @@ -0,0 +1,41 @@ +package com.vir.ai.admin.utils; + +import cn.hutool.json.JSONObject; +import com.vir.ai.admin.controller.vo.project.JwtUser; +import com.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.ObjectMapper; +import org.apache.commons.lang3.ArrayUtils; +import org.apache.commons.lang3.StringUtils; +import org.springframework.security.core.Authentication; +import org.springframework.security.core.context.SecurityContextHolder; +import org.springframework.web.context.request.RequestContextHolder; +import org.springframework.web.context.request.ServletRequestAttributes; + +import javax.servlet.http.HttpServletRequest; +import java.io.IOException; + + +public class SecurityUtils { + + public static Authentication getAuthentication() { + return SecurityContextHolder.getContext().getAuthentication(); + } + + + public static String getCurrentUser() { + Authentication authentication = SecurityContextHolder.getContext().getAuthentication(); + if (authentication != null) { + Object principal = authentication.getPrincipal(); + /* if (principal instanceof JwtUser) { + return (JwtUser) principal; + }*/ + JSONObject jsonObject = new JSONObject(principal); + Object claims = jsonObject.get("claims"); + JSONObject claimsJsonObject2 = new JSONObject(claims); + return claimsJsonObject2.get("userId").toString(); + } + return null; + } + + +} -- Gitee