From 3a9d45a7ddb9d82d961fb88c7ec1a14249455234 Mon Sep 17 00:00:00 2001
From: RuoYi <yzz_ivy@163.com>
Date: Mon, 30 Nov 2020 12:49:18 +0800
Subject: [PATCH] 服务之间feign调用传递用户请求头
---
ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/utils/ServletUtils.java | 48 ++++++++++++++++++++++++++++++++++++++++++++----
1 files changed, 44 insertions(+), 4 deletions(-)
diff --git a/ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/utils/ServletUtils.java b/ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/utils/ServletUtils.java
index 7de6fcb..5346483 100644
--- a/ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/utils/ServletUtils.java
+++ b/ruoyi-common/ruoyi-common-core/src/main/java/com/ruoyi/common/core/utils/ServletUtils.java
@@ -1,6 +1,9 @@
package com.ruoyi.common.core.utils;
import java.io.IOException;
+import java.util.Enumeration;
+import java.util.LinkedHashMap;
+import java.util.Map;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
@@ -53,7 +56,14 @@
*/
public static HttpServletRequest getRequest()
{
- return getRequestAttributes().getRequest();
+ try
+ {
+ return getRequestAttributes().getRequest();
+ }
+ catch (Exception e)
+ {
+ return null;
+ }
}
/**
@@ -61,7 +71,14 @@
*/
public static HttpServletResponse getResponse()
{
- return getRequestAttributes().getResponse();
+ try
+ {
+ return getRequestAttributes().getResponse();
+ }
+ catch (Exception e)
+ {
+ return null;
+ }
}
/**
@@ -74,8 +91,31 @@
public static ServletRequestAttributes getRequestAttributes()
{
- RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
- return (ServletRequestAttributes) attributes;
+ try
+ {
+ RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
+ return (ServletRequestAttributes) attributes;
+ }
+ catch (Exception e)
+ {
+ return null;
+ }
+ }
+
+ public static Map<String, String> getHeaders(HttpServletRequest request)
+ {
+ Map<String, String> map = new LinkedHashMap<>();
+ Enumeration<String> enumeration = request.getHeaderNames();
+ if (enumeration != null)
+ {
+ while (enumeration.hasMoreElements())
+ {
+ String key = enumeration.nextElement();
+ String value = request.getHeader(key);
+ map.put(key, value);
+ }
+ }
+ return map;
}
/**
--
Gitblit v1.9.3