Skip to content

Commit 82b9b64

Browse files
avibhstarburstRandgalt
authored andcommitted
Use GlueException.statusCode when available
1 parent 56ac119 commit 82b9b64

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,8 +86,9 @@ public Response gluePost(@Context Request request, @Context SigningMetadata sign
8686
catch (GlueException e) {
8787
requestLoggingSession.logException(e);
8888

89-
return Response.status(BAD_REQUEST)
89+
return Response.status(e.statusCode() != 0 ? e.statusCode() : BAD_REQUEST.getStatusCode())
9090
.header(X_AMZN_ERROR_TYPE, e.getClass().getSimpleName() + ":" + e.getMessage())
91+
.entity(e)
9192
.build();
9293
}
9394
catch (Exception e) {

trino-aws-proxy-glue/src/test/java/io/trino/aws/proxy/glue/TestingGlueRequestHandler.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,7 @@ public GlueResponse handleRequest(ParsedGlueRequest request, SigningMetadata sig
9494

9595
// TODO: add more cases
9696
case GetDatabaseRequest getDatabaseRequest -> throw EntityNotFoundException.builder()
97+
.statusCode(400)
9798
.message("'%s.%s' not found".formatted(getDatabaseRequest.catalogId(), getDatabaseRequest.name()))
9899
.build();
99100

0 commit comments

Comments
 (0)