com.facebook.presto.jdbc.internal.apache.http.impl.AbstractHttpServerConnection Maven / Gradle / Ivy
/*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package com.facebook.presto.jdbc.internal.apache.http.impl;
import java.io.IOException;
import com.facebook.presto.jdbc.internal.apache.http.HttpConnectionMetrics;
import com.facebook.presto.jdbc.internal.apache.http.HttpEntity;
import com.facebook.presto.jdbc.internal.apache.http.HttpEntityEnclosingRequest;
import com.facebook.presto.jdbc.internal.apache.http.HttpException;
import com.facebook.presto.jdbc.internal.apache.http.HttpRequest;
import com.facebook.presto.jdbc.internal.apache.http.HttpRequestFactory;
import com.facebook.presto.jdbc.internal.apache.http.HttpResponse;
import com.facebook.presto.jdbc.internal.apache.http.HttpServerConnection;
import com.facebook.presto.jdbc.internal.apache.http.impl.entity.DisallowIdentityContentLengthStrategy;
import com.facebook.presto.jdbc.internal.apache.http.impl.entity.EntityDeserializer;
import com.facebook.presto.jdbc.internal.apache.http.impl.entity.EntitySerializer;
import com.facebook.presto.jdbc.internal.apache.http.impl.entity.LaxContentLengthStrategy;
import com.facebook.presto.jdbc.internal.apache.http.impl.entity.StrictContentLengthStrategy;
import com.facebook.presto.jdbc.internal.apache.http.impl.io.DefaultHttpRequestParser;
import com.facebook.presto.jdbc.internal.apache.http.impl.io.HttpResponseWriter;
import com.facebook.presto.jdbc.internal.apache.http.io.EofSensor;
import com.facebook.presto.jdbc.internal.apache.http.io.HttpMessageParser;
import com.facebook.presto.jdbc.internal.apache.http.io.HttpMessageWriter;
import com.facebook.presto.jdbc.internal.apache.http.io.HttpTransportMetrics;
import com.facebook.presto.jdbc.internal.apache.http.io.SessionInputBuffer;
import com.facebook.presto.jdbc.internal.apache.http.io.SessionOutputBuffer;
import com.facebook.presto.jdbc.internal.apache.http.params.HttpParams;
import com.facebook.presto.jdbc.internal.apache.http.util.Args;
/**
* Abstract server-side HTTP connection capable of transmitting and receiving
* data using arbitrary {@link SessionInputBuffer} and
* {@link SessionOutputBuffer} implementations.
*
* The following parameters can be used to customize the behavior of this
* class:
*
* - {@link com.facebook.presto.jdbc.internal.apache.http.params.CoreProtocolPNames#STRICT_TRANSFER_ENCODING}
* - {@link com.facebook.presto.jdbc.internal.apache.http.params.CoreConnectionPNames#MAX_HEADER_COUNT}
* - {@link com.facebook.presto.jdbc.internal.apache.http.params.CoreConnectionPNames#MAX_LINE_LENGTH}
*
*
* @since 4.0
*
* @deprecated (4.3) use {@link DefaultBHttpServerConnection}
*/
@Deprecated
public abstract class AbstractHttpServerConnection implements HttpServerConnection {
private final EntitySerializer entityserializer;
private final EntityDeserializer entitydeserializer;
private SessionInputBuffer inbuffer = null;
private SessionOutputBuffer outbuffer = null;
private EofSensor eofSensor = null;
private HttpMessageParser requestParser = null;
private HttpMessageWriter responseWriter = null;
private HttpConnectionMetricsImpl metrics = null;
/**
* Creates an instance of this class.
*
* This constructor will invoke {@link #createEntityDeserializer()}
* and {@link #createEntitySerializer()} methods in order to initialize
* HTTP entity serializer and deserializer implementations for this
* connection.
*/
public AbstractHttpServerConnection() {
super();
this.entityserializer = createEntitySerializer();
this.entitydeserializer = createEntityDeserializer();
}
/**
* Asserts if the connection is open.
*
* @throws IllegalStateException if the connection is not open.
*/
protected abstract void assertOpen() throws IllegalStateException;
/**
* Creates an instance of {@link EntityDeserializer} with the
* {@link LaxContentLengthStrategy} implementation to be used for
* de-serializing entities received over this connection.
*
* This method can be overridden in a super class in order to create
* instances of {@link EntityDeserializer} using a custom
* {@link com.facebook.presto.jdbc.internal.apache.http.entity.ContentLengthStrategy}.
*
* @return HTTP entity deserializer
*/
protected EntityDeserializer createEntityDeserializer() {
return new EntityDeserializer(new DisallowIdentityContentLengthStrategy(
new LaxContentLengthStrategy(0)));
}
/**
* Creates an instance of {@link EntitySerializer} with the
* {@link StrictContentLengthStrategy} implementation to be used for
* serializing HTTP entities sent over this connection.
*
* This method can be overridden in a super class in order to create
* instances of {@link EntitySerializer} using a custom
* {@link com.facebook.presto.jdbc.internal.apache.http.entity.ContentLengthStrategy}.
*
* @return HTTP entity serialzier.
*/
protected EntitySerializer createEntitySerializer() {
return new EntitySerializer(new StrictContentLengthStrategy());
}
/**
* Creates an instance of {@link DefaultHttpRequestFactory} to be used
* for creating {@link HttpRequest} objects received by over this
* connection.
*
* This method can be overridden in a super class in order to provide
* a different implementation of the {@link HttpRequestFactory} interface.
*
* @return HTTP request factory.
*/
protected HttpRequestFactory createHttpRequestFactory() {
return DefaultHttpRequestFactory.INSTANCE;
}
/**
* Creates an instance of {@link HttpMessageParser} to be used for parsing
* HTTP requests received over this connection.
*
* This method can be overridden in a super class in order to provide
* a different implementation of the {@link HttpMessageParser} interface or
* to pass a different implementation of the
* {@link com.facebook.presto.jdbc.internal.apache.http.message.LineParser} to the
* {@link DefaultHttpRequestParser} constructor.
*
* @param buffer the session input buffer.
* @param requestFactory the HTTP request factory.
* @param params HTTP parameters.
* @return HTTP message parser.
*/
protected HttpMessageParser createRequestParser(
final SessionInputBuffer buffer,
final HttpRequestFactory requestFactory,
final HttpParams params) {
return new DefaultHttpRequestParser(buffer, null, requestFactory, params);
}
/**
* Creates an instance of {@link HttpMessageWriter} to be used for
* writing out HTTP responses sent over this connection.
*
* This method can be overridden in a super class in order to provide
* a different implementation of the {@link HttpMessageWriter} interface or
* to pass a different implementation of
* {@link com.facebook.presto.jdbc.internal.apache.http.message.LineFormatter} to the the default
* implementation {@link HttpResponseWriter}.
*
* @param buffer the session output buffer
* @param params HTTP parameters
* @return HTTP message writer
*/
protected HttpMessageWriter createResponseWriter(
final SessionOutputBuffer buffer,
final HttpParams params) {
return new HttpResponseWriter(buffer, null, params);
}
/**
* @since 4.1
*/
protected HttpConnectionMetricsImpl createConnectionMetrics(
final HttpTransportMetrics inTransportMetric,
final HttpTransportMetrics outTransportMetric) {
return new HttpConnectionMetricsImpl(inTransportMetric, outTransportMetric);
}
/**
* Initializes this connection object with {@link SessionInputBuffer} and
* {@link SessionOutputBuffer} instances to be used for sending and
* receiving data. These session buffers can be bound to any arbitrary
* physical output medium.
*
* This method will invoke {@link #createHttpRequestFactory},
* {@link #createRequestParser(SessionInputBuffer, HttpRequestFactory, HttpParams)}
* and {@link #createResponseWriter(SessionOutputBuffer, HttpParams)}
* methods to initialize HTTP request parser and response writer for this
* connection.
*
* @param inbuffer the session input buffer.
* @param outbuffer the session output buffer.
* @param params HTTP parameters.
*/
protected void init(
final SessionInputBuffer inbuffer,
final SessionOutputBuffer outbuffer,
final HttpParams params) {
this.inbuffer = Args.notNull(inbuffer, "Input session buffer");
this.outbuffer = Args.notNull(outbuffer, "Output session buffer");
if (inbuffer instanceof EofSensor) {
this.eofSensor = (EofSensor) inbuffer;
}
this.requestParser = createRequestParser(
inbuffer,
createHttpRequestFactory(),
params);
this.responseWriter = createResponseWriter(
outbuffer, params);
this.metrics = createConnectionMetrics(
inbuffer.getMetrics(),
outbuffer.getMetrics());
}
@Override
public HttpRequest receiveRequestHeader()
throws HttpException, IOException {
assertOpen();
final HttpRequest request = this.requestParser.parse();
this.metrics.incrementRequestCount();
return request;
}
@Override
public void receiveRequestEntity(final HttpEntityEnclosingRequest request)
throws HttpException, IOException {
Args.notNull(request, "HTTP request");
assertOpen();
final HttpEntity entity = this.entitydeserializer.deserialize(this.inbuffer, request);
request.setEntity(entity);
}
protected void doFlush() throws IOException {
this.outbuffer.flush();
}
@Override
public void flush() throws IOException {
assertOpen();
doFlush();
}
@Override
public void sendResponseHeader(final HttpResponse response)
throws HttpException, IOException {
Args.notNull(response, "HTTP response");
assertOpen();
this.responseWriter.write(response);
if (response.getStatusLine().getStatusCode() >= 200) {
this.metrics.incrementResponseCount();
}
}
@Override
public void sendResponseEntity(final HttpResponse response)
throws HttpException, IOException {
if (response.getEntity() == null) {
return;
}
this.entityserializer.serialize(
this.outbuffer,
response,
response.getEntity());
}
protected boolean isEof() {
return this.eofSensor != null && this.eofSensor.isEof();
}
@Override
public boolean isStale() {
if (!isOpen()) {
return true;
}
if (isEof()) {
return true;
}
try {
this.inbuffer.isDataAvailable(1);
return isEof();
} catch (final IOException ex) {
return true;
}
}
@Override
public HttpConnectionMetrics getMetrics() {
return this.metrics;
}
}