org.xnio.conduits.InflatingStreamSourceConduit 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 static org.xnio._private.Messages.msg;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;
import java.util.concurrent.TimeUnit;
import java.util.zip.DataFormatException;
import java.util.zip.Inflater;
import org.xnio.channels.StreamSinkChannel;
/**
* A filtering stream source conduit which decompresses the source data.
*
* @author David M. Lloyd
*/
public final class InflatingStreamSourceConduit extends AbstractStreamSourceConduit implements StreamSourceConduit {
private final Inflater inflater;
private final ByteBuffer buffer;
/**
* Construct a new instance.
*
* @param next the underlying conduit for this channel
* @param inflater the initialized inflater to use
*/
public InflatingStreamSourceConduit(final StreamSourceConduit next, final Inflater inflater) {
super(next);
this.inflater = inflater;
buffer = ByteBuffer.allocate(16384);
}
public long transferTo(final long position, final long count, final FileChannel target) throws IOException {
return target.transferFrom(new ConduitReadableByteChannel(this), position, count);
}
public long transferTo(final long count, final ByteBuffer throughBuffer, final StreamSinkChannel target) throws IOException {
return Conduits.transfer(this, count, throughBuffer, target);
}
public int read(final ByteBuffer dst) throws IOException {
final int remaining = dst.remaining();
final int position = dst.position();
final Inflater inflater = this.inflater;
int res;
if (dst.hasArray()) {
// fast path
final byte[] array = dst.array();
final int off = dst.arrayOffset();
for (;;) {
try {
res = inflater.inflate(array, off + position, remaining);
} catch (DataFormatException e) {
throw new IOException(e);
}
if (res > 0) {
dst.position(position + res);
return res;
}
if (inflater.needsDictionary()) {
throw msg.inflaterNeedsDictionary();
}
final ByteBuffer buffer = this.buffer;
buffer.clear();
res = next.read(buffer);
if (res > 0) {
inflater.setInput(buffer.array(), buffer.arrayOffset(), res);
} else {
return res;
}
}
} else {
final byte[] space = new byte[remaining];
for (;;) {
try {
res = inflater.inflate(space);
} catch (DataFormatException e) {
throw new IOException(e);
}
if (res > 0) {
dst.put(space, 0, res);
return res;
}
if (inflater.needsDictionary()) {
throw msg.inflaterNeedsDictionary();
}
final ByteBuffer buffer = this.buffer;
buffer.clear();
res = next.read(buffer);
if (res > 0) {
inflater.setInput(buffer.array(), buffer.arrayOffset(), res);
} else {
return res;
}
}
}
}
public long read(final ByteBuffer[] dsts) throws IOException {
return read(dsts, 0, dsts.length);
}
public long read(final ByteBuffer[] dsts, final int offset, final int length) throws IOException {
for (int i = 0; i < length; i ++) {
final ByteBuffer buffer = dsts[i + offset];
if (buffer.hasRemaining()) {
return read(buffer);
}
}
return 0L;
}
public void terminateReads() throws IOException {
inflater.end();
next.terminateReads();
}
public void awaitReadable() throws IOException {
if (! inflater.needsInput()) {
return;
}
next.awaitReadable();
}
public void awaitReadable(final long time, final TimeUnit timeUnit) throws IOException {
if (! inflater.needsInput()) {
return;
}
next.awaitReadable(time, timeUnit);
}
}