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 |   54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 54 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 0191920..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
@@ -1,7 +1,14 @@
 package com.doumee.api.web;
 
 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;
 
@@ -15,6 +22,15 @@
 @Slf4j
 public class ApiController {
 
+    @Autowired
+    UsersService usersService;
+    @Autowired
+    MemberService memberService;
+    /**
+     * 鏄惁寮�鍙戣��
+     */
+    @Value("${debug_model}")
+    private Boolean isDebug;
 
     /**
      * 寰楀埌request瀵硅薄
@@ -26,4 +42,42 @@
         return request;
     }
 
+    /**
+     * 鑾峰彇鐢ㄦ埛ID
+     *
+     * @return
+     */
+    protected Long getMemberId() {
+        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