Skip to content

Commit bd60529

Browse files
Chaho12Jaeho Yoo
authored andcommitted
Update ProxyRequestHandler.java
1 parent a120570 commit bd60529

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

gateway-ha/src/main/java/io/trino/gateway/proxyserver/ProxyRequestHandler.java

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -201,8 +201,7 @@ private void performRequest(
201201
for (String name : list(servletRequest.getHeaderNames())) {
202202
for (String value : list(servletRequest.getHeaders(name))) {
203203
// TODO: decide what should and shouldn't be forwarded
204-
if (!name.equalsIgnoreCase("Accept-Encoding")
205-
&& !name.equalsIgnoreCase("Host")
204+
if (!name.equalsIgnoreCase("Host")
206205
&& (addXForwardedHeaders || !name.startsWith("X-Forwarded"))) {
207206
requestBuilder.addHeader(name, value);
208207
}
@@ -262,7 +261,7 @@ else if (servletRequest.getCookies() != null) {
262261

263262
private Response buildResponse(ProxyResponse response, ImmutableList<NewCookie> cookie)
264263
{
265-
Response.ResponseBuilder builder = Response.status(response.statusCode()).entity(response.body());
264+
Response.ResponseBuilder builder = Response.status(response.statusCode()).entity(response.getRawBody());
266265
response.headers().forEach((headerName, value) -> builder.header(headerName.toString(), value));
267266
cookie.forEach(builder::cookie);
268267
return builder.build();
@@ -287,7 +286,7 @@ private FluentFuture<ProxyResponse> executeHttp(Request request)
287286
private ProxyResponse recordBackendForQueryId(Request request, ProxyResponse response, Optional<String> username,
288287
RoutingDestination routingDestination)
289288
{
290-
log.debug("For Request [%s] got Response [%s]", request.getUri(), response.body());
289+
log.debug("For Request [%s] got Response [%s]", request.getUri(), response.getDecompressedBody());
291290

292291
QueryHistoryManager.QueryDetail queryDetail = getQueryDetailsFromRequest(request, username);
293292

0 commit comments

Comments
 (0)