公众号img

nngkxxdp/src/main/java/com/example/nngkxxdp/util/CurrentLoginUtil.java

@@ -0,0 +1,43 @@
+package com.example.nngkxxdp.util;
+
+import com.example.nngkxxdp.entity.ProblemUser;
+
+import javax.servlet.http.Cookie;
+import javax.servlet.http.HttpServletRequest;
+
+/**
+ * @author zhoupeng
+ * @date 2022/7/14
+ */
+public class CurrentLoginUtil {
+
+    public String getToken(HttpServletRequest request, String tokenName) {
+        String token = null;
+        // 1. header
+        token = request.getHeader(tokenName);
+        if (Blank.isNotEmpty(token)) {
+            return token;
+        }
+        // 2. cookie
+        Cookie[] cookies = request.getCookies();
+        if (cookies != null && cookies.length != 0) {
+            for (Cookie cookie : cookies) {
+                if (cookie != null && tokenName.equals(cookie.getName())) {
+                    token = cookie.getValue();
+                    break;
+                }
+            }
+        }
+        if (Blank.isNotEmpty(token)) {
+            return token;
+        }
+        // 3. parameter
+        token = request.getParameter(tokenName);
+        return token;
+    }
+
+//    public ProblemUser getUser(String token){
+//
+//        ProblemUser
+//    }
+}