Merge branch 'master' into 3.0.0
Nikita
8 years ago
commit
e97ebdeb35
@ -0,0 +1,139 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2016 Nikita Koksharov
|
||||||
|
*
|
||||||
|
* 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 org.redisson.connection;
|
||||||
|
|
||||||
|
import java.net.InetAddress;
|
||||||
|
import java.net.URI;
|
||||||
|
import java.net.UnknownHostException;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Map.Entry;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
|
||||||
|
import org.redisson.client.RedisConnectionException;
|
||||||
|
import org.redisson.connection.ClientConnectionsEntry.FreezeReason;
|
||||||
|
import org.redisson.misc.URIBuilder;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import io.netty.util.concurrent.GlobalEventExecutor;
|
||||||
|
import io.netty.util.concurrent.ScheduledFuture;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* DNS changes monitor.
|
||||||
|
*
|
||||||
|
* @author Nikita Koksharov
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
public class DNSMonitor {
|
||||||
|
|
||||||
|
private static final Logger log = LoggerFactory.getLogger(DNSMonitor.class);
|
||||||
|
|
||||||
|
private ScheduledFuture<?> dnsMonitorFuture;
|
||||||
|
|
||||||
|
private ConnectionManager connectionManager;
|
||||||
|
|
||||||
|
private final Map<URI, InetAddress> masters = new HashMap<URI, InetAddress>();
|
||||||
|
private final Map<URI, InetAddress> slaves = new HashMap<URI, InetAddress>();
|
||||||
|
|
||||||
|
private long dnsMonitoringInterval;
|
||||||
|
|
||||||
|
public DNSMonitor(ConnectionManager connectionManager, Set<URI> masterHosts, Set<URI> slaveHosts, long dnsMonitoringInterval) {
|
||||||
|
for (URI host : masterHosts) {
|
||||||
|
try {
|
||||||
|
masters.put(host, InetAddress.getByName(host.getHost()));
|
||||||
|
} catch (UnknownHostException e) {
|
||||||
|
throw new RedisConnectionException("Unknown host: " + host.getHost(), e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for (URI host : slaveHosts) {
|
||||||
|
try {
|
||||||
|
slaves.put(host, InetAddress.getByName(host.getHost()));
|
||||||
|
} catch (UnknownHostException e) {
|
||||||
|
throw new RedisConnectionException("Unknown host: " + host.getHost(), e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
this.connectionManager = connectionManager;
|
||||||
|
this.dnsMonitoringInterval = dnsMonitoringInterval;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void start() {
|
||||||
|
monitorDnsChange();
|
||||||
|
log.debug("DNS monitoring enabled; Current masters: {}, slaves: {}", masters, slaves);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void stop() {
|
||||||
|
if (dnsMonitorFuture != null) {
|
||||||
|
dnsMonitorFuture.cancel(true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void monitorDnsChange() {
|
||||||
|
dnsMonitorFuture = GlobalEventExecutor.INSTANCE.schedule(new Runnable() {
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
// As InetAddress.getByName call is blocking. Method should be run in dedicated thread
|
||||||
|
connectionManager.getExecutor().execute(new Runnable() {
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
try {
|
||||||
|
for (Entry<URI, InetAddress> entry : masters.entrySet()) {
|
||||||
|
InetAddress master = entry.getValue();
|
||||||
|
InetAddress now = InetAddress.getByName(entry.getKey().getHost());
|
||||||
|
if (!now.getHostAddress().equals(master.getHostAddress())) {
|
||||||
|
log.info("Detected DNS change. {} has changed from {} to {}", entry.getKey().getHost(), master.getHostAddress(), now.getHostAddress());
|
||||||
|
for (MasterSlaveEntry entrySet : connectionManager.getEntrySet()) {
|
||||||
|
if (entrySet.getClient().getAddr().getHostName().equals(entry.getKey().getHost())
|
||||||
|
&& entrySet.getClient().getAddr().getPort() == entry.getKey().getPort()) {
|
||||||
|
entrySet.changeMaster(entry.getKey());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
masters.put(entry.getKey(), now);
|
||||||
|
log.info("Master {} has been changed", entry.getKey().getHost());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (Entry<URI, InetAddress> entry : slaves.entrySet()) {
|
||||||
|
InetAddress slave = entry.getValue();
|
||||||
|
InetAddress updatedSlave = InetAddress.getByName(entry.getKey().getHost());
|
||||||
|
if (!updatedSlave.getHostAddress().equals(slave.getHostAddress())) {
|
||||||
|
log.info("Detected DNS change. {} has changed from {} to {}", entry.getKey().getHost(), slave.getHostAddress(), updatedSlave.getHostAddress());
|
||||||
|
for (MasterSlaveEntry masterSlaveEntry : connectionManager.getEntrySet()) {
|
||||||
|
URI uri = URIBuilder.create("redis://" + slave.getHostAddress() + ":" + entry.getKey().getPort());
|
||||||
|
if (masterSlaveEntry.slaveDown(uri, FreezeReason.MANAGER)) {
|
||||||
|
masterSlaveEntry.slaveUp(entry.getKey(), FreezeReason.MANAGER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
slaves.put(entry.getKey(), updatedSlave);
|
||||||
|
log.info("Slave {} has been changed", entry.getKey().getHost());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (Exception e) {
|
||||||
|
log.error(e.getMessage(), e);
|
||||||
|
} finally {
|
||||||
|
monitorDnsChange();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
}, dnsMonitoringInterval, TimeUnit.MILLISECONDS);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue