update

Author: thinaih

trino-aws-proxy/src/main/java/io/trino/aws/proxy/server/rest/RequestHeadersBuilder.java

@@ -107,6 +107,13 @@ private static class Builder
         private Optional<String> contentSha256 = Optional.empty();
         private Set<ContentType> seenRequestPayloadContentTypes = new HashSet<>();
 
+        private static final Set<ContentType> CHUNKED_CONTENT_TYPES = Set.of(
+                ContentType.AWS_CHUNKED,
+                ContentType.AWS_CHUNKED_IN_W3C_CHUNKED,
+                ContentType.W3C_CHUNKED,
+                ContentType.AWS_CHUNKED_UNSIGNED,
+                ContentType.AWS_CHUNKED_IN_W3C_CHUNKED_UNSIGNED);
+
         private Builder() {}
 
         private static Optional<String> parseHeaderValuesAsSingle(List<String> allValues)
@@ -203,16 +210,11 @@ private String requiredContentSha256()
 
         private void assertContentTypeValid(ContentType actualContentType)
         {
-            if (actualContentType == ContentType.AWS_CHUNKED
-                    || actualContentType == ContentType.AWS_CHUNKED_IN_W3C_CHUNKED
-                    || actualContentType == ContentType.W3C_CHUNKED
-                    || actualContentType == ContentType.AWS_CHUNKED_UNSIGNED
-                    || actualContentType == ContentType.AWS_CHUNKED_IN_W3C_CHUNKED_UNSIGNED) {
+            if (CHUNKED_CONTENT_TYPES.contains(actualContentType)) {
                 if (decodedContentLength.isEmpty()) {
                     throw new WebApplicationException(LENGTH_REQUIRED);
                 }
-                String sha256 = requiredContentSha256();
-                if (actualContentType != ContentType.W3C_CHUNKED && !sha256.startsWith("STREAMING-")) {
+                if (actualContentType != ContentType.W3C_CHUNKED && !requiredContentSha256().startsWith("STREAMING-")) {
                     throw new WebApplicationException(BAD_REQUEST);
                 }
             }