Spring环境下Dubbo优雅停机

要做优雅停机，一般都会用java.lang.Runtime#addShutdownHook。dubbo也不例外。调用的地方在org.apache.dubbo.config.AbstractConfig。具体代码如下：

static {
    // 其他代码
    // this is only for compatibility
    DubboShutdownHook.getDubboShutdownHook().register();
}

org.apache.dubbo.config.DubboShutdownHook相关代码如下：

public void register() {
    if (!registered.get() && registered.compareAndSet(false, true)) {
        Runtime.getRuntime().addShutdownHook(getDubboShutdownHook());
    }
}

public void run() {
    if (logger.isInfoEnabled()) {
        logger.info("Run shutdown hook now.");
    }
    doDestroy();
}

public void doDestroy() {
    if (!destroyed.compareAndSet(false, true)) {
        return;
    }
    // destroy all the registries
    AbstractRegistryFactory.destroyAll();
    // destroy all the protocols
    destroyProtocols();
}

spring也是通过java.lang.Runtime#addShutdownHook实现的。因此如果不特殊处理，spring和dubbo的shutdown hook会独立执行。结果就是服务还没从注册中心移除，spring容器就已经关闭。这种情况下，请求还会进来，但会报错。于是dubbo做了特殊处理。代码在org.apache.dubbo.config.spring.extension.SpringExtensionFactory。这个类做的事情主要是移除之前注册的shutdown hook，然后注册一个关注spring context关闭事件的listener。spring在关闭时会先执行这个listener，再执行容器的销毁。

理论上通过上述手段，应该不会出现停机不平滑的问题。但还是在实际中碰到了。可能的原因是dubbo销毁注册之后，还有请求正在执行。想到的解决方法是销毁注册后，sleep一段时间。代码如下：

public void doDestroy() {
    if (!destroyed.compareAndSet(false, true)) {
        return;
    }
    // destroy all the registries
    AbstractRegistryFactory.destroyAll();
    try {
        Thread.sleep(2000L);
    } catch (InterruptedException ignored) {
    }
    // destroy all the protocols
    destroyProtocols();
}