io.edurt.datacap.pinot.com.ning.http.client.multipart.PartBase Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of datacap-shaded-pinot Show documentation
Show all versions of datacap-shaded-pinot Show documentation
DataCap - Shaded for Apache Pinot
/*
* Copyright (c) 2014 AsyncHttpClient Project. All rights reserved.
*
* This program is licensed to you under the Apache License Version 2.0,
* and you may not use this file except in compliance with the Apache License Version 2.0.
* You may obtain a copy of the Apache License Version 2.0 at http://www.apache.org/licenses/LICENSE-2.0.
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the Apache License Version 2.0 is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the Apache License Version 2.0 for the specific language governing permissions and limitations there under.
*/
package io.edurt.datacap.pinot.com.ning.http.client.multipart;
import static java.nio.charset.StandardCharsets.*;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.Charset;
public abstract class PartBase implements Part {
/**
* The name of the form field, part of the Content-Disposition header
*/
private final String name;
/**
* The main part of the Content-Type header
*/
private final String contentType;
/**
* The charset (part of Content-Type header)
*/
private final Charset charset;
/**
* The Content-Transfer-Encoding header value.
*/
private final String transferEncoding;
/**
* The Content-Id
*/
private final String contentId;
/**
* The disposition type (part of Content-Disposition)
*/
private String dispositionType;
public PartBase(String name, String contentType, Charset charset, String contentId) {
this(name, contentType, charset, contentId, null);
}
/**
* Constructor.
*
* @param name The name of the part, or null
* @param contentType The content type, or null
* @param charset The character encoding, or null
* @param contentId The content id, or null
* @param transferEncoding The transfer encoding, or null
*/
public PartBase(String name, String contentType, Charset charset, String contentId, String transferEncoding) {
this.name = name;
this.contentType = contentType;
this.charset = charset;
this.contentId = contentId;
this.transferEncoding = transferEncoding;
}
protected void visitStart(PartVisitor visitor, byte[] boundary) throws IOException {
visitor.withBytes(EXTRA_BYTES);
visitor.withBytes(boundary);
}
protected void visitDispositionHeader(PartVisitor visitor) throws IOException {
visitor.withBytes(CRLF_BYTES);
visitor.withBytes(CONTENT_DISPOSITION_BYTES);
visitor.withBytes(getDispositionType() != null ? getDispositionType().getBytes(US_ASCII) : FORM_DATA_DISPOSITION_TYPE_BYTES);
if (getName() != null) {
visitor.withBytes(NAME_BYTES);
visitor.withByte(QUOTE_BYTE);
visitor.withBytes(getName().getBytes(US_ASCII));
visitor.withByte(QUOTE_BYTE);
}
}
protected void visitContentTypeHeader(PartVisitor visitor) throws IOException {
String contentType = getContentType();
if (contentType != null) {
visitor.withBytes(CRLF_BYTES);
visitor.withBytes(CONTENT_TYPE_BYTES);
visitor.withBytes(contentType.getBytes(US_ASCII));
Charset charset = getCharset();
if (charset != null) {
visitor.withBytes(CHARSET_BYTES);
visitor.withBytes(charset.name().getBytes(US_ASCII));
}
}
}
protected void visitTransferEncodingHeader(PartVisitor visitor) throws IOException {
String transferEncoding = getTransferEncoding();
if (transferEncoding != null) {
visitor.withBytes(CRLF_BYTES);
visitor.withBytes(CONTENT_TRANSFER_ENCODING_BYTES);
visitor.withBytes(transferEncoding.getBytes(US_ASCII));
}
}
protected void visitContentIdHeader(PartVisitor visitor) throws IOException {
String contentId = getContentId();
if (contentId != null) {
visitor.withBytes(CRLF_BYTES);
visitor.withBytes(CONTENT_ID_BYTES);
visitor.withBytes(contentId.getBytes(US_ASCII));
}
}
protected void visitEndOfHeader(PartVisitor visitor) throws IOException {
visitor.withBytes(CRLF_BYTES);
visitor.withBytes(CRLF_BYTES);
}
protected void visitEnd(PartVisitor visitor) throws IOException {
visitor.withBytes(CRLF_BYTES);
}
protected abstract long getDataLength();
protected abstract void sendData(OutputStream out) throws IOException;
/**
* Write all the data to the output stream. If you override this method make sure to override #length() as well
*
* @param out
* The output stream
* @param boundary
* the boundary
* @throws IOException
* If an IO problem occurs.
*/
public void write(OutputStream out, byte[] boundary) throws IOException {
OutputStreamPartVisitor visitor = new OutputStreamPartVisitor(out);
visitStart(visitor, boundary);
visitDispositionHeader(visitor);
visitContentTypeHeader(visitor);
visitTransferEncodingHeader(visitor);
visitContentIdHeader(visitor);
visitEndOfHeader(visitor);
sendData(visitor.getOutputStream());
visitEnd(visitor);
}
/**
* Return the full length of all the data. If you override this method make sure to override #send(OutputStream) as well
*
* @return long The length.
*/
public long length(byte[] boundary) {
long dataLength = getDataLength();
try {
if (dataLength < 0L) {
return -1L;
} else {
CounterPartVisitor visitor = new CounterPartVisitor();
visitStart(visitor, boundary);
visitDispositionHeader(visitor);
visitContentTypeHeader(visitor);
visitTransferEncodingHeader(visitor);
visitContentIdHeader(visitor);
visitEndOfHeader(visitor);
visitEnd(visitor);
return dataLength + visitor.getCount();
}
} catch (IOException e) {
// can't happen
throw new RuntimeException("IOException while computing length, WTF", e);
}
}
public String toString() {
return new StringBuilder()//
.append(getClass().getSimpleName())//
.append(" name=").append(getName())//
.append(" contentType=").append(getContentType())//
.append(" charset=").append(getCharset())//
.append(" tranferEncoding=").append(getTransferEncoding())//
.append(" contentId=").append(getContentId())//
.append(" dispositionType=").append(getDispositionType())//
.toString();
}
@Override
public String getName() {
return this.name;
}
@Override
public String getContentType() {
return this.contentType;
}
@Override
public Charset getCharset() {
return this.charset;
}
@Override
public String getTransferEncoding() {
return transferEncoding;
}
@Override
public String getContentId() {
return contentId;
}
@Override
public String getDispositionType() {
return dispositionType;
}
public void setDispositionType(String dispositionType) {
this.dispositionType = dispositionType;
}
}