org.xnio.conduits.BufferedStreamSourceConduit Maven / Gradle / Ivy
Go to download
This artifact provides a single jar that contains all classes required to use remote EJB and JMS, including
all dependencies. It is intended for use by those not using maven, maven users should just import the EJB and
JMS BOM's instead (shaded JAR's cause lots of problems with maven, as it is very easy to inadvertently end up
with different versions on classes on the class path).
/*
* JBoss, Home of Professional Open Source
*
* Copyright 2013 Red Hat, Inc. and/or its affiliates.
*
* Licensed 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.
*/
package org.xnio.conduits;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;
import java.util.Arrays;
import org.xnio.Buffers;
import org.xnio.Pooled;
import org.xnio.channels.StreamSinkChannel;
/**
* A stream source conduit which buffers input.
*
* @author David M. Lloyd
*/
public final class BufferedStreamSourceConduit extends AbstractStreamSourceConduit {
private final Pooled pooledBuffer;
/**
* Construct a new instance.
*
* @param next the delegate conduit to set
* @param pooledBuffer the buffer to use
*/
public BufferedStreamSourceConduit(final StreamSourceConduit next, final Pooled pooledBuffer) {
super(next);
this.pooledBuffer = pooledBuffer;
}
public long transferTo(final long position, final long count, final FileChannel target) throws IOException {
try {
final ByteBuffer buffer = pooledBuffer.getResource();
final int lim = buffer.limit();
final int pos = buffer.position();
final int rem = lim - pos;
if (rem > 0) {
if ((long)rem > count) {
buffer.limit(pos + (int)count);
try {
return target.write(buffer, position);
} finally {
buffer.limit(lim);
}
} else {
return target.write(buffer, position);
}
} else {
return super.transferTo(position, count, target);
}
} catch (IllegalStateException ignored) {
return 0L;
}
}
public long transferTo(final long count, final ByteBuffer throughBuffer, final StreamSinkChannel target) throws IOException {
try {
final ByteBuffer buffer = pooledBuffer.getResource();
int lim;
int pos;
int rem;
lim = buffer.limit();
pos = buffer.position();
rem = lim - pos;
throughBuffer.clear();
int res;
long t = 0L;
while (rem > 0) {
if ((long)rem > count) {
buffer.limit(pos + (int)count);
try {
t += res = target.write(buffer);
} finally {
buffer.limit(lim);
}
} else {
t += res = target.write(buffer);
}
if (res == 0) {
return t;
}
pos = buffer.position();
rem = lim - pos;
}
final long lres = Conduits.transfer(next, count, throughBuffer, target);
if (lres > 0) t += lres;
return t == 0L && lres == -1L ? -1L : t;
} catch (IllegalStateException ignored) {
return -1L;
}
}
public int read(final ByteBuffer dst) throws IOException {
try {
final ByteBuffer buffer = pooledBuffer.getResource();
final int lim = buffer.limit();
final int pos = buffer.position();
final int rem = lim - pos;
if (rem > 0) {
return Buffers.copy(dst, buffer);
} else {
final int dstRem = dst.remaining();
buffer.clear();
try {
final long rres = next.read(new ByteBuffer[] { dst, buffer }, 0, 2);
if (rres == -1L) {
return -1;
}
} finally {
buffer.flip();
}
return dst.remaining() - dstRem;
}
} catch (IllegalStateException ignored) {
return -1;
}
}
public long read(final ByteBuffer[] dsts, final int offs, final int len) throws IOException {
if (len == 0) {
return 0L;
} else if (len == 1) {
return read(dsts[offs]);
} else try {
final ByteBuffer buffer = pooledBuffer.getResource();
final int lim = buffer.limit();
final int pos = buffer.position();
final int rem = lim - pos;
if (rem > 0) {
return Buffers.copy(dsts, offs, len, buffer);
} else {
final long dstRem = Buffers.remaining(dsts, offs, len);
buffer.clear();
try {
final ByteBuffer[] buffers = Arrays.copyOfRange(dsts, offs, offs + len + 1);
buffers[buffers.length - 1] = buffer;
final long rres = next.read(buffers, 0, buffers.length);
if (rres == -1) {
return -1L;
}
} finally {
buffer.flip();
}
return Buffers.remaining(dsts, offs, len) - dstRem;
}
} catch (IllegalStateException ignored) {
return -1;
}
}
public void terminateReads() throws IOException {
pooledBuffer.free();
super.terminateReads();
}
}