From c86848ad0a97033f7a8bdcdf901b46bf866d9fb7 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 03 Jun 2020 19:00:05 +0800
Subject: [PATCH] modify

---
 src/main/java/com/xcong/excoin/common/LoginUserUtils.java |   32 +++++++++++++++++++++++++++++++-
 1 files changed, 31 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/common/LoginUserUtils.java b/src/main/java/com/xcong/excoin/common/LoginUserUtils.java
index 7840a12..0c24019 100644
--- a/src/main/java/com/xcong/excoin/common/LoginUserUtils.java
+++ b/src/main/java/com/xcong/excoin/common/LoginUserUtils.java
@@ -1,6 +1,8 @@
 package com.xcong.excoin.common;
 
+import com.xcong.excoin.common.exception.GlobalException;
 import com.xcong.excoin.modules.member.entity.MemberEntity;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.security.core.context.SecurityContextHolder;
 
 /**
@@ -9,9 +11,37 @@
  * @author wzy
  * @date 2020-05-14
  **/
+@Slf4j
 public class LoginUserUtils {
 
+    private static final String ANON = "anonymousUser";
+
     public static MemberEntity getAppLoginUser() {
-        return (MemberEntity) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
+        if (SecurityContextHolder.getContext().getAuthentication().getPrincipal().equals(ANON)) {
+            throw new GlobalException("无法获取登陆信息");
+        } else {
+            return (MemberEntity) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
+        }
+    }
+
+    public static String getAppLoginUserToken() {
+        return (String) SecurityContextHolder.getContext().getAuthentication().getCredentials();
+    }
+
+    /**
+     * mybatis 拦截器专用
+     *
+     * @return MemberEntity
+     */
+    public static MemberEntity getUser() {
+        if (SecurityContextHolder.getContext().getAuthentication() == null) {
+            return null;
+        }
+
+        if (SecurityContextHolder.getContext().getAuthentication().getPrincipal().equals(ANON)) {
+            return null;
+        } else {
+            return (MemberEntity) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
+        }
     }
 }

--
Gitblit v1.9.1