Refine search
private long getTotalCpuTime() { long totalCpuTime = 0; for (URI server : nodes) { URI addressUri = uriBuilderFrom(server).replacePath("/v1/jmx/mbean/java.lang:type=OperatingSystem/ProcessCpuTime").build(); String data = httpClient.execute(prepareGet().setUri(addressUri).build(), createStringResponseHandler()).getBody(); totalCpuTime += parseLong(data.trim()); } return TimeUnit.NANOSECONDS.toNanos(totalCpuTime); }
@Override public Optional<QueryStats> getQueryStats(String queryId) { URI uri = uriBuilderFrom(baseUri).appendPath("/v1/query").appendPath(queryId).build(); Request request = prepareGet().setUri(uri).build(); return httpClient.execute(request, new GetQueryStatsResponseHandler()); }
HttpUriBuilder httpUriBuilder = uriBuilderFrom(taskStatus.getSelf()); URI uri = summarizeTaskInfo ? httpUriBuilder.addParameter("summarize").build() : httpUriBuilder.build(); Request request = prepareGet() .setUri(uri) .setHeader(CONTENT_TYPE, JSON_UTF_8.toString()) .build(); future = httpClient.executeAsync(request, createFullJsonResponseHandler(taskInfoCodec)); currentRequestStartNanos.set(System.nanoTime()); Futures.addCallback(future, new SimpleHttpResponseHandler<>(this, request.getUri(), stats), executor);
@Override public URI createQueryLocation(QueryId queryId) { requireNonNull(queryId, "queryId is null"); return uriBuilderFrom(baseUri) .appendPath("/v1/query") .appendPath(queryId.toString()) .build(); }
private HttpUriBuilder getHttpUriBuilder(TaskStatus taskStatus) { HttpUriBuilder uriBuilder = uriBuilderFrom(taskStatus.getSelf()); if (summarizeTaskInfo) { uriBuilder.addParameter("summarize"); } return uriBuilder; }
@Test public void testReplaceParameters() { URI uri = uriBuilderFrom(URI.create("http://www.example.com:8081/?k1=1&k1=2&k2=3")) .replaceParameter("k1", "4") .build(); assertEquals(uri.toASCIIString(), "http://www.example.com:8081/?k2=3&k1=4"); }
@Test public void testAddMixedEmptyAndNonEmptyParameters() { URI uri = uriBuilder() .scheme("http") .host("www.example.com") .addParameter("pretty") .addParameter("pretty", "true") .addParameter("pretty") .build(); assertEquals(uri.toASCIIString(), "http://www.example.com/?pretty&pretty=true&pretty"); }
@Test public void testAppendRelativePath() { URI uri = uriBuilder() .scheme("http") .host("www.example.com") .appendPath("/a/b/c") .appendPath("x/y/z") .build(); assertEquals(uri.toASCIIString(), "http://www.example.com/a/b/c/x/y/z"); }
@Test public void testDefaultPort() { URI uri = uriBuilderFrom(URI.create("http://www.example.com:8081")) .defaultPort() .build(); assertEquals(uri.toASCIIString(), "http://www.example.com"); }
private static void assertResource(URI baseUri, HttpClient client, String path, String contents) { HttpUriBuilder uriBuilder = uriBuilderFrom(baseUri); StringResponseHandler.StringResponse data = client.execute(prepareGet().setUri(uriBuilder.appendPath(path).build()).build(), createStringResponseHandler()); assertEquals(data.getStatusCode(), HttpStatus.OK.code()); MediaType contentType = MediaType.parse(data.getHeader(HttpHeaders.CONTENT_TYPE)); assertTrue(PLAIN_TEXT_UTF_8.is(contentType), "Expected text/plain but got " + contentType); assertEquals(data.getBody().trim(), contents); }
private JsonNode jsonpRequest(URI uri) throws IOException { uri = uriBuilderFrom(uri) .addParameter("jsonp", "test") .build(); Request request = prepareGet().setUri(uri).build(); StringResponse response = client.execute(request, createStringResponseHandler()); assertEquals(response.getStatusCode(), 200, response.getBody()); assertContentType(response, JSON_UTF_8); String jsonp = response.getBody().trim(); assertTrue(jsonp.startsWith("test("), jsonp); assertTrue(jsonp.endsWith(")"), jsonp); jsonp = jsonp.substring(5, jsonp.length() - 1); return new ObjectMapperProvider().get().readTree(jsonp); }
private void runToCompletion(String sql) { URI uri = uriBuilderFrom(server.getBaseUrl().resolve("/v1/statement")).build(); Request request = preparePost() .setHeader(PRESTO_USER, "user") .setUri(uri) .setBodyGenerator(createStaticBodyGenerator(sql, UTF_8)) .build(); QueryResults queryResults = client.execute(request, createJsonResponseHandler(jsonCodec(QueryResults.class))); while (queryResults.getNextUri() != null) { request = prepareGet() .setHeader(PRESTO_USER, "user") .setUri(queryResults.getNextUri()) .build(); queryResults = client.execute(request, createJsonResponseHandler(jsonCodec(QueryResults.class))); } }
URI uri = HttpUriBuilder.uriBuilderFrom(location).appendPath(String.valueOf(result.getNextToken())).appendPath("acknowledge").build(); httpClient.executeAsync(prepareGet().setUri(uri).build(), new ResponseHandler<Void, RuntimeException>()
@GET @Path("/v1/info") @Produces(APPLICATION_JSON) public void getInfo( @Context HttpServletRequest servletRequest, @Suspended AsyncResponse asyncResponse) { Request.Builder request = prepareGet() .setUri(uriBuilderFrom(remoteUri).replacePath("/v1/info").build()); performRequest(servletRequest, asyncResponse, request, response -> responseWithHeaders(Response.ok(response.getBody()), response)); }
Request request = preparePost() .setUri(uriBuilder.build()) .setHeader(HttpHeaders.CONTENT_TYPE, MediaType.JSON_UTF_8.toString()) .setBodyGenerator(createStaticBodyGenerator(taskUpdateRequestJson)) .build(); ListenableFuture<JsonResponse<TaskInfo>> future = httpClient.executeAsync(request, createFullJsonResponseHandler(taskInfoCodec)); currentRequest = future; currentRequestStartNanos = System.nanoTime();
@Override public synchronized void cancel() { try (SetThreadName ignored = new SetThreadName("HttpRemoteTask-%s", taskId)) { TaskStatus taskStatus = getTaskStatus(); if (taskStatus.getState().isDone()) { return; } // send cancel to task and ignore response HttpUriBuilder uriBuilder = getHttpUriBuilder(taskStatus).addParameter("abort", "false"); Request request = prepareDelete() .setUri(uriBuilder.build()) .build(); scheduleAsyncCleanupRequest(createCleanupBackoff(), request, "cancel"); } }
private synchronized void abort(TaskStatus status) { checkState(status.getState().isDone(), "cannot abort task with an incomplete status"); try (SetThreadName ignored = new SetThreadName("HttpRemoteTask-%s", taskId)) { taskStatusFetcher.updateTaskStatus(status); // send abort to task HttpUriBuilder uriBuilder = getHttpUriBuilder(getTaskStatus()); Request request = prepareDelete() .setUri(uriBuilder.build()) .build(); scheduleAsyncCleanupRequest(createCleanupBackoff(), request, "abort"); } }
public HttpUriBuilder replaceParameter(String name, String... values) { return replaceParameter(name, asList(values)); }
@Test public void testReplaceParameterMultivalued() { URI uri = uriBuilderFrom(URI.create("http://www.example.com/?k1=1&k1=2&k2=3")) .replaceParameter("k1", "a", "b", "c") .build(); assertEquals(uri.toASCIIString(), "http://www.example.com/?k2=3&k1=a&k1=b&k1=c"); }
@Test public void testAddMixedEmptyAndNonEmptyParameters() { URI uri = uriBuilder() .scheme("http") .host("www.example.com") .addParameter("pretty") .addParameter("pretty", "true") .addParameter("pretty") .build(); assertEquals(uri.toASCIIString(), "http://www.example.com/?pretty&pretty=true&pretty"); }