From 7915bccb915fad88e885d54832b120cf2d36a036 Mon Sep 17 00:00:00 2001 From: k94314517 <8417338+k94314517@user.noreply.gitee.com> Date: 星期三, 17 七月 2024 18:26:56 +0800 Subject: [PATCH] 代码提交 --- server/web/src/main/java/com/doumee/api/web/ApiController.java | 43 ++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 42 insertions(+), 1 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 7a5bf06..64e7118 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 @@ -2,8 +2,13 @@ import com.doumee.config.Jwt.JwtTokenUtil; import com.doumee.dao.business.model.Member; +import com.doumee.dao.business.model.Users; +import com.doumee.service.business.MemberService; +import com.doumee.service.business.UsersService; import lombok.extern.slf4j.Slf4j; import org.apache.poi.ss.formula.ptg.MemAreaPtg; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.ServletRequestAttributes; @@ -17,6 +22,15 @@ @Slf4j public class ApiController { + @Autowired + UsersService usersService; + @Autowired + MemberService memberService; + /** + * 鏄惁寮�鍙戣�� + */ + @Value("${debug_model}") + private Boolean isDebug; /** * 寰楀埌request瀵硅薄 @@ -28,7 +42,6 @@ return request; } - /** * 鑾峰彇鐢ㄦ埛ID * @@ -38,5 +51,33 @@ Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserId_Name); return obj != null ? (Long) obj : null; } + protected Long getUserId() { + if(isDebug){ + return 2L; + } + Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserId_Name); + return obj != null ? (Long) obj : null; + } + protected Users getLoginUserInfo() { + Long userId = getUserId(); + if(userId== null){ + return null; + } + Users user = usersService.findById(userId); + return user; + } + protected Member getLoginMemberInfo() { + Long userId = getMemberId(); + if(userId== null){ + return null; + } + Member user = memberService.findById(userId); + return user; + } + + protected Long getUserType() { + Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserType); + return obj != null ? (Long) obj : null; + } } -- Gitblit v1.9.3