forked from spring-cloud/spring-cloud-gateway
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add predicate to detect requests made for a Cloud Foundry route servi…
…ce. (spring-cloud#241) See spring-cloudgh-190
- Loading branch information
1 parent
3258204
commit 1eedb66
Showing
2 changed files
with
99 additions
and
0 deletions.
There are no files selected for viewing
38 changes: 38 additions & 0 deletions
38
...ngframework/cloud/gateway/handler/predicate/CloudFoundryRouteServicePredicateFactory.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,38 @@ | ||
package org.springframework.cloud.gateway.handler.predicate; | ||
|
||
import java.util.function.Predicate; | ||
|
||
import org.springframework.web.server.ServerWebExchange; | ||
|
||
/** | ||
* Creates a predicate which indicates if the request is intended for a Cloud Foundry Route Service. | ||
* @see <a href="https://docs.cloudfoundry.org/services/route-services.html">Cloud Foundry Route Service documentation</a>. | ||
* @author Andrew Fitzgerald | ||
*/ | ||
public class CloudFoundryRouteServicePredicateFactory extends | ||
AbstractRoutePredicateFactory<Object> { | ||
|
||
public static final String X_CF_FORWARDED_URL = "X-CF-Forwarded-Url"; | ||
public static final String X_CF_PROXY_SIGNATURE = "X-CF-Proxy-Signature"; | ||
public static final String X_CF_PROXY_METADATA = "X-CF-Proxy-Metadata"; | ||
private final HeaderRoutePredicateFactory factory = new HeaderRoutePredicateFactory(); | ||
|
||
public CloudFoundryRouteServicePredicateFactory() { | ||
super(Object.class); | ||
} | ||
|
||
@Override | ||
public Predicate<ServerWebExchange> apply( | ||
Object unused) { | ||
return headerPredicate(X_CF_FORWARDED_URL) | ||
.and(headerPredicate(X_CF_PROXY_SIGNATURE)) | ||
.and(headerPredicate(X_CF_PROXY_METADATA)); | ||
} | ||
|
||
private Predicate<ServerWebExchange> headerPredicate(String header) { | ||
HeaderRoutePredicateFactory.Config config = factory.newConfig(); | ||
config.setHeader(header); | ||
config.setRegexp(".*"); | ||
return factory.apply(config); | ||
} | ||
} |
61 changes: 61 additions & 0 deletions
61
...amework/cloud/gateway/handler/predicate/CloudFoundryRouteServicePredicateFactoryTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,61 @@ | ||
package org.springframework.cloud.gateway.handler.predicate; | ||
|
||
import static org.assertj.core.api.Assertions.assertThat; | ||
|
||
import java.util.function.Predicate; | ||
|
||
import org.junit.Before; | ||
import org.junit.Test; | ||
import org.springframework.mock.http.server.reactive.MockServerHttpRequest; | ||
import org.springframework.mock.web.server.MockServerWebExchange; | ||
import org.springframework.web.server.ServerWebExchange; | ||
|
||
/** | ||
* @author Andrew Fitzgerald | ||
*/ | ||
public class CloudFoundryRouteServicePredicateFactoryTest { | ||
|
||
private Predicate<ServerWebExchange> predicate; | ||
|
||
@Before | ||
public void setUp() throws Exception { | ||
CloudFoundryRouteServicePredicateFactory factory = new CloudFoundryRouteServicePredicateFactory(); | ||
predicate = factory.apply(factory.newConfig()); | ||
} | ||
|
||
@Test | ||
public void itReturnsTrueWithAllHeadersPresent() { | ||
MockServerHttpRequest request = MockServerHttpRequest.get("someurl") | ||
.header(CloudFoundryRouteServicePredicateFactory.X_CF_FORWARDED_URL, | ||
"url") | ||
.header(CloudFoundryRouteServicePredicateFactory.X_CF_PROXY_METADATA, | ||
"metadata") | ||
.header(CloudFoundryRouteServicePredicateFactory.X_CF_PROXY_SIGNATURE, | ||
"signature") | ||
.build(); | ||
MockServerWebExchange exchange = MockServerWebExchange.from(request); | ||
|
||
assertThat(predicate.test(exchange)).isTrue(); | ||
} | ||
|
||
@Test | ||
public void itReturnsFalseWithAHeadersMissing() { | ||
MockServerHttpRequest request = MockServerHttpRequest.get("someurl") | ||
.header(CloudFoundryRouteServicePredicateFactory.X_CF_FORWARDED_URL, | ||
"url") | ||
.header(CloudFoundryRouteServicePredicateFactory.X_CF_PROXY_METADATA, | ||
"metadata") | ||
.build(); | ||
MockServerWebExchange exchange = MockServerWebExchange.from(request); | ||
|
||
assertThat(predicate.test(exchange)).isFalse(); | ||
} | ||
|
||
@Test | ||
public void itReturnsFalseWithNoHeaders() { | ||
MockServerHttpRequest request = MockServerHttpRequest.get("someurl").build(); | ||
MockServerWebExchange exchange = MockServerWebExchange.from(request); | ||
|
||
assertThat(predicate.test(exchange)).isFalse(); | ||
} | ||
} |