org.tarantool.protocol.ReadableViaSelectorChannel Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of connector Show documentation

Tarantool client for java

The newest version!

package org.tarantool.protocol;

import org.tarantool.CommunicationException;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.ReadableByteChannel;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.nio.channels.spi.SelectorProvider;

/**
 * Implements blocking-read based on non-blocking socket channel.
 * 

 * Implied to use in {@link ProtoUtils} class for blocking-read operations.
 */
public class ReadableViaSelectorChannel implements ReadableByteChannel {
    private final SocketChannel channel;
    private final Selector selector;

    public ReadableViaSelectorChannel(SocketChannel channel) throws IOException {
        if (channel.isBlocking()) {
            throw new IllegalArgumentException("Channel have to be non-blocking");
        }

        this.channel = channel;
        selector = SelectorProvider.provider().openSelector();
        channel.register(selector, SelectionKey.OP_READ);
    }

    @Override
    public int read(ByteBuffer buffer) throws IOException {
        int count;
        int n;

        count = n = channel.read(buffer);

        if (n < 0) {
            throw new CommunicationException("Channel read failed " + formatReadBytes(n));
        }

        while (buffer.remaining() > 0) {
            selector.select();
            n = channel.read(buffer);
            if (n < 0) {
                throw new CommunicationException("Channel read failed: " + formatReadBytes(n));
            }
            count += n;
        }
        return count;
    }

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

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

    /**
     * Formats the bytes count to a human readable message.
     *
     * @param bytes number of bytes
     *
     * @return formatted message
     */
    private String formatReadBytes(int bytes) {
        return bytes < 0 ? "EOF" : bytes + " bytes";
    }
}