forked from D-Net/dnet-hadoop
54 lines
1.1 KiB
Java
54 lines
1.1 KiB
Java
|
|
package eu.dnetlib.dhp.common.api;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
|
|
import okhttp3.MediaType;
|
|
import okhttp3.RequestBody;
|
|
import okhttp3.internal.Util;
|
|
import okio.BufferedSink;
|
|
import okio.Okio;
|
|
import okio.Source;
|
|
|
|
public class InputStreamRequestBody extends RequestBody {
|
|
|
|
private final InputStream inputStream;
|
|
private final MediaType mediaType;
|
|
private final long lenght;
|
|
|
|
public static RequestBody create(final MediaType mediaType, final InputStream inputStream, final long len) {
|
|
|
|
return new InputStreamRequestBody(inputStream, mediaType, len);
|
|
}
|
|
|
|
private InputStreamRequestBody(InputStream inputStream, MediaType mediaType, long len) {
|
|
this.inputStream = inputStream;
|
|
this.mediaType = mediaType;
|
|
this.lenght = len;
|
|
}
|
|
|
|
@Override
|
|
public MediaType contentType() {
|
|
return mediaType;
|
|
}
|
|
|
|
@Override
|
|
public long contentLength() {
|
|
|
|
return lenght;
|
|
|
|
}
|
|
|
|
@Override
|
|
public void writeTo(BufferedSink sink) throws IOException {
|
|
Source source = null;
|
|
try {
|
|
source = Okio.source(inputStream);
|
|
sink.writeAll(source);
|
|
} finally {
|
|
Util.closeQuietly(source);
|
|
}
|
|
}
|
|
}
|