Skip to content

Commit 327e2cd

Browse files
committed
#509 - Polishing.
Formatting and author note. Original pull request: #519.
1 parent 47cefe6 commit 327e2cd

File tree

2 files changed

+28
-27
lines changed

2 files changed

+28
-27
lines changed

src/main/java/org/springframework/hateoas/mvc/ControllerLinkBuilder.java

Lines changed: 23 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,11 @@
1515
*/
1616
package org.springframework.hateoas.mvc;
1717

18+
import static org.springframework.util.StringUtils.*;
19+
1820
import lombok.RequiredArgsConstructor;
1921
import lombok.experimental.Delegate;
2022

21-
import static org.springframework.util.StringUtils.hasText;
22-
2323
import java.lang.reflect.Method;
2424
import java.net.URI;
2525
import java.util.Map;
@@ -32,7 +32,7 @@
3232
import org.springframework.hateoas.core.DummyInvocationUtils;
3333
import org.springframework.hateoas.core.LinkBuilderSupport;
3434
import org.springframework.hateoas.core.MappingDiscoverer;
35-
import org.springframework.http.server.ServletServerHttpRequest;
35+
import org.springframework.http.server.ServletServerHttpRequest;
3636
import org.springframework.util.Assert;
3737
import org.springframework.util.ConcurrentReferenceHashMap;
3838
import org.springframework.web.bind.annotation.RequestMapping;
@@ -53,7 +53,7 @@
5353
* @author Greg Turnquist
5454
* @author Kevin Conaway
5555
* @author Andrew Naydyonock
56-
* @author Oliver Trosien
56+
* @author Oliver Trosien
5757
*/
5858
public class ControllerLinkBuilder extends LinkBuilderSupport<ControllerLinkBuilder> {
5959

@@ -255,29 +255,29 @@ public String toString() {
255255
}
256256

257257
/**
258-
* Returns a {@link UriComponentsBuilder} obtained from the current servlet mapping with
259-
* scheme tweaked in case the request contains an {@code X-Forwarded-Ssl} header, which is not (yet)
260-
* supported by the underlying {@link UriComponentsBuilder}.
258+
* Returns a {@link UriComponentsBuilder} obtained from the current servlet mapping with scheme tweaked in case the
259+
* request contains an {@code X-Forwarded-Ssl} header, which is not (yet) supported by the underlying
260+
* {@link UriComponentsBuilder}.
261261
*
262262
* @return
263263
*/
264264
static UriComponentsBuilder getBuilder() {
265-
266-
HttpServletRequest request = getCurrentRequest();
267-
UriComponentsBuilder builder = UriComponentsBuilder.fromHttpRequest(new ServletServerHttpRequest(request));
268-
269-
// special case handling for X-Forwarded-Ssl:
270-
// apply it, but only if X-Forwarded-Proto is unset.
271-
272-
String forwardedSsl = request.getHeader("X-Forwarded-Ssl");
273-
ForwardedHeader forwarded = ForwardedHeader.of(request.getHeader(ForwardedHeader.NAME));
274-
String proto = hasText(forwarded.getProto()) ? forwarded.getProto() : request.getHeader("X-Forwarded-Proto");
275-
276-
if (!hasText(proto) && hasText(forwardedSsl) && forwardedSsl.equalsIgnoreCase("on")) {
277-
builder.scheme("https");
278-
}
279-
280-
return builder;
265+
266+
HttpServletRequest request = getCurrentRequest();
267+
UriComponentsBuilder builder = UriComponentsBuilder.fromHttpRequest(new ServletServerHttpRequest(request));
268+
269+
// special case handling for X-Forwarded-Ssl:
270+
// apply it, but only if X-Forwarded-Proto is unset.
271+
272+
String forwardedSsl = request.getHeader("X-Forwarded-Ssl");
273+
ForwardedHeader forwarded = ForwardedHeader.of(request.getHeader(ForwardedHeader.NAME));
274+
String proto = hasText(forwarded.getProto()) ? forwarded.getProto() : request.getHeader("X-Forwarded-Proto");
275+
276+
if (!hasText(proto) && hasText(forwardedSsl) && forwardedSsl.equalsIgnoreCase("on")) {
277+
builder.scheme("https");
278+
}
279+
280+
return builder;
281281
}
282282

283283
/**

src/test/java/org/springframework/hateoas/mvc/ControllerLinkBuilderUnitTest.java

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@
5252
* @author Oemer Yildiz
5353
* @author Greg Turnquist
5454
* @author Kevin Conaway
55+
* @author Oliver Trosien
5556
*/
5657
public class ControllerLinkBuilderUnitTest extends TestUtils {
5758

@@ -537,23 +538,23 @@ public void addsOptionalRequestParameterTemplateForMissingValue() {
537538
*/
538539
@Test
539540
public void supportsTwoProxiesAddingXForwardedPort() {
541+
540542
request.addHeader("X-Forwarded-Port", "1443,8443");
541543
request.addHeader("X-Forwarded-Host", "proxy1,proxy2");
542544

543-
Link link = linkTo(PersonControllerImpl.class).withSelfRel();
544-
assertThat(link.getHref(), startsWith("http://proxy1:1443"));
545+
assertThat(linkTo(PersonControllerImpl.class).withSelfRel().getHref(), startsWith("http://proxy1:1443"));
545546
}
546547

547548
/**
548549
* @see #509
549550
*/
550551
@Test
551552
public void resolvesAmbiguousXForwardedHeaders() {
553+
552554
request.addHeader("X-Forwarded-Proto", "http");
553555
request.addHeader("X-Forwarded-Ssl", "on");
554556

555-
Link link = linkTo(PersonControllerImpl.class).withSelfRel();
556-
assertThat(link.getHref(), startsWith("http://"));
557+
assertThat(linkTo(PersonControllerImpl.class).withSelfRel().getHref(), startsWith("http://"));
557558
}
558559

559560
private static UriComponents toComponents(Link link) {

0 commit comments

Comments
 (0)