com.hazelcast.nio.NodeIOService Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2015, 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.client.impl.protocol.ClientMessage;
import com.hazelcast.internal.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.HazelcastThreadGroup;
import com.hazelcast.instance.MemberImpl;
import com.hazelcast.instance.Node;
import com.hazelcast.instance.OutOfMemoryErrorDispatcher;
import com.hazelcast.logging.ILogger;
import com.hazelcast.nio.serialization.Data;
import com.hazelcast.nio.serialization.SerializationService;
import com.hazelcast.nio.tcp.PacketReader;
import com.hazelcast.nio.tcp.PacketWriter;
import com.hazelcast.nio.tcp.SocketChannelWrapperFactory;
import com.hazelcast.nio.tcp.TcpIpConnection;
import com.hazelcast.spi.EventService;
import com.hazelcast.spi.ExecutionService;
import com.hazelcast.spi.impl.NodeEngineImpl;
import com.hazelcast.spi.impl.transceiver.PacketTransceiver;
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;
private final PacketTransceiver packetTransceiver;
public NodeIOService(Node node, NodeEngineImpl nodeEngine) {
this.node = node;
this.nodeEngine = nodeEngine;
this.packetTransceiver = nodeEngine.getPacketTransceiver();
}
@Override
public boolean isActive() {
return node.isActive();
}
@Override
public ILogger getLogger(String name) {
return node.getLogger(name);
}
@Override
public void onOutOfMemory(OutOfMemoryError oom) {
OutOfMemoryErrorDispatcher.onOutOfMemory(oom);
}
@Override
public Address getThisAddress() {
return node.getThisAddress();
}
@Override
public void onFatalError(Exception e) {
HazelcastThreadGroup threadGroup = node.getHazelcastThreadGroup();
new Thread(threadGroup.getInternalThreadGroup(), threadGroup.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();
}
}
packetTransceiver.receive(packet);
}
@Override
public void handleClientPacket(Packet p) {
node.clientEngine.handlePacket(p);
}
@Override
public void handleClientMessage(ClientMessage cm, Connection connection) {
node.clientEngine.handleClientMessage(cm, connection);
}
@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() {
HazelcastThreadGroup threadGroup = node.getHazelcastThreadGroup();
return threadGroup.getThreadPoolNamePrefix("IO");
}
@Override
public ThreadGroup getThreadGroup() {
HazelcastThreadGroup threadGroup = node.getHazelcastThreadGroup();
return threadGroup.getInternalThreadGroup();
}
@Override
public void onSuccessfulConnection(Address address) {
if (!node.joined()) {
node.getJoiner().unblacklist(address);
}
}
@Override
public void onFailedConnection(Address address) {
if (!node.joined()) {
node.getJoiner().blacklist(address, false);
}
}
@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 int getSocketConnectTimeoutSeconds() {
return this.node.getGroupProperties().SOCKET_CONNECT_TIMEOUT_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 int getBalancerIntervalSeconds() {
return node.groupProperties.IO_BALANCER_INTERVAL_SECONDS.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 SocketChannelWrapperFactory getSocketChannelWrapperFactory() {
return node.getNodeExtension().getSocketChannelWrapperFactory();
}
@Override
public MemberSocketInterceptor getMemberSocketInterceptor() {
return node.getNodeExtension().getMemberSocketInterceptor();
}
@Override
public PacketReader createPacketReader(TcpIpConnection connection) {
return node.getNodeExtension().createPacketReader(connection, this);
}
@Override
public PacketWriter createPacketWriter(TcpIpConnection connection) {
return node.getNodeExtension().createPacketWriter(connection, this);
}
@Override
public Collection getOutboundPorts() {
final NetworkConfig networkConfig = node.getConfig().getNetworkConfig();
final Collection portDefinitions = getPortDefinitions(networkConfig);
final Set 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 portDefinitions, Set ports) {
// not checking port ranges...
for (String portDef : portDefinitions) {
String[] portDefs = portDef.split("[,; ]");
for (String def : portDefs) {
def = def.trim();
if (def.isEmpty()) {
continue;
}
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 getPorts(NetworkConfig networkConfig) {
return networkConfig.getOutboundPorts() == null
? new HashSet() : new HashSet(networkConfig.getOutboundPorts());
}
private Collection getPortDefinitions(NetworkConfig networkConfig) {
return networkConfig.getOutboundPortDefinitions() == null
? Collections.emptySet() : networkConfig.getOutboundPortDefinitions();
}
}