From 678115e978e2eac37874363d89d6db1d9a12f4ec Mon Sep 17 00:00:00 2001
From: k94314517 <8417338+k94314517@user.noreply.gitee.com>
Date: 星期一, 22 七月 2024 18:29:31 +0800
Subject: [PATCH] 代码提交
---
server/web/src/main/java/com/doumee/api/web/ApiController.java | 36 ++++++++++++++++++++++++++----------
1 files changed, 26 insertions(+), 10 deletions(-)
diff --git a/server/web/src/main/java/com/doumee/api/web/ApiController.java b/server/web/src/main/java/com/doumee/api/web/ApiController.java
index 52287ca..6efbc98 100644
--- a/server/web/src/main/java/com/doumee/api/web/ApiController.java
+++ b/server/web/src/main/java/com/doumee/api/web/ApiController.java
@@ -1,5 +1,6 @@
package com.doumee.api.web;
+import com.alibaba.fastjson.JSONObject;
import com.doumee.config.Jwt.JwtTokenUtil;
import com.doumee.core.constants.ResponseStatus;
import com.doumee.core.exception.BusinessException;
@@ -8,6 +9,7 @@
import com.doumee.service.business.MemberService;
import com.doumee.service.business.UsersService;
import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.formula.ptg.MemAreaPtg;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
@@ -54,6 +56,10 @@
return obj != null ? (Long) obj : null;
}
+ protected String getToken() {
+ Object obj = this.getRequest().getAttribute(JwtTokenUtil.HEADER_KEY);
+ return obj != null ? (String) obj : null;
+ }
protected Long getUserId() {
@@ -64,25 +70,35 @@
return obj != null ? (Long) obj : null;
}
protected Users getLoginUserInfo() {
- Long userId = getUserId();
- if(userId== null){
- throw new BusinessException(ResponseStatus.SERVER_ERROR.getCode(),"鏈櫥褰�");
- }
- Users user = usersService.findById(userId);
- return user;
+ Long userId = getUserId();
+ if(userId== null){
+ throw new BusinessException(ResponseStatus.SERVER_ERROR.getCode(),"鏈櫥褰�");
+ }
+ Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserInfo);
+ String userInfo = obj != null ? (String) obj : null;
+ Users user = new Users();
+ if(StringUtils.isNotBlank(userInfo)){
+ user = JSONObject.toJavaObject(JSONObject.parseObject(userInfo),Users.class);
+ }
+ return user;
}
protected Member getLoginMemberInfo() {
Long userId = getMemberId();
if(userId== null){
return null;
}
- Member user = memberService.findById(userId);
- return user;
+ Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserInfo);
+ String userInfo = obj != null ? (String) obj : null;
+ Member member = new Member();
+ if(StringUtils.isNotBlank(userInfo)){
+ member = JSONObject.toJavaObject(JSONObject.parseObject(userInfo),Member.class);
+ }
+ return member;
}
- protected Long getUserType() {
+ protected String getUserType() {
Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserType);
- return obj != null ? (Long) obj : null;
+ return obj != null ? (String) obj : null;
}
}
--
Gitblit v1.9.3