通过Apollo SPI机制实现默认环境变量配置支持

前言

最近公司项目要接入配置中心，后来调研下来决定使用apollo，但是在使用的时候发现有
个小细节特别难受，apollo不支持通过项目代码配置默认的environment，官网文档如下：

这样会导致代码拉下来不能直接启动服务，还需要通过上面的三种方式之一来指定environment，非常的麻烦。

进行魔改

我想着是能不能项目代码的配置文件里配置一个默认的environment，这样在没有外部指定environment的情况下就默认使用配置文件里的。

通过跟踪源码可以看到加载environment是通过DefaultServerProvider类的initialize方法进行加载的，然后通过DefaultProviderManager类中的register方法进行注册，时序图：

本来理论上来说魔改DefaultServerProvider这个类就可以实现目的了，然而这源码里没有为接口ServerProvider实现SPI机制，是直接new出来的：

那么只能从DefaultProviderManager类进行入手了，这个接口是通过SPI机制去加载的实现类：

现在唯一的入口点就是register方法了，通过这个方法可以把注册的DefaultServerProvider魔改掉，这个时候设计模式就派上用场了，我直接进行一个装饰器模式的秀：

public class HookServerProviderManager extends DefaultProviderManager {

    @Override
    public synchronized void register(Provider provider) {
        // 如果是注册DefaultServerProvider的时候替换成装饰之后的类
        if (provider instanceof DefaultServerProvider) {
            super.register(new DefaultServerProviderWrapper((DefaultServerProvider) provider));
        } else {
            super.register(provider);
        }
    }
}

public class DefaultServerProviderWrapper implements ServerProvider {

    private DefaultServerProvider serverProvider;

    private String defaultEnv;

    public DefaultServerProviderWrapper(DefaultServerProvider serverProvider) {
        this.serverProvider = serverProvider;

        // 如果没有读取到环境变量，则加载META-INF/app.properties文件中的default.env变量
        if (serverProvider.getEnvType() == null) {
            Properties prop = new Properties();
            try (InputStream inputStream = Thread.currentThread().getContextClassLoader()
                    .getResourceAsStream(DefaultApplicationProvider.APP_PROPERTIES_CLASSPATH.substring(1))) {
                prop.load(inputStream);
                this.defaultEnv = prop.getProperty("default.env");
            } catch (Exception e) {
                log.warn("load app.properties fail", e);
            }
        }
    }

    @Override
    public String getEnvType() {
        String originEnv = serverProvider.getEnvType();
        return originEnv != null ? originEnv : defaultEnv;
    }

    @Override
    public boolean isEnvTypeSet() {
        return getEnvType() != null;
    }

    @Override
    public String getDataCenter() {
        return serverProvider.getDataCenter();
    }

    @Override
    public boolean isDataCenterSet() {
        return serverProvider.isDataCenterSet();
    }

    @Override
    public void initialize(InputStream in) throws IOException {
        serverProvider.initialize(in);
    }

    @Override
    public Class<? extends Provider> getType() {
        return serverProvider.getType();
    }

    @Override
    public String getProperty(String name, String defaultValue) {
        return serverProvider.getProperty(name, defaultValue);
    }

    @Override
    public void initialize() {
        serverProvider.initialize();
    }
}