123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674 |
- /*
- * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
- * or more contributor license agreements. Licensed under the Elastic License
- * 2.0 and the Server Side Public License, v 1; you may not use this file except
- * in compliance with, at your election, the Elastic License 2.0 or the Server
- * Side Public License, v 1.
- */
- package org.elasticsearch.http;
- import org.apache.logging.log4j.Level;
- import org.apache.logging.log4j.LogManager;
- import org.apache.logging.log4j.Logger;
- import org.apache.lucene.util.BytesRef;
- import org.elasticsearch.ElasticsearchTimeoutException;
- import org.elasticsearch.ExceptionsHelper;
- import org.elasticsearch.action.ActionListener;
- import org.elasticsearch.action.support.PlainActionFuture;
- import org.elasticsearch.common.Strings;
- import org.elasticsearch.common.component.AbstractLifecycleComponent;
- import org.elasticsearch.common.network.CloseableChannel;
- import org.elasticsearch.common.network.NetworkAddress;
- import org.elasticsearch.common.network.NetworkService;
- import org.elasticsearch.common.recycler.Recycler;
- import org.elasticsearch.common.settings.ClusterSettings;
- import org.elasticsearch.common.settings.Settings;
- import org.elasticsearch.common.transport.BoundTransportAddress;
- import org.elasticsearch.common.transport.NetworkExceptionHelper;
- import org.elasticsearch.common.transport.PortsRange;
- import org.elasticsearch.common.transport.TransportAddress;
- import org.elasticsearch.common.unit.ByteSizeValue;
- import org.elasticsearch.common.util.concurrent.ConcurrentCollections;
- import org.elasticsearch.common.util.concurrent.FutureUtils;
- import org.elasticsearch.common.util.concurrent.ThreadContext;
- import org.elasticsearch.common.xcontent.LoggingDeprecationHandler;
- import org.elasticsearch.core.AbstractRefCounted;
- import org.elasticsearch.core.RefCounted;
- import org.elasticsearch.rest.RestChannel;
- import org.elasticsearch.rest.RestRequest;
- import org.elasticsearch.rest.RestResponse;
- import org.elasticsearch.tasks.Task;
- import org.elasticsearch.telemetry.tracing.Tracer;
- import org.elasticsearch.threadpool.ThreadPool;
- import org.elasticsearch.transport.BindTransportException;
- import org.elasticsearch.transport.TransportSettings;
- import org.elasticsearch.xcontent.NamedXContentRegistry;
- import org.elasticsearch.xcontent.XContentParserConfiguration;
- import java.io.IOException;
- import java.net.InetAddress;
- import java.net.InetSocketAddress;
- import java.nio.channels.CancelledKeyException;
- import java.util.ArrayList;
- import java.util.Arrays;
- import java.util.HashSet;
- import java.util.List;
- import java.util.Map;
- import java.util.Set;
- import java.util.concurrent.ConcurrentHashMap;
- import java.util.concurrent.TimeUnit;
- import java.util.concurrent.atomic.AtomicLong;
- import java.util.concurrent.atomic.AtomicReference;
- import java.util.concurrent.locks.ReadWriteLock;
- import java.util.concurrent.locks.StampedLock;
- import static org.elasticsearch.core.Strings.format;
- import static org.elasticsearch.http.HttpTransportSettings.SETTING_HTTP_BIND_HOST;
- import static org.elasticsearch.http.HttpTransportSettings.SETTING_HTTP_MAX_CONTENT_LENGTH;
- import static org.elasticsearch.http.HttpTransportSettings.SETTING_HTTP_PORT;
- import static org.elasticsearch.http.HttpTransportSettings.SETTING_HTTP_PUBLISH_HOST;
- import static org.elasticsearch.http.HttpTransportSettings.SETTING_HTTP_PUBLISH_PORT;
- import static org.elasticsearch.http.HttpTransportSettings.SETTING_HTTP_SERVER_SHUTDOWN_GRACE_PERIOD;
- public abstract class AbstractHttpServerTransport extends AbstractLifecycleComponent implements HttpServerTransport {
- private static final Logger logger = LogManager.getLogger(AbstractHttpServerTransport.class);
- protected final Settings settings;
- public final HttpHandlingSettings handlingSettings;
- protected final NetworkService networkService;
- protected final Recycler<BytesRef> recycler;
- protected final ThreadPool threadPool;
- protected final Dispatcher dispatcher;
- protected final CorsHandler corsHandler;
- private final XContentParserConfiguration parserConfig;
- protected final PortsRange port;
- protected final ByteSizeValue maxContentLength;
- private final String[] bindHosts;
- private final String[] publishHosts;
- private volatile BoundTransportAddress boundAddress;
- private final AtomicLong totalChannelsAccepted = new AtomicLong();
- private final Map<HttpChannel, RequestTrackingHttpChannel> httpChannels = new ConcurrentHashMap<>();
- private final PlainActionFuture<Void> allClientsClosedListener = new PlainActionFuture<>();
- private final RefCounted refCounted = AbstractRefCounted.of(() -> allClientsClosedListener.onResponse(null));
- private final Set<HttpServerChannel> httpServerChannels = ConcurrentCollections.newConcurrentSet();
- private final long shutdownGracePeriodMillis;
- private final HttpClientStatsTracker httpClientStatsTracker;
- private final HttpTracer httpLogger;
- private final Tracer tracer;
- private volatile boolean shuttingDown;
- private final ReadWriteLock shuttingDownRWLock = new StampedLock().asReadWriteLock();
- private volatile long slowLogThresholdMs;
- protected AbstractHttpServerTransport(
- Settings settings,
- NetworkService networkService,
- Recycler<BytesRef> recycler,
- ThreadPool threadPool,
- NamedXContentRegistry xContentRegistry,
- Dispatcher dispatcher,
- ClusterSettings clusterSettings,
- Tracer tracer
- ) {
- this.settings = settings;
- this.networkService = networkService;
- this.recycler = recycler;
- this.threadPool = threadPool;
- this.parserConfig = XContentParserConfiguration.EMPTY.withRegistry(xContentRegistry)
- .withDeprecationHandler(LoggingDeprecationHandler.INSTANCE);
- this.dispatcher = dispatcher;
- this.handlingSettings = HttpHandlingSettings.fromSettings(settings);
- this.corsHandler = CorsHandler.fromSettings(settings);
- // we can't make the network.bind_host a fallback since we already fall back to http.host hence the extra conditional here
- List<String> httpBindHost = SETTING_HTTP_BIND_HOST.get(settings);
- this.bindHosts = (httpBindHost.isEmpty() ? NetworkService.GLOBAL_NETWORK_BIND_HOST_SETTING.get(settings) : httpBindHost).toArray(
- Strings.EMPTY_ARRAY
- );
- // we can't make the network.publish_host a fallback since we already fall back to http.host hence the extra conditional here
- List<String> httpPublishHost = SETTING_HTTP_PUBLISH_HOST.get(settings);
- this.publishHosts = (httpPublishHost.isEmpty() ? NetworkService.GLOBAL_NETWORK_PUBLISH_HOST_SETTING.get(settings) : httpPublishHost)
- .toArray(Strings.EMPTY_ARRAY);
- this.port = SETTING_HTTP_PORT.get(settings);
- this.maxContentLength = SETTING_HTTP_MAX_CONTENT_LENGTH.get(settings);
- this.tracer = tracer;
- this.httpLogger = new HttpTracer(settings, clusterSettings);
- clusterSettings.addSettingsUpdateConsumer(
- TransportSettings.SLOW_OPERATION_THRESHOLD_SETTING,
- slowLogThreshold -> this.slowLogThresholdMs = slowLogThreshold.getMillis()
- );
- slowLogThresholdMs = TransportSettings.SLOW_OPERATION_THRESHOLD_SETTING.get(settings).getMillis();
- httpClientStatsTracker = new HttpClientStatsTracker(settings, clusterSettings, threadPool);
- shutdownGracePeriodMillis = SETTING_HTTP_SERVER_SHUTDOWN_GRACE_PERIOD.get(settings).getMillis();
- }
- public Recycler<BytesRef> recycler() {
- return recycler;
- }
- @Override
- public BoundTransportAddress boundAddress() {
- return this.boundAddress;
- }
- @Override
- public HttpInfo info() {
- BoundTransportAddress boundTransportAddress = boundAddress();
- if (boundTransportAddress == null) {
- return null;
- }
- return new HttpInfo(boundTransportAddress, maxContentLength.getBytes());
- }
- @Override
- public HttpStats stats() {
- return new HttpStats(
- httpChannels.size(),
- totalChannelsAccepted.get(),
- httpClientStatsTracker.getClientStats(),
- dispatcher.getStats()
- );
- }
- protected void bindServer() {
- // Bind and start to accept incoming connections.
- final InetAddress[] hostAddresses;
- try {
- hostAddresses = networkService.resolveBindHostAddresses(bindHosts);
- } catch (IOException e) {
- throw new BindHttpException("Failed to resolve host [" + Arrays.toString(bindHosts) + "]", e);
- }
- List<TransportAddress> boundAddresses = new ArrayList<>(hostAddresses.length);
- for (InetAddress address : hostAddresses) {
- boundAddresses.add(bindAddress(address));
- }
- final InetAddress publishInetAddress;
- try {
- publishInetAddress = networkService.resolvePublishHostAddresses(publishHosts);
- } catch (Exception e) {
- throw new BindTransportException("Failed to resolve publish address", e);
- }
- final int publishPort = resolvePublishPort(settings, boundAddresses, publishInetAddress);
- TransportAddress publishAddress = new TransportAddress(new InetSocketAddress(publishInetAddress, publishPort));
- this.boundAddress = new BoundTransportAddress(boundAddresses.toArray(new TransportAddress[0]), publishAddress);
- logger.info("{}", boundAddress);
- }
- private TransportAddress bindAddress(final InetAddress hostAddress) {
- final AtomicReference<Exception> lastException = new AtomicReference<>();
- final AtomicReference<InetSocketAddress> boundSocket = new AtomicReference<>();
- boolean success = port.iterate(portNumber -> {
- try {
- synchronized (httpServerChannels) {
- HttpServerChannel httpServerChannel = bind(new InetSocketAddress(hostAddress, portNumber));
- httpServerChannels.add(httpServerChannel);
- boundSocket.set(httpServerChannel.getLocalAddress());
- }
- } catch (Exception e) {
- lastException.set(e);
- return false;
- }
- return true;
- });
- if (success == false) {
- throw new BindHttpException("Failed to bind to " + NetworkAddress.format(hostAddress, port), lastException.get());
- }
- if (logger.isDebugEnabled()) {
- logger.debug("Bound http to address {{}}", NetworkAddress.format(boundSocket.get()));
- }
- return new TransportAddress(boundSocket.get());
- }
- protected abstract HttpServerChannel bind(InetSocketAddress hostAddress) throws Exception;
- /**
- * Gracefully shut down. If {@link HttpTransportSettings#SETTING_HTTP_SERVER_SHUTDOWN_GRACE_PERIOD} is zero, the default, then
- * forcefully close all open connections immediately.
- * Serially run through the following steps:
- * <ol>
- * <li> Stop listening for new HTTP connections, which means no new HttpChannel are added to the {@link #httpChannels} list.
- * {@link #serverAcceptedChannel(HttpChannel)} will close any new channels to ensure this is true.
- * <li> Close the HttpChannel after a new request completes on all existing channels.
- * <li> Close all idle channels.
- * <li> If grace period is set, wait for all httpChannels to close via 2 for up to the configured grace period,
- * {@link #shutdownGracePeriodMillis}.
- * If all connections are closed before the expiration of the grace period, stop waiting early.
- * <li> Close all remaining open httpChannels even if requests are in flight.
- * </ol>
- */
- @Override
- protected void doStop() {
- synchronized (httpServerChannels) {
- if (httpServerChannels.isEmpty() == false) {
- try {
- CloseableChannel.closeChannels(new ArrayList<>(httpServerChannels), true);
- } catch (Exception e) {
- logger.warn("exception while closing channels", e);
- } finally {
- httpServerChannels.clear();
- }
- }
- }
- var wlock = shuttingDownRWLock.writeLock();
- try {
- wlock.lock();
- shuttingDown = true;
- refCounted.decRef();
- httpChannels.values().forEach(RequestTrackingHttpChannel::setCloseWhenIdle);
- } finally {
- wlock.unlock();
- }
- boolean closed = false;
- if (shutdownGracePeriodMillis > 0) {
- try {
- logger.debug(format("waiting [%d]ms for clients to close connections", shutdownGracePeriodMillis));
- FutureUtils.get(allClientsClosedListener, shutdownGracePeriodMillis, TimeUnit.MILLISECONDS);
- closed = true;
- } catch (ElasticsearchTimeoutException t) {
- logger.warn(format("timed out while waiting [%d]ms for clients to close connections", shutdownGracePeriodMillis));
- }
- } else {
- logger.debug("closing all client connections immediately");
- }
- if (closed == false) {
- try {
- CloseableChannel.closeChannels(new ArrayList<>(httpChannels.values()), true);
- } catch (Exception e) {
- logger.warn("unexpected exception while closing http channels", e);
- }
- try {
- allClientsClosedListener.get();
- } catch (Exception e) {
- assert false : e;
- logger.warn("unexpected exception while waiting for http channels to close", e);
- }
- }
- stopInternal();
- }
- boolean isAcceptingConnections() {
- return shuttingDown == false;
- }
- @Override
- protected void doClose() {}
- /**
- * Called to tear down internal resources
- */
- protected abstract void stopInternal();
- // package private for tests
- static int resolvePublishPort(Settings settings, List<TransportAddress> boundAddresses, InetAddress publishInetAddress) {
- int publishPort = SETTING_HTTP_PUBLISH_PORT.get(settings);
- if (publishPort < 0) {
- for (TransportAddress boundAddress : boundAddresses) {
- InetAddress boundInetAddress = boundAddress.address().getAddress();
- if (boundInetAddress.isAnyLocalAddress() || boundInetAddress.equals(publishInetAddress)) {
- publishPort = boundAddress.getPort();
- break;
- }
- }
- }
- // if no matching boundAddress found, check if there is a unique port for all bound addresses
- if (publishPort < 0) {
- final Set<Integer> ports = new HashSet<>();
- for (TransportAddress boundAddress : boundAddresses) {
- ports.add(boundAddress.getPort());
- }
- if (ports.size() == 1) {
- publishPort = ports.iterator().next();
- }
- }
- if (publishPort < 0) {
- throw new BindHttpException(
- "Failed to auto-resolve http publish port, multiple bound addresses "
- + boundAddresses
- + " with distinct ports and none of them matched the publish address ("
- + publishInetAddress
- + "). "
- + "Please specify a unique port by setting "
- + SETTING_HTTP_PORT.getKey()
- + " or "
- + SETTING_HTTP_PUBLISH_PORT.getKey()
- );
- }
- return publishPort;
- }
- public void onException(HttpChannel channel, Exception e) {
- try {
- if (lifecycle.started() == false) {
- // just close and ignore - we are already stopped and just need to make sure we release all resources
- return;
- }
- if (NetworkExceptionHelper.getCloseConnectionExceptionLevel(e, false) != Level.OFF) {
- logger.trace(
- () -> format("close connection exception caught while handling client http traffic, closing connection %s", channel),
- e
- );
- } else if (NetworkExceptionHelper.isConnectException(e)) {
- logger.trace(
- () -> format("connect exception caught while handling client http traffic, closing connection %s", channel),
- e
- );
- } else if (e instanceof HttpReadTimeoutException) {
- logger.trace(() -> format("http read timeout, closing connection %s", channel), e);
- } else if (e instanceof CancelledKeyException) {
- logger.trace(
- () -> format("cancelled key exception caught while handling client http traffic, closing connection %s", channel),
- e
- );
- } else {
- logger.warn(() -> format("caught exception while handling client http traffic, closing connection %s", channel), e);
- }
- } finally {
- CloseableChannel.closeChannel(channel);
- }
- }
- protected static void onServerException(HttpServerChannel channel, Exception e) {
- logger.error(() -> "exception from http server channel caught on transport layer [channel=" + channel + "]", e);
- }
- protected void serverAcceptedChannel(HttpChannel httpChannel) {
- var rlock = shuttingDownRWLock.readLock();
- try {
- rlock.lock();
- if (shuttingDown) {
- throw new IllegalStateException("Server cannot accept new channel while shutting down");
- }
- RequestTrackingHttpChannel trackingChannel = httpChannels.putIfAbsent(httpChannel, new RequestTrackingHttpChannel(httpChannel));
- assert trackingChannel == null : "Channel should only be added to http channel set once";
- } finally {
- rlock.unlock();
- }
- refCounted.incRef();
- httpChannel.addCloseListener(ActionListener.running(() -> {
- httpChannels.remove(httpChannel);
- refCounted.decRef();
- }));
- totalChannelsAccepted.incrementAndGet();
- httpClientStatsTracker.addClientStats(httpChannel);
- logger.trace(() -> format("Http channel accepted: %s", httpChannel));
- }
- /**
- * This method handles an incoming http request.
- *
- * @param httpRequest that is incoming
- * @param httpChannel that received the http request
- */
- public void incomingRequest(final HttpRequest httpRequest, final HttpChannel httpChannel) {
- httpClientStatsTracker.updateClientStats(httpRequest, httpChannel);
- final RequestTrackingHttpChannel trackingChannel = httpChannels.get(httpChannel);
- final long startTime = threadPool.rawRelativeTimeInMillis();
- try {
- // The channel may not be present if the close listener (set in serverAcceptedChannel) runs before this method because the
- // connection closed early
- if (trackingChannel == null) {
- httpRequest.release();
- logger.warn(
- "http channel [{}] closed before starting to handle [{}][{}][{}]",
- httpChannel,
- httpRequest.header(Task.X_OPAQUE_ID_HTTP_HEADER),
- httpRequest.method(),
- httpRequest.uri()
- );
- return;
- }
- trackingChannel.incomingRequest();
- handleIncomingRequest(httpRequest, trackingChannel, httpRequest.getInboundException());
- } finally {
- final long took = threadPool.rawRelativeTimeInMillis() - startTime;
- networkService.getHandlingTimeTracker().addHandlingTime(took);
- final long logThreshold = slowLogThresholdMs;
- if (logThreshold > 0 && took > logThreshold) {
- logger.warn(
- "handling request [{}][{}][{}][{}] took [{}ms] which is above the warn threshold of [{}ms]",
- httpRequest.header(Task.X_OPAQUE_ID_HTTP_HEADER),
- httpRequest.method(),
- httpRequest.uri(),
- httpChannel,
- took,
- logThreshold
- );
- }
- }
- }
- // Visible for testing
- void dispatchRequest(final RestRequest restRequest, final RestChannel channel, final Throwable badRequestCause) {
- final ThreadContext threadContext = threadPool.getThreadContext();
- try (ThreadContext.StoredContext ignore = threadContext.stashContext()) {
- if (badRequestCause != null) {
- dispatcher.dispatchBadRequest(channel, threadContext, badRequestCause);
- } else {
- populatePerRequestThreadContext0(restRequest, channel, threadContext);
- dispatcher.dispatchRequest(restRequest, channel, threadContext);
- }
- }
- }
- private void populatePerRequestThreadContext0(RestRequest restRequest, RestChannel channel, ThreadContext threadContext) {
- try {
- populatePerRequestThreadContext(restRequest, threadContext);
- } catch (Exception e) {
- try {
- channel.sendResponse(new RestResponse(channel, e));
- } catch (Exception inner) {
- inner.addSuppressed(e);
- logger.error(() -> "failed to send failure response for uri [" + restRequest.uri() + "]", inner);
- }
- }
- }
- protected void populatePerRequestThreadContext(RestRequest restRequest, ThreadContext threadContext) {}
- private void handleIncomingRequest(final HttpRequest httpRequest, final HttpChannel httpChannel, final Exception exception) {
- if (exception == null) {
- HttpResponse earlyResponse = corsHandler.handleInbound(httpRequest);
- if (earlyResponse != null) {
- httpChannel.sendResponse(earlyResponse, earlyResponseListener(httpRequest, httpChannel));
- httpRequest.release();
- return;
- }
- }
- Exception badRequestCause = exception;
- /*
- * We want to create a REST request from the incoming request from Netty. However, creating this request could fail if there
- * are incorrectly encoded parameters, or the Content-Type header is invalid. If one of these specific failures occurs, we
- * attempt to create a REST request again without the input that caused the exception (e.g., we remove the Content-Type header,
- * or skip decoding the parameters). Once we have a request in hand, we then dispatch the request as a bad request with the
- * underlying exception that caused us to treat the request as bad.
- */
- final RestRequest restRequest;
- {
- RestRequest innerRestRequest;
- try {
- innerRestRequest = RestRequest.request(parserConfig, httpRequest, httpChannel);
- } catch (final RestRequest.MediaTypeHeaderException e) {
- badRequestCause = ExceptionsHelper.useOrSuppress(badRequestCause, e);
- innerRestRequest = requestWithoutFailedHeader(httpRequest, httpChannel, badRequestCause, e.getFailedHeaderNames());
- } catch (final RestRequest.BadParameterException e) {
- badRequestCause = ExceptionsHelper.useOrSuppress(badRequestCause, e);
- innerRestRequest = RestRequest.requestWithoutParameters(parserConfig, httpRequest, httpChannel);
- }
- restRequest = innerRestRequest;
- }
- final HttpTracer maybeHttpLogger = httpLogger.maybeLogRequest(restRequest, exception);
- /*
- * We now want to create a channel used to send the response on. However, creating this channel can fail if there are invalid
- * parameter values for any of the filter_path, human, or pretty parameters. We detect these specific failures via an
- * IllegalArgumentException from the channel constructor and then attempt to create a new channel that bypasses parsing of these
- * parameter values.
- */
- final RestChannel channel;
- {
- RestChannel innerChannel;
- ThreadContext threadContext = threadPool.getThreadContext();
- try {
- innerChannel = new DefaultRestChannel(
- httpChannel,
- httpRequest,
- restRequest,
- recycler,
- handlingSettings,
- threadContext,
- corsHandler,
- maybeHttpLogger,
- tracer
- );
- } catch (final IllegalArgumentException e) {
- badRequestCause = ExceptionsHelper.useOrSuppress(badRequestCause, e);
- final RestRequest innerRequest = RestRequest.requestWithoutParameters(parserConfig, httpRequest, httpChannel);
- innerChannel = new DefaultRestChannel(
- httpChannel,
- httpRequest,
- innerRequest,
- recycler,
- handlingSettings,
- threadContext,
- corsHandler,
- httpLogger,
- tracer
- );
- }
- channel = innerChannel;
- }
- dispatchRequest(restRequest, channel, badRequestCause);
- }
- private RestRequest requestWithoutFailedHeader(
- HttpRequest httpRequest,
- HttpChannel httpChannel,
- Exception badRequestCause,
- Set<String> failedHeaderNames
- ) {
- assert failedHeaderNames.size() > 0;
- HttpRequest httpRequestWithoutHeader = httpRequest;
- for (String failedHeaderName : failedHeaderNames) {
- httpRequestWithoutHeader = httpRequestWithoutHeader.removeHeader(failedHeaderName);
- }
- try {
- return RestRequest.request(parserConfig, httpRequestWithoutHeader, httpChannel);
- } catch (final RestRequest.MediaTypeHeaderException e) {
- badRequestCause = ExceptionsHelper.useOrSuppress(badRequestCause, e);
- return requestWithoutFailedHeader(httpRequestWithoutHeader, httpChannel, badRequestCause, e.getFailedHeaderNames());
- } catch (final RestRequest.BadParameterException e) {
- badRequestCause.addSuppressed(e);
- return RestRequest.requestWithoutParameters(parserConfig, httpRequestWithoutHeader, httpChannel);
- }
- }
- private static ActionListener<Void> earlyResponseListener(HttpRequest request, HttpChannel httpChannel) {
- if (HttpUtils.shouldCloseConnection(request)) {
- return ActionListener.running(() -> CloseableChannel.closeChannel(httpChannel));
- } else {
- return ActionListener.noop();
- }
- }
- public ThreadPool getThreadPool() {
- return threadPool;
- }
- /**
- * A {@link HttpChannel} that tracks number of requests via a {@link RefCounted}.
- */
- private static class RequestTrackingHttpChannel implements HttpChannel {
- /**
- * Only counts down to zero via {@link #setCloseWhenIdle()}.
- */
- final RefCounted refCounted = AbstractRefCounted.of(this::closeInner);
- final HttpChannel inner;
- RequestTrackingHttpChannel(HttpChannel inner) {
- this.inner = inner;
- }
- public void incomingRequest() throws IllegalStateException {
- refCounted.incRef();
- }
- /**
- * Close the channel when there are no more requests in flight.
- */
- public void setCloseWhenIdle() {
- refCounted.decRef();
- }
- @Override
- public void close() {
- closeInner();
- }
- /**
- * Synchronized to avoid double close due to a natural close and a close via {@link #setCloseWhenIdle()}
- */
- private void closeInner() {
- synchronized (inner) {
- if (inner.isOpen()) {
- inner.close();
- } else {
- logger.info("channel [{}] already closed", inner);
- }
- }
- }
- @Override
- public void addCloseListener(ActionListener<Void> listener) {
- inner.addCloseListener(listener);
- }
- @Override
- public boolean isOpen() {
- return inner.isOpen();
- }
- @Override
- public void sendResponse(HttpResponse response, ActionListener<Void> listener) {
- inner.sendResponse(
- response,
- listener != null ? ActionListener.runAfter(listener, refCounted::decRef) : ActionListener.running(refCounted::decRef)
- );
- }
- @Override
- public InetSocketAddress getLocalAddress() {
- return inner.getLocalAddress();
- }
- @Override
- public InetSocketAddress getRemoteAddress() {
- return inner.getRemoteAddress();
- }
- @Override
- public String toString() {
- return inner.toString();
- }
- }
- }
|