org.xnio.conduits.InflatingStreamSourceConduit Maven / Gradle / Ivy
/*
* 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);
}
}