use of com.hazelcast.nio.IOService in project hazelcast by hazelcast.
the class SocketWriterInitializerImpl method initHandler.
private void initHandler(TcpIpConnection connection, SocketWriter writer, String protocol) {
WriteHandler handler;
if (CLUSTER.equals(protocol)) {
IOService ioService = connection.getConnectionManager().getIoService();
handler = ioService.createWriteHandler(connection);
} else if (CLIENT_BINARY_NEW.equals(protocol)) {
handler = new ClientWriteHandler();
} else {
handler = new TextWriteHandler(connection);
}
writer.initWriteHandler(handler);
}
use of com.hazelcast.nio.IOService in project hazelcast by hazelcast.
the class SocketReaderInitializerImpl method init.
@Override
public void init(TcpIpConnection connection, SocketReader reader) throws IOException {
TcpIpConnectionManager connectionManager = connection.getConnectionManager();
IOService ioService = connectionManager.getIoService();
ByteBuffer protocolBuffer = reader.getProtocolBuffer();
SocketChannelWrapper socketChannel = reader.getSocketChannel();
int readBytes = socketChannel.read(protocolBuffer);
if (readBytes == -1) {
throw new EOFException("Could not read protocol type!");
}
if (readBytes == 0 && connectionManager.isSSLEnabled()) {
// when using SSL, we can read 0 bytes since data read from socket can be handshake frames.
return;
}
if (protocolBuffer.hasRemaining()) {
// we have not yet received all protocol bytes
return;
}
ReadHandler readHandler;
String protocol = bytesToString(protocolBuffer.array());
SocketWriter socketWriter = connection.getSocketWriter();
if (CLUSTER.equals(protocol)) {
initInputBuffer(connection, reader, ioService.getSocketReceiveBufferSize());
connection.setType(MEMBER);
socketWriter.setProtocol(CLUSTER);
readHandler = ioService.createReadHandler(connection);
} else if (CLIENT_BINARY_NEW.equals(protocol)) {
initInputBuffer(connection, reader, ioService.getSocketClientReceiveBufferSize());
socketWriter.setProtocol(CLIENT_BINARY_NEW);
readHandler = new ClientReadHandler(reader.getNormalFramesReadCounter(), connection, ioService);
} else {
ByteBuffer inputBuffer = initInputBuffer(connection, reader, ioService.getSocketReceiveBufferSize());
socketWriter.setProtocol(TEXT);
inputBuffer.put(protocolBuffer.array());
readHandler = new TextReadHandler(connection);
connectionManager.incrementTextConnections();
}
if (readHandler == null) {
throw new IOException("Could not initialize ReadHandler!");
}
reader.initReadHandler(readHandler);
}
use of com.hazelcast.nio.IOService in project hazelcast by hazelcast.
the class SocketWriterInitializerImpl method initOutputBuffer.
private void initOutputBuffer(TcpIpConnection connection, SocketWriter writer, String protocol) {
IOService ioService = connection.getConnectionManager().getIoService();
int sizeKb = CLUSTER.equals(protocol) ? ioService.getSocketSendBufferSize() : ioService.getSocketClientSendBufferSize();
int size = KILO_BYTE * sizeKb;
ByteBuffer outputBuffer = newByteBuffer(size, ioService.isSocketBufferDirect());
if (CLUSTER.equals(protocol)) {
outputBuffer.put(stringToBytes(CLUSTER));
}
writer.initOutputBuffer(outputBuffer);
try {
connection.setSendBufferSize(size);
} catch (SocketException e) {
logger.finest("Failed to adjust TCP send buffer of " + connection + " to " + size + " B.", e);
}
}
Aggregations