Skip to content

Commit 8134b59

Browse files
avibhstarburstRandgalt
authored andcommitted
Throw AWS InvalidInputException instead of WebApplicationException(based on AWS docs)
1 parent 8d5a922 commit 8134b59

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

trino-aws-proxy-glue/src/main/java/io/trino/aws/proxy/glue/rest/TrinoGlueResource.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
import software.amazon.awssdk.services.glue.model.GlueRequest;
3131
import software.amazon.awssdk.services.glue.model.GlueResponse;
3232
import software.amazon.awssdk.services.glue.model.InternalServiceException;
33+
import software.amazon.awssdk.services.glue.model.InvalidInputException;
3334

3435
import java.io.InputStream;
3536

@@ -61,7 +62,7 @@ public Response gluePost(@Context Request request, @Context SigningMetadata sign
6162
requestLoggingSession.logProperty("request.glue.emulated.key", signingMetadata.credentials().emulated().secretKey());
6263

6364
String target = request.requestHeaders().unmodifiedHeaders().getFirst("x-amz-target")
64-
.orElseThrow(() -> new WebApplicationException(BAD_REQUEST));
65+
.orElseThrow(() -> InvalidInputException.builder().statusCode(BAD_REQUEST.getStatusCode()).build());
6566

6667
requestLoggingSession.logProperty("request.glue.target", target);
6768

@@ -102,7 +103,9 @@ public Response gluePost(@Context Request request, @Context SigningMetadata sign
102103
private GlueRequest unmarshal(Request request, Class<?> clazz, ObjectMapper objectMapper)
103104
{
104105
try {
105-
InputStream inputStream = request.requestContent().inputStream().orElseThrow(() -> new WebApplicationException(BAD_REQUEST));
106+
InputStream inputStream = request.requestContent()
107+
.inputStream()
108+
.orElseThrow(() -> InvalidInputException.builder().statusCode(BAD_REQUEST.getStatusCode()).build());
106109
return (GlueRequest) objectMapper.readValue(inputStream, clazz);
107110
}
108111
catch (Exception e) {

0 commit comments

Comments
 (0)