From f11317925fe0319bbacf6c4bc32a3a87d6c1ee9b Mon Sep 17 00:00:00 2001
From: jiangping <jp@doumee.com>
Date: 星期二, 16 七月 2024 17:19:13 +0800
Subject: [PATCH] 提交

---
 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