问题
用户数上升，服务要集群，如何实现主从机制，并且当主服务挂掉或停机维护时，其它任意从服务可自动变成主服务？
程序猿A：这还不简单，用zookeeper就行了，配上Apache curator更方便，直接帮你实现好Leader Election了。
程序猿B：嗯...zookeeper又要安装个服务，不想只为了这个主从又引入一个新东西，我们已经有redis了，能不能基于redis来实现？
答案是：yes。
思路
开始前，先说一下基本的实现思路：
- 先有一个redisson的分布式锁RLock，名称为：leader-lock
- 所有服务在启动的时候都去尝试获取leader锁
- 获取锁成功的服务为主服务
- 未获取锁的其它服务为从服务
- 从服务每隔几秒钟一直去尝试获取leader锁，当主服务挂掉或停机时，其中一个从服务就会获取到锁变成主服务
码起来
分布式锁的初始化
RLock leaderLock = redissonClient.getLock(“leader-lock”);
ElectionThread
有了锁后，我们需要一个专门的线程用于获取锁
class ElectionThread extends Thread {
private boolean isMaster = false;
public ElectionThread() {
setName("leader-election");
}
@Override
public void run() {
while (!stop) {
try {
if (isMaster) {
synchronized (masterLock) {
//leader锁获取到了，就不需要再去获取了，进入阻塞状态
masterLock.wait();
}
} else {
//所有从服务尝试获取leader锁，尝试并等待一定时间，如果未获取成功，就一直重试
isMaster = leaderLock.tryLock(WAIT_SECONDS, TimeUnit.SECONDS);
if (isMaster) {
//leader锁获取成功，当前服务为主服务
logger.info("got leadership");
}
}
} catch (InterruptedException e) {
}
}
}
//判断leader锁是否获取成功
public boolean isMaster() {
return isMaster;
}
}
tryHold
有了ElectionThread，需要提供一个方法启动它去获取锁
public void tryHold(String leaderName) {
//分布式锁的初始化
leaderLock = redissonClient.getLock(leaderName);
//启动获取锁的线程
electionThread.start();
}
锁释放
锁获取到了，如果要释放怎么释放？ElectionThread需要加上释放锁的逻辑
class ElectionThread extends Thread {
private boolean isMaster = false;
public ElectionThread() {
setName("leader-election");
}
@Override
public void run() {
while (!stop) {
try {
if (isMaster) {
synchronized (masterLock) {
//leader锁获取到了，就不需要再去获取了，进入阻塞状态
masterLock.wait();
}
} else {
//所有从服务尝试获取leader锁，尝试并等待一定时间，如果未获取成功，就一直重试
isMaster = leaderLock.tryLock(WAIT_SECONDS, TimeUnit.SECONDS);
if (isMaster) {
//leader锁获取成功，当前服务为主服务
logger.info("got leadership");
}
}
} catch (InterruptedException e) {
}
}
//如果leader锁被当前线程占用，就释放锁
if (leaderLock.isLocked() && leaderLock.isHeldByCurrentThread()) {
leaderLock.unlock();
}
if (isMaster) {
isMaster = false;
}
}
//判断leader锁是否获取成功
public boolean isMaster() {
return isMaster;
}
}
shutdown
等等，ElectionThread是加上了释放锁的逻辑了，但当ElectionThread得到锁的时候，线程已经阻塞了，我们需要在外部唤醒ElectionThread线程并跳出while循环
public void shutdown() {
//stop设为true，ElectionThread中的while循环即可退出
stop = true;
try {
synchronized (masterLock) {
//唤醒ElectionThread
masterLock.notifyAll();
}
//等待ElectionThread死亡
electionThread.join();
} catch (InterruptedException e) {
}
logger.info("shutdown and give up leadership");
}
shutdown hook
有了shutdown方法后，我们再加个shutdownHook，就可以在jvm停止时调用shutdown方法，leader锁就会被释放
public void tryHold(String leaderName) {
//分布式锁的初始化
leaderLock = redissonClient.getLock(leaderName);
//启动获取锁的线程
electionThread.start();
//shutdown hook
Runtime.getRuntime().addShutdownHook(new Thread(() -> shutdown()));
}
jvm的shutdownHook是在服务正常退出的情况下才会生效，如果服务异常退出，会怎样？leader锁会释放吗？放心，redisson有个lockWatchdogTimeout配置，这个配置会让redisson客户端处于正常状态的时候，给那些不会自动释放的锁延长过期时间，如果服务异常了，那些不会自动释放的锁由于没有延长过期时间，会被redis自动清除，所以leader锁即使在服务异常退出的情况下，也会自动释放。
lockWatchdogTimeout
lockWatchdogTimeout的配置可以用代码方式设置，也可以用配置文件方式设置，一定要注意它的单位是毫秒(我测试的时候设成了10，结果找了半天问题)，默认值是30000毫秒，即30秒。
import org.redisson.config.Config
//从yaml配置文件中读取配置
Config config = Config.fromYAML(configFile.getInputStream());
//代码方式设置，如果希望主从切换更快，时间可以设置成5秒
config.setLockWatchdogTimeout(5000)
应用
看起来差不多了，现在我们提供一个方法用于判断当前服务是否是主服务
public boolean isMaster() {
return electionThread.isMaster();
}
Ok，现在可以拿来用了，首先看下定时任务的情况，定时任务一般只想在主服务上运行，这时就可以这样写了
scheduler.scheduleAtFixedRate(() -> {
if (!isMaster()) {
//do nothing
return;
}
//do something
}, 0, 60, TimeUnit.MINUTES);
初始化问题
这个判断方法能不能在锁状态初始化完成之前阻塞，这样在类似上面的定时任务里（如1小时）做判断时，主服务不至于因为没初始化，然后就得等到下个小时才能执行，我们修改两个地方。
public boolean isMaster() {
//这里加上一个初始化锁，当没有初始化时，阻塞当前线程
synchronized (initLock) {
if (!isInit) {
try {
initLock.wait();
} catch (InterruptedException e) {
}
}
}
return electionThread.isMaster();
}
class ElectionThread extends Thread {
private boolean isMaster = false;
public ElectionThread() {
setName("leader-election");
}
@Override
public void run() {
while (!stop) {
try {
if (isMaster) {
synchronized (masterLock) {
if (isInit) {
//获得锁且已经初始化过就进入阻塞状态
masterLock.wait();
} else {
//获得锁，还未设置初始化状态，就等待一会儿，给线程机会设置初始化状态
masterLock.wait(Duration.ofSeconds(WAIT_SECONDS).toMillis());
}
}
} else {
isMaster = leaderLock.tryLock(WAIT_SECONDS, TimeUnit.SECONDS);
if (isMaster) {
logger.info("got leadership");
}
}
} catch (InterruptedException e) {
} finally {
//设置线程初始化状态
synchronized (initLock) {
if (!isInit) {
//初始化完成唤醒调用isMaster方法处于阻塞状态的线程，并设置初始化状态
initLock.notifyAll();
isInit = true;
}
}
}
}
if (leaderLock.isLocked() && leaderLock.isHeldByCurrentThread()) {
leaderLock.unlock();
}
if (isMaster) {
isMaster = false;
}
}
public boolean isMaster() {
return isMaster;
}
}
从服务变成主服务
还有一种情况，我们需要考虑，当主服务挂了，从服务变成主服务时，在上面的1小时定时任务已经过了执行时间，现在想在从服务变成主服务时，马上就执行任务，要怎么办？我们可以加上一个从服务变成主服务的监听器
private List listeners = new ArrayList<>();
public interface ElectionListener {
void onElected();
}
public void addElectionListener(ElectionListener electionListener) {
if (listeners.contains(electionListener)) {
return;
}
listeners.add(electionListener);
}
private void notifyElected() {
for (ElectionListener listener : listeners) {
listener.onElected();
}
}
//ElectionThread代码片断
isMaster = leaderLock.tryLock(WAIT_SECONDS, TimeUnit.SECONDS);
if (isMaster) {
logger.info("got leadership");
notifyElected();
}
总结
大功告成，最后看一下完整版的代码
LeaderElection
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.time.Duration;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;
public class LeaderElection {
private static final Logger logger = LoggerFactory.getLogger(LeaderElection.class);
private static final int WAIT_SECONDS = 1;
private RedissonClient redissonClient;
private RLock leaderLock;
private boolean stop = false;
private boolean isInit = false;
private Object masterLock = new Object();
private Object initLock = new Object();
private ElectionThread electionThread = new ElectionThread();
private List listeners = new ArrayList<>();
public void tryHold(String leaderName) {
leaderLock = redissonClient.getLock(leaderName);
electionThread.start();
Runtime.getRuntime().addShutdownHook(new Thread(() -> shutdown()));
}
public boolean isMaster() {
synchronized (initLock) {
if (!isInit) {
try {
initLock.wait();
} catch (InterruptedException e) {
}
}
}
return electionThread.isMaster();
}
public void addElectionListener(ElectionListener electionListener) {
if (listeners.contains(electionListener)) {
return;
}
listeners.add(electionListener);
}
public void shutdown() {
stop = true;
try {
synchronized (masterLock) {
masterLock.notifyAll();
}
electionThread.join();
listeners.clear();
} catch (InterruptedException e) {
}
logger.info("shutdown and give up leadership");
}
class ElectionThread extends Thread {
private boolean isMaster = false;
public ElectionThread() {
setName("leader-election");
}
@Override
public void run() {
while (!stop) {
try {
if (isMaster) {
synchronized (masterLock) {
if (isInit) {
masterLock.wait();
} else {
masterLock.wait(Duration.ofSeconds(WAIT_SECONDS).toMillis());
}
}
} else {
isMaster = leaderLock.tryLock(WAIT_SECONDS, TimeUnit.SECONDS);
if (isMaster) {
logger.info("got leadership");
notifyElected();
}
}
} catch (InterruptedException e) {
} finally {
synchronized (initLock) {
if (!isInit) {
initLock.notifyAll();
isInit = true;
}
}
}
}
if (leaderLock.isLocked() && leaderLock.isHeldByCurrentThread()) {
leaderLock.unlock();
}
if (isMaster) {
isMaster = false;
}
}
public boolean isMaster() {
return isMaster;
}
}
private void notifyElected() {
for (ElectionListener listener : listeners) {
listener.onElected();
}
}
public void setRedissonClient(RedissonClient redissonClient) {
this.redissonClient = redissonClient;
}
}
ElectionListener
public interface ElectionListener {
void onElected();
}
在springboot中的配置应用
@Configuration
public class LeaderElectionConfig {
@Value("${spring.application.name}")
private String appName;
@Bean(destroyMethod = "shutdown")
public LeaderElection leaderElection(RedissonClient redissonClient) {
LeaderElection leaderElection = new LeaderElection();
leaderElection.setRedissonClient(redissonClient);
leaderElection.tryHold("leader-lock-" + appName);
return leaderElection;
}
}
源码
https://github.com/huangyemin/wheel
https://gitee.com/huangyemin/wheel
原文地址