瀏覽代碼

Merge branch 'master' of https://gitee.com/zouxuan/mec

zouxuan 5 年之前
父節點
當前提交
cdfa6f035e

+ 5 - 6
mec-auth/mec-auth-api/src/main/java/com/ym/mec/auth/api/client/SysUserFeignService.java

@@ -1,23 +1,22 @@
 package com.ym.mec.auth.api.client;
 
-import com.ym.mec.auth.api.entity.SysUser;
 import org.springframework.cloud.openfeign.FeignClient;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
-import org.springframework.web.bind.annotation.RequestHeader;
 
 import com.ym.mec.auth.api.dto.SysUserInfo;
-import org.springframework.web.bind.annotation.RequestParam;
+import com.ym.mec.auth.api.entity.SysUser;
+import com.ym.mec.common.config.FeignConfiguration;
 
-@FeignClient(name = "auth-server")
+@FeignClient(name = "auth-server", configuration = FeignConfiguration.class)
 public interface SysUserFeignService {
 
 	@GetMapping(value = "query/{username}")
-	public SysUserInfo getUser(@PathVariable("username") String username, @RequestHeader("from") String from);
+	public SysUserInfo getUser(@PathVariable("username") String username);
 
 	@GetMapping(value = "task/test")
 	public Object test();
 
 	@GetMapping(value = "queryUserInfo")
-	SysUser queryUserInfo(@RequestHeader("Authorization") String access_token);
+	SysUser queryUserInfo();
 }

+ 32 - 0
mec-common/common-core/src/main/java/com/ym/mec/common/config/FeignConfiguration.java

@@ -0,0 +1,32 @@
+package com.ym.mec.common.config;
+
+import java.util.Enumeration;
+
+import javax.servlet.http.HttpServletRequest;
+
+import org.springframework.stereotype.Component;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import feign.RequestInterceptor;
+import feign.RequestTemplate;
+
+@Component
+public class FeignConfiguration implements RequestInterceptor {
+
+	@Override
+	public void apply(RequestTemplate template) {
+		ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+		HttpServletRequest request = attributes.getRequest();
+		Enumeration<String> headerNames = request.getHeaderNames();
+		if (headerNames != null) {
+			while (headerNames.hasMoreElements()) {
+				String name = headerNames.nextElement();
+				String values = request.getHeader(name);
+				template.header(name, values);
+
+			}
+		}
+	}
+
+}