com.emc.mongoose.base.item.io.AsyncChannelWrapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongoose-base Show documentation

Mongoose is a high-load storage performance testing tool

There is a newer version: 4.3.10

package com.emc.mongoose.base.item.io;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.AsynchronousByteChannel;
import java.nio.channels.AsynchronousChannel;
import java.nio.channels.AsynchronousFileChannel;
import java.nio.channels.CompletionHandler;

public final class AsyncChannelWrapper
implements AsyncChannel {

	private final AsynchronousChannel channel;
	private final boolean isFileChannel;

	public AsyncChannelWrapper(final AsynchronousChannel channel) {
		this.channel = channel;
		this.isFileChannel = channel instanceof AsynchronousFileChannel;
	}

	@Override
	public final boolean isFileChannel() {
		return isFileChannel;
	}

	@Override
	public final AsynchronousChannel wrapped() {
		return channel;
	}

	@Override
	public final  void read(
		final ByteBuffer dst, final long position, final A attach,
		final CompletionHandler handler
	) {
		if(isFileChannel) {
			((AsynchronousFileChannel) channel).read(dst, position, attach, handler);
		} else {
			((AsynchronousByteChannel) channel).read(dst, attach, handler);
		}
	}

	@Override
	public final  void write(
		final ByteBuffer src, final long position, final A attach, final CompletionHandler handler
	) {
		if(isFileChannel) {
			((AsynchronousFileChannel) channel).write(src, position, attach, handler);
		} else {
			((AsynchronousByteChannel) channel).write(src, attach, handler);
		}
	}

	@Override
	public final boolean isOpen() {
		return channel.isOpen();
	}

	@Override
	public final void close()
	throws IOException {
		channel.close();
	}
}