/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.cassandra.locator;
import java.net.InetAddress;
import java.net.UnknownHostException;
import com.google.common.annotations.VisibleForTesting;
import org.apache.cassandra.gms.*;
import org.apache.cassandra.net.MessagingService;
import org.apache.cassandra.net.OutboundTcpConnectionPool;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Sidekick helper for snitches that want to reconnect from one IP addr for a node to another.
* Typically, this is for situations like EC2 where a node will have a public address and a private address,
* where we connect on the public, discover the private, and reconnect on the private.
*/
public class ReconnectableSnitchHelper implements IEndpointStateChangeSubscriber
{
private static final Logger logger = LoggerFactory.getLogger(ReconnectableSnitchHelper.class);
private final IEndpointSnitch snitch;
private final String localDc;
private final boolean preferLocal;
public ReconnectableSnitchHelper(IEndpointSnitch snitch, String localDc, boolean preferLocal)
{
this.snitch = snitch;
this.localDc = localDc;
this.preferLocal = preferLocal;
}
private void reconnect(InetAddress publicAddress, VersionedValue localAddressValue)
{
try
{
reconnect(publicAddress, InetAddress.getByName(localAddressValue.value), snitch, localDc);
}
catch (UnknownHostException e)
{
logger.error("Error in getting the IP address resolved: ", e);
}
}
@VisibleForTesting
static void reconnect(InetAddress publicAddress, InetAddress localAddress, IEndpointSnitch snitch, String localDc)
{
OutboundTcpConnectionPool cp = MessagingService.instance().getConnectionPool(publicAddress);
//InternodeAuthenticator said don't connect
if (cp == null)
{
logger.debug("InternodeAuthenticator said don't reconnect to {} on {}", publicAddress, localAddress);
return;
}
if (snitch.getDatacenter(publicAddress).equals(localDc)
&& !cp.endPoint().equals(localAddress))
{
cp.reset(localAddress);
logger.debug("Initiated reconnect to an Internal IP {} for the {}", localAddress, publicAddress);
}
}
public void beforeChange(InetAddress endpoint, EndpointState currentState, ApplicationState newStateKey, VersionedValue newValue)
{
// no-op
}
public void onJoin(InetAddress endpoint, EndpointState epState)
{
if (preferLocal && !Gossiper.instance.isDeadState(epState) && epState.getApplicationState(ApplicationState.INTERNAL_IP) != null)
reconnect(endpoint, epState.getApplicationState(ApplicationState.INTERNAL_IP));
}
public void onChange(InetAddress endpoint, ApplicationState state, VersionedValue value)
{
if (preferLocal && state == ApplicationState.INTERNAL_IP && !Gossiper.instance.isDeadState(Gossiper.instance.getEndpointStateForEndpoint(endpoint)))
reconnect(endpoint, value);
}
public void onAlive(InetAddress endpoint, EndpointState state)
{
if (preferLocal && state.getApplicationState(ApplicationState.INTERNAL_IP) != null)
reconnect(endpoint, state.getApplicationState(ApplicationState.INTERNAL_IP));
}
public void onDead(InetAddress endpoint, EndpointState state)
{
// do nothing.
}
public void onRemove(InetAddress endpoint)
{
// do nothing.
}
public void onRestart(InetAddress endpoint, EndpointState state)
{
// do nothing.
}
}