52 lines
1.2 KiB
Java
52 lines
1.2 KiB
Java
package com.jenkov.nioserver;
|
|
|
|
import java.io.IOException;
|
|
import java.nio.ByteBuffer;
|
|
import java.nio.channels.SocketChannel;
|
|
|
|
/**
|
|
* Project: <strong>java-nio-server</strong><br>
|
|
* File: <strong>Socket.java</strong><br>
|
|
* Created: <strong>16 Oct 2015</strong><br>
|
|
*
|
|
* @author jjenkov
|
|
*/
|
|
public class Socket {
|
|
|
|
public long socketId;
|
|
|
|
public SocketChannel socketChannel;
|
|
public IMessageReader messageReader;
|
|
public MessageWriter messageWriter;
|
|
|
|
public boolean endOfStreamReached;
|
|
|
|
public Socket(SocketChannel socketChannel) { this.socketChannel = socketChannel; }
|
|
|
|
public int read(ByteBuffer byteBuffer) throws IOException {
|
|
int bytesRead = socketChannel.read(byteBuffer);
|
|
int totalBytesRead = bytesRead;
|
|
|
|
while (bytesRead > 0) {
|
|
bytesRead = socketChannel.read(byteBuffer);
|
|
totalBytesRead += bytesRead;
|
|
}
|
|
if (bytesRead == -1) endOfStreamReached = true;
|
|
|
|
return totalBytesRead;
|
|
}
|
|
|
|
public int write(ByteBuffer byteBuffer) throws IOException {
|
|
int bytesWritten = socketChannel.write(byteBuffer);
|
|
int totalBytesWritten = bytesWritten;
|
|
|
|
while (bytesWritten > 0 && byteBuffer.hasRemaining()) {
|
|
bytesWritten = socketChannel.write(byteBuffer);
|
|
totalBytesWritten += bytesWritten;
|
|
}
|
|
|
|
return totalBytesWritten;
|
|
}
|
|
|
|
}
|