/*
* Copyright (c) 2008-2013, Hazelcast, Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hazelcast.nio.tcp;
import com.hazelcast.logging.ILogger;
import com.hazelcast.logging.SystemLogService;
import com.hazelcast.nio.ConnectionType;
import java.io.IOException;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.util.logging.Level;
abstract class AbstractSelectionHandler implements SelectionHandler {
protected final ILogger logger;
protected final SocketChannelWrapper socketChannel;
protected final TcpIpConnection connection;
protected final TcpIpConnectionManager connectionManager;
protected final SystemLogService systemLogService;
private SelectionKey sk;
public AbstractSelectionHandler(final TcpIpConnection connection) {
this.connection = connection;
this.socketChannel = connection.getSocketChannelWrapper();
this.connectionManager = connection.getConnectionManager();
this.logger = connectionManager.ioService.getLogger(this.getClass().getName());
this.systemLogService = connectionManager.ioService.getSystemLogService();
}
protected void shutdown() {
}
final void handleSocketException(Throwable e) {
if (e instanceof OutOfMemoryError) {
connectionManager.ioService.onOutOfMemory((OutOfMemoryError) e);
}
if (sk != null) {
sk.cancel();
}
connection.close(e);
final ConnectionType connectionType = connection.getType();
if (connectionType.isClient() && !connectionType.isBinary()) {
return;
}
StringBuilder sb = new StringBuilder();
sb.append(Thread.currentThread().getName());
sb.append(" Closing socket to endpoint ");
sb.append(connection.getEndPoint());
sb.append(", Cause:").append(e);
final Level level = connectionManager.ioService.isActive() ? Level.WARNING : Level.FINEST;
if (e instanceof IOException) {
logger.log(level, sb.toString());
} else {
logger.log(level, sb.toString(), e);
}
}
final void registerOp(final Selector selector, final int operation) {
try {
if (!connection.live()) {
return;
}
if (sk == null) {
sk = socketChannel.keyFor(selector);
}
if (sk == null) {
sk = socketChannel.register(selector, operation, this);
} else {
sk.interestOps(sk.interestOps() | operation);
if (sk.attachment() != this) {
sk.attach(this);
}
}
} catch (Throwable e) {
handleSocketException(e);
}
}
}