From 0743a3120f3c74839254e77cd77ae9dba0a54868 Mon Sep 17 00:00:00 2001 From: jiangping <jp@doumee.com> Date: 星期三, 17 七月 2024 12:03:26 +0800 Subject: [PATCH] 提交 --- server/web/src/main/java/com/doumee/api/web/ApiController.java | 39 +++++++++++++++++++++++++++++++++++++-- 1 files changed, 37 insertions(+), 2 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 f52186c..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,7 +51,29 @@ 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); -- Gitblit v1.9.3