Skip to content

Commit 9b189fd

Browse files
committed
Clean up getServerName() and getServerPort()
1 parent a8aaa55 commit 9b189fd

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

aws-serverless-java-container-core/src/main/java/com/amazonaws/serverless/proxy/internal/servlet/AwsProxyHttpServletRequest.java

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ public String getRequestURI() {
268268
@Override
269269
public StringBuffer getRequestURL() {
270270
String url = "";
271-
url += getHeaderCaseInsensitive(HttpHeaders.HOST);
271+
url += getServerName();
272272
url += "/";
273273
url += getContextPath();
274274
url += "/";
@@ -613,13 +613,18 @@ public String getScheme() {
613613

614614
@Override
615615
public String getServerName() {
616-
return "lambda.amazonaws.com";
616+
String name = getHeaderCaseInsensitive(HttpHeaders.HOST);
617+
618+
if (name == null || name.length() == 0) {
619+
name = "lambda.amazonaws.com";
620+
}
621+
return name;
617622
}
618623

619624

620625
@Override
621626
public int getServerPort() {
622-
return 0;
627+
return getLocalPort();
623628
}
624629

625630

@@ -725,7 +730,14 @@ public String getLocalAddr() {
725730

726731
@Override
727732
public int getLocalPort() {
728-
return 0;
733+
int port = 0;
734+
735+
if ("https".equals(getScheme())) {
736+
port = 443;
737+
} else if ("http".equals(getScheme())) {
738+
port = 80;
739+
}
740+
return port;
729741
}
730742

731743

0 commit comments

Comments
 (0)