沟通非阻塞IO与阻塞IO - 出入流

最新推荐文章于 2024-04-28 03:24:50 发布

原创最新推荐文章于 2024-04-28 03:24:50 发布 · 1.6k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#io #import #buffer #byte #null #less

此博客展示了Java中ChannelOutputStream类的实现代码。该类继承自OutputStream，利用WritableByteChannel和ByteBuffer进行数据处理，包含构造方法、write、flush和close方法，可处理字节写入、缓冲区刷新和通道关闭等操作。

import java.io.IOException;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import java.nio.channels.WritableByteChannel;

public class ChannelOutputStream extends OutputStream {

private WritableByteChannel channel;

private ByteBuffer buffer;

private int bufferSize;

private int bufferCount;

public ChannelOutputStream(WritableByteChannel channel) throws IllegalArgumentException {
this(channel, 1024);
}

public ChannelOutputStream(WritableByteChannel channel, int bufferSize) throws IllegalArgumentException {
    if (channel == null) {
      throw new IllegalArgumentException("The writable byte channel is null");
    }

    if (bufferSize < 1) {
      throw new IllegalArgumentException("The buffer size is less than 1");
    }

    this.channel = channel;
    this.buffer = ByteBuffer.allocate(bufferSize);
    this.bufferSize = bufferSize;
    this.bufferCount = 0;
}

public void write(int b) throws IOException {
    buffer.put((byte) b);
    bufferCount++;
    if (bufferCount >= bufferSize) {
      flush();
    }
}

public void flush() throws IOException {
    buffer.flip();
    channel.write(buffer);
    buffer.clear();
    bufferCount = 0;
}

public void close() throws IOException {
    if (bufferCount > 0) {
      flush();
    }
    channel.close();
}
}