Examples with ContentProvider org.eclipse.jetty.client.api.ContentProvider used on opensource projects

Example 1 with ContentProvider

use of org.eclipse.jetty.client.api.ContentProvider in project jetty.project by eclipse.

the class HttpSender method send.

public void send(HttpExchange exchange) {
    if (!queuedToBegin(exchange))
        return;
    Request request = exchange.getRequest();
    ContentProvider contentProvider = request.getContent();
    HttpContent content = this.content = new HttpContent(contentProvider);
    SenderState newSenderState = SenderState.SENDING;
    if (expects100Continue(request))
        newSenderState = content.hasContent() ? SenderState.EXPECTING_WITH_CONTENT : SenderState.EXPECTING;
    out: while (true) {
        SenderState current = senderState.get();
        switch(current) {
            case IDLE:
            case COMPLETED:
                {
                    if (updateSenderState(current, newSenderState))
                        break out;
                    break;
                }
            default:
                {
                    illegalSenderState(current);
                    return;
                }
        }
    }
    // threads so we must set it only after the sender state has been updated
    if (contentProvider instanceof AsyncContentProvider)
        ((AsyncContentProvider) contentProvider).setListener(this);
    if (!beginToHeaders(exchange))
        return;
    sendHeaders(exchange, content, commitCallback);
}

Example 2 with ContentProvider

use of org.eclipse.jetty.client.api.ContentProvider in project jetty.project by eclipse.

the class HttpConnection method normalizeRequest.

protected void normalizeRequest(Request request) {
    HttpVersion version = request.getVersion();
    HttpFields headers = request.getHeaders();
    ContentProvider content = request.getContent();
    ProxyConfiguration.Proxy proxy = destination.getProxy();
    // Make sure the path is there
    String path = request.getPath();
    if (path.trim().length() == 0) {
        path = "/";
        request.path(path);
    }
    URI uri = request.getURI();
    if (proxy instanceof HttpProxy && !HttpClient.isSchemeSecure(request.getScheme()) && uri != null) {
        path = uri.toString();
        request.path(path);
    }
    // If we are HTTP 1.1, add the Host header
    if (version.getVersion() <= 11) {
        if (!headers.containsKey(HttpHeader.HOST.asString()))
            headers.put(getHttpDestination().getHostField());
    }
    // Add content headers
    if (content != null) {
        if (!headers.containsKey(HttpHeader.CONTENT_TYPE.asString())) {
            String contentType = null;
            if (content instanceof ContentProvider.Typed)
                contentType = ((ContentProvider.Typed) content).getContentType();
            if (contentType != null)
                headers.put(HttpHeader.CONTENT_TYPE, contentType);
            else
                headers.put(HttpHeader.CONTENT_TYPE, "application/octet-stream");
        }
        long contentLength = content.getLength();
        if (contentLength >= 0) {
            if (!headers.containsKey(HttpHeader.CONTENT_LENGTH.asString()))
                headers.put(HttpHeader.CONTENT_LENGTH, String.valueOf(contentLength));
        }
    }
    // Cookies
    CookieStore cookieStore = getHttpClient().getCookieStore();
    if (cookieStore != null) {
        StringBuilder cookies = null;
        if (uri != null)
            cookies = convertCookies(cookieStore.get(uri), null);
        cookies = convertCookies(request.getCookies(), cookies);
        if (cookies != null)
            request.header(HttpHeader.COOKIE.asString(), cookies.toString());
    }
    // Authentication
    applyAuthentication(request, proxy != null ? proxy.getURI() : null);
    applyAuthentication(request, uri);
}

Example 3 with ContentProvider

use of org.eclipse.jetty.client.api.ContentProvider in project jetty.project by eclipse.

the class ProxyServlet method service.

@Override
protected void service(final HttpServletRequest request, final HttpServletResponse response) throws ServletException, IOException {
    final int requestId = getRequestId(request);
    String rewrittenTarget = rewriteTarget(request);
    if (_log.isDebugEnabled()) {
        StringBuffer uri = request.getRequestURL();
        if (request.getQueryString() != null)
            uri.append("?").append(request.getQueryString());
        if (_log.isDebugEnabled())
            _log.debug("{} rewriting: {} -> {}", requestId, uri, rewrittenTarget);
    }
    if (rewrittenTarget == null) {
        onProxyRewriteFailed(request, response);
        return;
    }
    final Request proxyRequest = getHttpClient().newRequest(rewrittenTarget).method(request.getMethod()).version(HttpVersion.fromString(request.getProtocol()));
    copyRequestHeaders(request, proxyRequest);
    addProxyHeaders(request, proxyRequest);
    final AsyncContext asyncContext = request.startAsync();
    // We do not timeout the continuation, but the proxy request
    asyncContext.setTimeout(0);
    proxyRequest.timeout(getTimeout(), TimeUnit.MILLISECONDS);
    if (hasContent(request)) {
        if (expects100Continue(request)) {
            DeferredContentProvider deferred = new DeferredContentProvider();
            proxyRequest.content(deferred);
            proxyRequest.attribute(CLIENT_REQUEST_ATTRIBUTE, request);
            proxyRequest.attribute(CONTINUE_ACTION_ATTRIBUTE, (Runnable) () -> {
                try {
                    ContentProvider provider = proxyRequestContent(request, response, proxyRequest);
                    new DelegatingContentProvider(request, proxyRequest, response, provider, deferred).iterate();
                } catch (Throwable failure) {
                    onClientRequestFailure(request, proxyRequest, response, failure);
                }
            });
        } else {
            proxyRequest.content(proxyRequestContent(request, response, proxyRequest));
        }
    }
    sendProxyRequest(request, response, proxyRequest);
}

Example 4 with ContentProvider

use of org.eclipse.jetty.client.api.ContentProvider in project jersey by jersey.

the class JettyConnector method apply.

@Override
public Future<?> apply(final ClientRequest jerseyRequest, final AsyncConnectorCallback callback) {
    final Request jettyRequest = translateRequest(jerseyRequest);
    final Map<String, String> clientHeadersSnapshot = writeOutBoundHeaders(jerseyRequest.getHeaders(), jettyRequest);
    final ContentProvider entity = getStreamProvider(jerseyRequest);
    if (entity != null) {
        jettyRequest.content(entity);
    }
    final AtomicBoolean callbackInvoked = new AtomicBoolean(false);
    final Throwable failure;
    try {
        final CompletableFuture<ClientResponse> responseFuture = new CompletableFuture<ClientResponse>().whenComplete((clientResponse, throwable) -> {
            if (throwable != null && throwable instanceof CancellationException) {
                // take care of future cancellation
                jettyRequest.abort(throwable);
            }
        });
        final AtomicReference<ClientResponse> jerseyResponse = new AtomicReference<>();
        final ByteBufferInputStream entityStream = new ByteBufferInputStream();
        jettyRequest.send(new Response.Listener.Adapter() {

            @Override
            public void onHeaders(final Response jettyResponse) {
                HeaderUtils.checkHeaderChanges(clientHeadersSnapshot, jerseyRequest.getHeaders(), JettyConnector.this.getClass().getName());
                if (responseFuture.isDone()) {
                    if (!callbackInvoked.compareAndSet(false, true)) {
                        return;
                    }
                }
                final ClientResponse response = translateResponse(jerseyRequest, jettyResponse, entityStream);
                jerseyResponse.set(response);
                callback.response(response);
            }

            @Override
            public void onContent(final Response jettyResponse, final ByteBuffer content) {
                try {
                    entityStream.put(content);
                } catch (final InterruptedException ex) {
                    final ProcessingException pe = new ProcessingException(ex);
                    entityStream.closeQueue(pe);
                    // try to complete the future with an exception
                    responseFuture.completeExceptionally(pe);
                    Thread.currentThread().interrupt();
                }
            }

            @Override
            public void onComplete(final Result result) {
                entityStream.closeQueue();
                // try to complete the future with the response only once truly done
                responseFuture.complete(jerseyResponse.get());
            }

            @Override
            public void onFailure(final Response response, final Throwable t) {
                entityStream.closeQueue(t);
                // try to complete the future with an exception
                responseFuture.completeExceptionally(t);
                if (callbackInvoked.compareAndSet(false, true)) {
                    callback.failure(t);
                }
            }
        });
        processContent(jerseyRequest, entity);
        return responseFuture;
    } catch (final Throwable t) {
        failure = t;
    }
    if (callbackInvoked.compareAndSet(false, true)) {
        callback.failure(failure);
    }
    CompletableFuture<Object> future = new CompletableFuture<>();
    future.completeExceptionally(failure);
    return future;
}

Example 5 with ContentProvider

use of org.eclipse.jetty.client.api.ContentProvider in project jersey by jersey.

the class JettyConnector method apply.

@Override
public ClientResponse apply(final ClientRequest jerseyRequest) throws ProcessingException {
    final Request jettyRequest = translateRequest(jerseyRequest);
    final Map<String, String> clientHeadersSnapshot = writeOutBoundHeaders(jerseyRequest.getHeaders(), jettyRequest);
    final ContentProvider entity = getBytesProvider(jerseyRequest);
    if (entity != null) {
        jettyRequest.content(entity);
    }
    try {
        final ContentResponse jettyResponse = jettyRequest.send();
        HeaderUtils.checkHeaderChanges(clientHeadersSnapshot, jerseyRequest.getHeaders(), JettyConnector.this.getClass().getName());
        final javax.ws.rs.core.Response.StatusType status = jettyResponse.getReason() == null ? Statuses.from(jettyResponse.getStatus()) : Statuses.from(jettyResponse.getStatus(), jettyResponse.getReason());
        final ClientResponse jerseyResponse = new ClientResponse(status, jerseyRequest);
        processResponseHeaders(jettyResponse.getHeaders(), jerseyResponse);
        try {
            jerseyResponse.setEntityStream(new HttpClientResponseInputStream(jettyResponse));
        } catch (final IOException e) {
            LOGGER.log(Level.SEVERE, null, e);
        }
        return jerseyResponse;
    } catch (final Exception e) {
        throw new ProcessingException(e);
    }
}