Adds config support for TokenRelayFilterFunctions.tokenRelay()

Fixes gh-3176
spring-cloud · Dec 11, 2023 · 9a63aaf · 9a63aaf
1 parent 6163200
commit 9a63aaf
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/...n/java/org/springframework/cloud/gateway/server/mvc/filter/TokenRelayFilterFunctions.java b/...n/java/org/springframework/cloud/gateway/server/mvc/filter/TokenRelayFilterFunctions.java
@@ -16,8 +16,12 @@
 
 package org.springframework.cloud.gateway.server.mvc.filter;
 
+import java.lang.reflect.Method;
 import java.security.Principal;
+import java.util.Arrays;
+import java.util.Collection;
 
+import org.springframework.cloud.gateway.server.mvc.common.Shortcut;
 import org.springframework.security.oauth2.client.OAuth2AuthorizeRequest;
 import org.springframework.security.oauth2.client.OAuth2AuthorizedClient;
 import org.springframework.security.oauth2.client.OAuth2AuthorizedClientManager;
@@ -34,6 +38,7 @@ public abstract class TokenRelayFilterFunctions {
 	private TokenRelayFilterFunctions() {
 	}
 
+	@Shortcut
 	public static HandlerFilterFunction<ServerResponse, ServerResponse> tokenRelay() {
 		return (request, next) -> {
 			Principal principle = request.servletRequest().getUserPrincipal();
@@ -53,4 +58,13 @@ public static HandlerFilterFunction<ServerResponse, ServerResponse> tokenRelay()
 		};
 	}
 
+	class FilterSupplier implements org.springframework.cloud.gateway.server.mvc.filter.FilterSupplier {
+
+		@Override
+		public Collection<Method> get() {
+			return Arrays.asList(TokenRelayFilterFunctions.class.getMethods());
+		}
+
+	}
+
 }
diff --git a/spring-cloud-gateway-server-mvc/src/main/resources/META-INF/spring.factories b/spring-cloud-gateway-server-mvc/src/main/resources/META-INF/spring.factories
@@ -1,6 +1,7 @@
 org.springframework.cloud.gateway.server.mvc.filter.FilterSupplier=\
   org.springframework.cloud.gateway.server.mvc.filter.Bucket4jFilterFunctions.FilterSupplier,\
   org.springframework.cloud.gateway.server.mvc.filter.CircuitBreakerFilterFunctions.FilterSupplier,\
+  org.springframework.cloud.gateway.server.mvc.filter.TokenRelayFilterFunctions.FilterSupplier,\
   org.springframework.cloud.gateway.server.mvc.filter.FilterFunctions.FilterSupplier
 
 org.springframework.cloud.gateway.server.mvc.handler.HandlerSupplier=\

diff --git a/...eway-server-mvc/src/test/resources/application-propertiesbeandefinitionregistrartests.yml b/...eway-server-mvc/src/test/resources/application-propertiesbeandefinitionregistrartests.yml
@@ -32,6 +32,7 @@ spring.cloud.gateway.mvc:
         - Path=/anything/listRoute2
       filters:
         - HttpbinUriResolver=
+        - TokenRelay
         - AddRequestHeader=X-Test,listRoute2
     - id: listRoute3
       uri: lb://httpbin