From 53e0cba6de254a1d6f19d0287da0b212f4a5ae90 Mon Sep 17 00:00:00 2001 From: jiangping <jp@doumee.com> Date: 星期二, 16 七月 2024 17:22:05 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- server/web/src/main/java/com/doumee/api/web/ApiController.java | 43 +++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 43 insertions(+), 0 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..969c732 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瀵硅薄 @@ -38,5 +52,34 @@ 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