consul重新注册,避免注册失败的情况。

/**
 * consul服务重新注册
 * @author xxm
 * @date 2020/4/30
 * @since 1.8
 */
@Slf4j
public class ConsulRetryRegistry implements CommandLineRunner {
    private ConsulAutoRegistration consulAutoRegistration;
    private ConsulServiceRegistry consulServiceRegistry;
    private DiscoveryClient discoveryClient;
    private ConsulRetryProperties properties;

    public ConsulRetryRegistry(
            ConsulAutoRegistration consulAutoRegistration,
            ConsulServiceRegistry consulServiceRegistry,
            DiscoveryClient discoveryClient,
            ConsulRetryProperties properties
    ) {
        this.consulAutoRegistration = consulAutoRegistration;
        this.consulServiceRegistry = consulServiceRegistry;
        this.discoveryClient = discoveryClient;
        this.properties = properties;
    }

    @Override
    public void run(String... args) {
        // 获取当前服务
        final NewService service = this.consulAutoRegistration.getService();
        // 启动一个线程进行服务监测
        ScheduledExecutorService executorService = new ScheduledThreadPoolExecutor(1,new BasicThreadFactory.Builder().daemon(true).build());
        executorService.scheduleAtFixedRate(() -> {
            // 健康检查
            if (!checkStatus(service)) {
                try {
                    //重新注册
                    registry();
                }catch (Exception e) {
                    log.error(e.getMessage());
                }
            }
        }, 0, this.properties.getInitialInterval(), TimeUnit.MILLISECONDS);
    }

    /**
     * 服务注册
     */
    private void registry() {
        this.consulServiceRegistry.register(this.consulAutoRegistration);
    }

    /**
     * 检查服务状态
     * @param service 服务
     * @return 返回布尔值,正常true,异常false
     */
    private boolean checkStatus(NewService service) {
        // 检测标志
        boolean flag = false;
        try {
            // 获取所有服务实例
            List<ServiceInstance> instances = this.discoveryClient.getInstances(service.getName());
            for (ServiceInstance instance : instances) {
                // 判断是否为当前服务
                if (instance.getServiceId().equals(service.getName())) {
                    flag = true;
                    break;
                }
            }
        }catch (Exception e){
            log.error(e.getMessage());
        }
        return flag;
    }
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值