From 3dcf05955b72998d60a86c5b4ab7c5ca5caa2a62 Mon Sep 17 00:00:00 2001
From: huanghongfa <18228131219@163.com>
Date: 星期六, 19 十二月 2020 17:22:17 +0800
Subject: [PATCH] 修改部分数据

---
 springcloud_k8s_panzhihuazhihuishequ/common/src/main/java/com/panzhihua/common/controller/BaseController.java |   64 +++++++++++++++++++++++++------
 1 files changed, 51 insertions(+), 13 deletions(-)

diff --git a/springcloud_k8s_panzhihuazhihuishequ/common/src/main/java/com/panzhihua/common/controller/BaseController.java b/springcloud_k8s_panzhihuazhihuishequ/common/src/main/java/com/panzhihua/common/controller/BaseController.java
index 71fd6c9..778dbb3 100644
--- a/springcloud_k8s_panzhihuazhihuishequ/common/src/main/java/com/panzhihua/common/controller/BaseController.java
+++ b/springcloud_k8s_panzhihuazhihuishequ/common/src/main/java/com/panzhihua/common/controller/BaseController.java
@@ -1,12 +1,18 @@
 package com.panzhihua.common.controller;
 
+import com.alibaba.fastjson.JSONObject;
+import com.panzhihua.common.constants.Constants;
 import com.panzhihua.common.constants.TokenConstant;
 import com.panzhihua.common.constants.UserConstants;
+import com.panzhihua.common.exceptions.UnAuthenticationException;
 import com.panzhihua.common.model.vos.LoginUserInfoVO;
+import com.panzhihua.common.utlis.AES;
 import io.swagger.models.auth.In;
+import lombok.SneakyThrows;
 import org.springframework.util.ObjectUtils;
 import org.springframework.web.context.request.RequestContextHolder;
 import org.springframework.web.context.request.ServletRequestAttributes;
+import sun.security.krb5.internal.crypto.Aes256;
 
 import javax.servlet.http.HttpServletRequest;
 
@@ -26,29 +32,61 @@
 
     /**
      * 获取登录对象信息
-     * @return
+     * @return 对象userid
      */
     public Long getUserId(){
-        HttpServletRequest request = this.getRequest();
-        Long header = Long.valueOf(request.getHeader(UserConstants.USER_ID));
-        boolean empty = ObjectUtils.isEmpty(header);
-        if (empty) {
-            return null;
-        }
-        return header;
+        LoginUserInfoVO loginUserInfo = this.getLoginUserInfo();
+        Long userId = loginUserInfo.getUserId();
+        return userId;
     }
+
+    /**
+     * 获取登录对象所在社区id
+     * @return 社区id
+     */
+    public Long getCommunityId(){
+        LoginUserInfoVO loginUserInfo = this.getLoginUserInfo();
+        Long communityId = loginUserInfo.getCommunityId();
+        return communityId;
+    }
+
+    /**
+     * 获取登录对象所在小区
+     * @return 小区id
+     */
+    public Long getAreaId(){
+        LoginUserInfoVO loginUserInfo = this.getLoginUserInfo();
+        Long areaId = loginUserInfo.getAreaId();
+        return areaId;
+    }
+
     /**
      * 获取登录token
-     * @return
+     * @return token
      */
     public String getToken(){
         HttpServletRequest request = this.getRequest();
         String header = request.getHeader(TokenConstant.TOKEN_LOGOUT);
-        boolean empty = ObjectUtils.isEmpty(header);
-        if (empty) {
-            return null;
-        }
         return header;
     }
 
+    /**
+     * 获取登录对象所有信息
+     * @return 所有信息
+     */
+    @SneakyThrows
+    public LoginUserInfoVO getLoginUserInfo(){
+        HttpServletRequest request = this.getRequest();
+        String userInfo = request.getHeader(TokenConstant.TOKEN_USERINFO);
+        boolean empty = ObjectUtils.isEmpty(userInfo);
+        if (empty) {
+            throw new UnAuthenticationException("获取登录人信息失败");
+        }
+        byte[] bytes = AES.parseHexStr2Byte(userInfo);
+        byte[] decrypt = AES.decrypt(bytes, Constants.AES_KEY);
+        userInfo=new String(decrypt);
+        LoginUserInfoVO loginUserInfoVO= JSONObject.parseObject(userInfo,LoginUserInfoVO.class);
+        return loginUserInfoVO;
+    }
+
 }

--
Gitblit v1.7.1