一.前言
我们平时发布服务端新版本时总会担心一个事情，就是发布时如果某些服务在运行，但因为发版服务器宕机了，而还没完跑完的服务刚好是很重要的业务，有时候会带来不可估量的损失，所以我们需要一个让服务安全关闭的方案，即保证等服务所有在完全跑完后才宕机。
二.背景
我现在负责这个项目，因为前期进度比较赶，所以队列(RocketMQ)消费整合到接口里面，消费使用线程池并发进行，但因为目前是初期运营，接口需要频繁发布，但队列的消费量也相当不小了(出于性能考虑，大部分耗时业务使用队列消费完成)，所以没办法保证某些关键业务在接口发布时完整跑完，所以使用SpringBoot拆出一个专门进行队列消费的服务项目，但这样还是没有根本解决问题，因为队列消费项目也是需要发布的。
三.解决思路
为了所有服务都能完全跑完，首先我们要保证没有新的消费任务进来，所以在需要宕机时必须先保证消费者shutdown掉，然后再保证所有服务完全运行后宕机，但直接宕机会把所有在跑的线程结束掉，所以我们可以使用线程堵塞来实现接受到宕机信号后进行某些处理再完全宕机；
这时我们需要一个判断任务完全跑完的机制，这里我们创建一个专门计算任务的类，当队列任务任务消费进来时，我们把任务数+1，当任务执行完毕时我们任务-1，以任务数最终是否为0来进行判断所有任务是否完全跑完，因为队列消费是基于线程池执行的，所以我们需要保证计数的线程安全，这时候我们可以使用Java原生类库提供的AtomicInteger类来实现线程计数，AtomicInteger的计数具有原子性，有兴趣的小伙伴可以去研究一下
最低0.47元/天 解锁文章
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
