dubbo优雅停机问题修复

问题总结

1.服务端
问题出线程池是否为空的判断环节，需要获取线程池对象，检查其中是否有未完成的任务，并等待任务全部完成后关闭线程池，但原生代码存在BUG，获取到的线程池对象为null。

修改方法

改为使用静态缓存存取线程池对象，保证停机时能正常获取线程池对象。

相关源码

com.alibaba.dubbo.remoting.transport.AbstractServer.AbstractServer(URL, ChannelHandler)

官方commit

https://github.com/alibaba/dubbo/commit/8a9392d58e6833cb5fe1f3318636dca3d77048d8
这样修改过后，服务端可以正常优雅停机

2.消费端
问题出在上图步骤3的环节，检查是否有未完成调用的源码存在BUG。简单来说，这里使用了一个静态Map作为缓存，每当有新的服务请求时，放一个NettyClient对象到缓存里，调用结束再将NettyClient对象移出缓存。因此通过检查该缓存，就能判断是否有未结束的服务调用。检查缓存时使用Map.containsValue(Object)，目前的BUG在于传入的对象错误，只要改为传入NettyClient即可修复该问题。

代码位置com.alibaba.dubbo.remoting.exchange.support.header.HeaderExchangeChannel.close(int)

官方commit

https://github.com/alibaba/dubbo/commit/8a9392d58e6833cb5fe1f3318636dca3d77048d8
在对官方commit的代码进行移植后，发现消费端还是无法优雅停机，查找问题发现是client.close()这个方法里没有传入超时时间，导致无法执行到DefaultFuture.hasFuture(channel)这一步，client.close()一共有2处需要改动，
分别在
com.alibaba.dubbo.rpc.protocol.dubbo.DubboProtocol.destroy()com.alibaba.dubbo.rpc.protocol.dubbo.DubboInvoker.destroy()
需要传入参数ConfigUtils.getServerShutdownTimeout()

 //client.close();
   client.close(ConfigUtils.getServerShutdownTimeout());
这样经过测试后消费端可以正常优雅停机

3.消费端的测试方法
由于消费端需要去调服务，触发正常停机需要程序正常退出，由于水平有限的原因，一直没想好该怎么正常退出程序，后来请教别人之后，他给出了重新起一个线程，然后退出这个线程的方法

package com.alibaba.dubbo.demo.main;

import com.alibaba.dubbo.common.Version;
import com.alibaba.dubbo.common.logger.Logger;
import com.alibaba.dubbo.common.logger.LoggerFactory;
import com.alibaba.dubbo.demo.DemoService;
import org.springframework.context.support.ClassPathXmlApplicationContext;

import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;

public class DemoConsumer {
    private static final Logger logger = LoggerFactory.getLogger(DemoConsumer.class);
    public static void main(String[] args) throws IOException, InterruptedException {
        ClassPathXmlApplicationContext context = new ClassPathXmlApplicationContext("META-INF/spring/dubbo-demo-consumer.xml");
        context.start();
        final DemoService demoService = (DemoService) context.getBean("demoService");
        //这里起一个线程，从这里退出
        Thread t = new Thread(new Runnable() {
            public void run() {
                try {
                    System.in.read();
                    System.exit(0);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        });
        t.start();

        while (true) {
            try {
                logger.info("-------------- start to call --------------");
                String hello = demoService.sayHello("world" + 1000000);
                System.out.println("[" + new SimpleDateFormat("HH:mm:ss").format(new Date()) + "] " + hello);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

    }


}