/*
* 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;
import com.hazelcast.ascii.TextCommandService;
import com.hazelcast.config.NetworkConfig;
import com.hazelcast.config.SSLConfig;
import com.hazelcast.config.SocketInterceptorConfig;
import com.hazelcast.config.SymmetricEncryptionConfig;
import com.hazelcast.instance.MemberImpl;
import com.hazelcast.instance.Node;
import com.hazelcast.instance.OutOfMemoryErrorDispatcher;
import com.hazelcast.logging.ILogger;
import com.hazelcast.logging.SystemLogService;
import com.hazelcast.nio.serialization.Data;
import com.hazelcast.nio.serialization.PortableContext;
import com.hazelcast.nio.serialization.SerializationService;
import com.hazelcast.spi.EventService;
import com.hazelcast.spi.ExecutionService;
import com.hazelcast.spi.impl.NodeEngineImpl;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;
public class NodeIOService implements IOService {
private final Node node;
private final NodeEngineImpl nodeEngine;
public NodeIOService(Node node) {
this.node = node;
this.nodeEngine = node.nodeEngine;
}
@Override
public boolean isActive() {
return node.isActive();
}
@Override
public ILogger getLogger(String name) {
return node.getLogger(name);
}
@Override
public SystemLogService getSystemLogService() {
return node.getSystemLogService();
}
@Override
public void onOutOfMemory(OutOfMemoryError oom) {
OutOfMemoryErrorDispatcher.onOutOfMemory(oom);
}
@Override
public Address getThisAddress() {
return node.getThisAddress();
}
@Override
public void onFatalError(Exception e) {
getSystemLogService().logConnection(e.getClass().getName() + ": " + e.getMessage());
new Thread(node.threadGroup, node.getThreadNamePrefix("io.error.shutdown")) {
public void run() {
node.shutdown(false);
}
} .start();
}
@Override
public SocketInterceptorConfig getSocketInterceptorConfig() {
return node.getConfig().getNetworkConfig().getSocketInterceptorConfig();
}
@Override
public SymmetricEncryptionConfig getSymmetricEncryptionConfig() {
return node.getConfig().getNetworkConfig().getSymmetricEncryptionConfig();
}
@Override
public SSLConfig getSSLConfig() {
return node.getConfig().getNetworkConfig().getSSLConfig();
}
@Override
public void handleMemberPacket(final Packet packet) {
final Address endPoint = packet.getConn().getEndPoint();
if (endPoint != null) {
final MemberImpl member = node.clusterService.getMember(endPoint);
if (member != null) {
member.didRead();
}
}
nodeEngine.handlePacket(packet);
}
@Override
public void handleClientPacket(Packet p) {
node.clientEngine.handlePacket(p);
}
@Override
public TextCommandService getTextCommandService() {
return node.getTextCommandService();
}
@Override
public boolean isMemcacheEnabled() {
return node.groupProperties.MEMCACHE_ENABLED.getBoolean();
}
@Override
public boolean isRestEnabled() {
return node.groupProperties.REST_ENABLED.getBoolean();
}
@Override
public void removeEndpoint(final Address endPoint) {
nodeEngine.getExecutionService().execute(ExecutionService.IO_EXECUTOR, new Runnable() {
@Override
public void run() {
node.clusterService.removeAddress(endPoint);
}
});
}
@Override
public String getThreadPrefix() {
return node.getThreadPoolNamePrefix("IO");
}
@Override
public ThreadGroup getThreadGroup() {
return node.threadGroup;
}
@Override
public void onFailedConnection(Address address) {
if (!node.joined()) {
node.failedConnection(address);
}
}
@Override
public void shouldConnectTo(Address address) {
if (node.getThisAddress().equals(address)) {
throw new RuntimeException("Connecting to self! " + address);
}
}
@Override
public boolean isSocketBind() {
return node.groupProperties.SOCKET_CLIENT_BIND.getBoolean();
}
@Override
public boolean isSocketBindAny() {
return node.groupProperties.SOCKET_CLIENT_BIND_ANY.getBoolean();
}
@Override
public int getSocketReceiveBufferSize() {
return this.node.getGroupProperties().SOCKET_RECEIVE_BUFFER_SIZE.getInteger();
}
@Override
public int getSocketSendBufferSize() {
return this.node.getGroupProperties().SOCKET_SEND_BUFFER_SIZE.getInteger();
}
@Override
public int getSocketLingerSeconds() {
return this.node.getGroupProperties().SOCKET_LINGER_SECONDS.getInteger();
}
@Override
public boolean getSocketKeepAlive() {
return this.node.getGroupProperties().SOCKET_KEEP_ALIVE.getBoolean();
}
@Override
public boolean getSocketNoDelay() {
return this.node.getGroupProperties().SOCKET_NO_DELAY.getBoolean();
}
@Override
public int getSelectorThreadCount() {
return node.groupProperties.IO_THREAD_COUNT.getInteger();
}
@Override
public void onDisconnect(final Address endpoint) {
}
@Override
public boolean isClient() {
return false;
}
@Override
public long getConnectionMonitorInterval() {
return node.groupProperties.CONNECTION_MONITOR_INTERVAL.getLong();
}
@Override
public int getConnectionMonitorMaxFaults() {
return node.groupProperties.CONNECTION_MONITOR_MAX_FAULTS.getInteger();
}
@Override
public void executeAsync(final Runnable runnable) {
nodeEngine.getExecutionService().execute(ExecutionService.IO_EXECUTOR, runnable);
}
@Override
public EventService getEventService() {
return nodeEngine.getEventService();
}
@Override
public Data toData(Object obj) {
return nodeEngine.toData(obj);
}
@Override
public Object toObject(Data data) {
return nodeEngine.toObject(data);
}
@Override
public SerializationService getSerializationService() {
return node.getSerializationService();
}
@Override
public PortableContext getPortableContext() {
return node.getSerializationService().getPortableContext();
}
@Override
public Collection<Integer> getOutboundPorts() {
final NetworkConfig networkConfig = node.getConfig().getNetworkConfig();
final Collection<String> portDefinitions = getPortDefinitions(networkConfig);
final Set<Integer> ports = getPorts(networkConfig);
if (portDefinitions.isEmpty() && ports.isEmpty()) {
// means any port
return Collections.emptySet();
}
if (portDefinitions.contains("*") || portDefinitions.contains("0")) {
// means any port
return Collections.emptySet();
}
transformPortDefinitionsToPorts(portDefinitions, ports);
if (ports.contains(0)) {
// means any port
return Collections.emptySet();
}
return ports;
}
private void transformPortDefinitionsToPorts(Collection<String> portDefinitions, Set<Integer> ports) {
// not checking port ranges...
for (String portDef : portDefinitions) {
String[] portDefs = portDef.split("[,; ]");
for (String def : portDefs) {
def = def.trim();
final int dashPos = def.indexOf('-');
if (dashPos > 0) {
final int start = Integer.parseInt(def.substring(0, dashPos));
final int end = Integer.parseInt(def.substring(dashPos + 1));
for (int port = start; port <= end; port++) {
ports.add(port);
}
} else {
ports.add(Integer.parseInt(def));
}
}
}
}
private Set<Integer> getPorts(NetworkConfig networkConfig) {
return networkConfig.getOutboundPorts() == null
? new HashSet<Integer>() : new HashSet<Integer>(networkConfig.getOutboundPorts());
}
private Collection<String> getPortDefinitions(NetworkConfig networkConfig) {
return networkConfig.getOutboundPortDefinitions() == null
? Collections.<String>emptySet() : networkConfig.getOutboundPortDefinitions();
}
}