private static long getToken(Response response) { String tokenHeader = response.getHeader(PRESTO_PAGE_TOKEN); if (tokenHeader == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_PAGE_TOKEN)); } return Long.parseLong(tokenHeader); }
@Override public ProxyResponse handle(Request request, Response response) { if (response.getStatusCode() == NO_CONTENT.code()) { return new ProxyResponse(response.getHeaders(), new byte[0]); } if (response.getStatusCode() != OK.code()) { throw new ProxyException(format("Bad status code from remote Presto server: %s: %s", response.getStatusCode(), readBody(response))); } String contentType = response.getHeader(CONTENT_TYPE); if (contentType == null) { throw new ProxyException("No Content-Type set in response from remote Presto server"); } if (!MediaType.parse(contentType).is(MEDIA_TYPE_JSON)) { throw new ProxyException("Bad Content-Type from remote Presto server:" + contentType); } try { return new ProxyResponse(response.getHeaders(), toByteArray(response.getInputStream())); } catch (IOException e) { throw new ProxyException("Failed reading response from remote Presto server", e); } }
private static boolean getComplete(Response response) { String bufferComplete = response.getHeader(PRESTO_BUFFER_COMPLETE); if (bufferComplete == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_BUFFER_COMPLETE)); } return Boolean.parseBoolean(bufferComplete); }
@Override public JsonResponse<T> handle(Request request, Response response) { byte[] bytes = readResponseBytes(response); String contentType = response.getHeader(CONTENT_TYPE); if ((contentType == null) || !MediaType.parse(contentType).is(MEDIA_TYPE_JSON)) { return new JsonResponse<>(response.getStatusCode(), response.getStatusMessage(), response.getHeaders(), bytes); } return new JsonResponse<>(response.getStatusCode(), response.getStatusMessage(), response.getHeaders(), jsonCodec, bytes); }
private static String getTaskInstanceId(Response response) { String taskInstanceId = response.getHeader(PRESTO_TASK_INSTANCE_ID); if (taskInstanceId == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_TASK_INSTANCE_ID)); } return taskInstanceId; }
private static long getNextToken(Response response) { String nextTokenHeader = response.getHeader(PRESTO_PAGE_NEXT_TOKEN); if (nextTokenHeader == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_PAGE_NEXT_TOKEN)); } return Long.parseLong(nextTokenHeader); }
@Override public InputStream handle(Request request, io.airlift.http.client.Response response) { try { if (APPLICATION_JSON.equals(response.getHeader(CONTENT_TYPE))) { return response.getInputStream(); } throw new RuntimeException("Response received was not of type " + APPLICATION_JSON); } catch (IOException e) { throw new RuntimeException("Unable to read response from worker", e); } } }
String contentType = response.getHeader(CONTENT_TYPE); if (contentType == null) { throw new PageTransportErrorException(format("%s header is not set: %s", CONTENT_TYPE, response));
private static long getNextToken(Response response) { String nextTokenHeader = response.getHeader(PRESTO_PAGE_NEXT_TOKEN); if (nextTokenHeader == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_PAGE_NEXT_TOKEN)); } return Long.parseLong(nextTokenHeader); }
private static boolean getComplete(Response response) { String bufferComplete = response.getHeader(PRESTO_BUFFER_COMPLETE); if (bufferComplete == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_BUFFER_COMPLETE)); } return Boolean.parseBoolean(bufferComplete); }
private static long getToken(Response response) { String tokenHeader = response.getHeader(PRESTO_PAGE_TOKEN); if (tokenHeader == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_PAGE_TOKEN)); } return Long.parseLong(tokenHeader); }
private static boolean getComplete(Response response) { String bufferComplete = response.getHeader(PRESTO_BUFFER_COMPLETE); if (bufferComplete == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_BUFFER_COMPLETE)); } return Boolean.parseBoolean(bufferComplete); }
private static long getToken(Response response) { String tokenHeader = response.getHeader(PRESTO_PAGE_TOKEN); if (tokenHeader == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_PAGE_TOKEN)); } return Long.parseLong(tokenHeader); }
private static String getTaskInstanceId(Response response) { String taskInstanceId = response.getHeader(PRESTO_TASK_INSTANCE_ID); if (taskInstanceId == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_TASK_INSTANCE_ID)); } return taskInstanceId; }
private static long getNextToken(Response response) { String nextTokenHeader = response.getHeader(PRESTO_PAGE_NEXT_TOKEN); if (nextTokenHeader == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_PAGE_NEXT_TOKEN)); } return Long.parseLong(nextTokenHeader); }
private static long getNextToken(Response response) { String nextTokenHeader = response.getHeader(PRESTO_PAGE_NEXT_TOKEN); if (nextTokenHeader == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_PAGE_NEXT_TOKEN)); } return Long.parseLong(nextTokenHeader); }
private static String getTaskInstanceId(Response response) { String taskInstanceId = response.getHeader(PRESTO_TASK_INSTANCE_ID); if (taskInstanceId == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_TASK_INSTANCE_ID)); } return taskInstanceId; }
private static boolean getComplete(Response response) { String bufferComplete = response.getHeader(PRESTO_BUFFER_COMPLETE); if (bufferComplete == null) { throw new PageTransportErrorException(format("Expected %s header", PRESTO_BUFFER_COMPLETE)); } return Boolean.parseBoolean(bufferComplete); }
private static Duration extractMaxAge(Response response) { String header = response.getHeader(HttpHeaders.CACHE_CONTROL); if (header != null) { CacheControl cacheControl = CacheControl.valueOf(header); if (cacheControl.getMaxAge() > 0) { return new Duration(cacheControl.getMaxAge(), TimeUnit.SECONDS); } } return DEFAULT_DELAY; }
private Duration extractMaxAge(Response response) { String header = response.getHeader(HttpHeaders.CACHE_CONTROL); if (header != null) { CacheControl cacheControl = CacheControl.valueOf(header); if (cacheControl.getMaxAge() > 0) { return new Duration(cacheControl.getMaxAge(), TimeUnit.SECONDS); } } return DEFAULT_DELAY; }