DynamicBodySourceContentProvider(DynamicBodySource dynamicBodySource, AtomicLong bytesWritten) { this.dynamicBodySource = dynamicBodySource; this.bytesWritten = bytesWritten; traceToken = getCurrentTraceToken(); }
JettyResponseFuture(JettyHttpClient jettyHttpClient, Request request, org.eclipse.jetty.client.api.Request jettyRequest, ResponseHandler<T, E> responseHandler, AtomicLong bytesWritten, RequestStats stats) { this.jettyHttpClient = requireNonNull(jettyHttpClient, "jettyHttpClientf is null"); this.request = requireNonNull(request, "request is null"); this.jettyRequest = requireNonNull(jettyRequest, "jettyRequest is null"); this.responseHandler = requireNonNull(responseHandler, "responseHandler is null"); this.bytesWritten = requireNonNull(bytesWritten, "bytesWritten is null"); this.stats = requireNonNull(stats, "stats is null"); traceToken = getCurrentTraceToken(); }
@Override protected Runnable wrapTask(Runnable command) { TraceToken token = getCurrentTraceToken(); return () -> { try (TraceTokenScope ignored = registerTraceToken(token)) { command.run(); } }; } }
@Override protected <T> Callable<T> wrapTask(Callable<T> callable) { TraceToken token = getCurrentTraceToken(); return () -> { try (TraceTokenScope ignored = registerTraceToken(token)) { return callable.call(); } }; }
@Override public <T> ListenableFuture<Void> post(final Iterable<T> events) throws IllegalArgumentException { requireNonNull(events, "eventsSupplier is null"); TraceToken token = getCurrentTraceToken(); Request request = preparePost() .setUri(URI.create("v2/event")) .setHeader("User-Agent", nodeInfo.getNodeId()) .setHeader("Content-Type", MEDIA_TYPE_JSON.toString()) .setBodySource((DynamicBodySource) out -> eventWriter.createEventWriter(events.iterator(), token, out)) .build(); return httpClient.executeAsync(request, EVENT_RESPONSE_HANDLER); }
@Override public Request filterRequest(Request request) { requireNonNull(request, "request is null"); TraceToken token = getCurrentTraceToken(); if (token == null) { return request; } String tokenString; if (token.size() == 1) { tokenString = token.toString(); } else { tokenString = TRACE_TOKEN_JSON_CODEC.toJson(token); } return fromRequest(request) .addHeader(TRACETOKEN_HEADER, tokenString) .build(); }
TraceToken traceToken = getCurrentTraceToken(); synchronized (subFutureLock) { Duration backoff = attemptBackoffPolicy.backoff(previousBackoff);