把 apollo client 打进 common

commons-spring/pom.xml

@@ -12,7 +12,6 @@
     <artifactId>commons-spring</artifactId>
 
     <dependencies>
-        <!--健康检查需要-->
         <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-actuator</artifactId>

pom.xml

@@ -51,16 +51,11 @@
         <module>idgenerator-spring-boot-starter</module>
         <module>enoch-agent-spring-boot-starter</module>
         <module>qg-application-dependencies</module>
+        <module>qg-apollo-starter</module>
     </modules>
 
 
     <dependencies>
-        <!--所有工程基础依赖. apollo 配置中心-->
-        <dependency>
-            <groupId>com.ctrip.framework.apollo</groupId>
-            <artifactId>apollo-client</artifactId>
-            <version>${apollo.client.version}</version>
-        </dependency>
         <!--所有工程基础依赖. lombok -->
         <dependency>
             <groupId>org.projectlombok</groupId>
@@ -116,6 +111,16 @@
                 <artifactId>enoch-agent-spring-boot-starter</artifactId>
                 <version>${revision}</version>
             </dependency>
+            <dependency>
+                <groupId>cn.quantgroup</groupId>
+                <artifactId>qg-apollo-starter</artifactId>
+                <version>${revision}</version>
+            </dependency>
+            <dependency>
+                <groupId>com.ctrip.framework.apollo</groupId>
+                <artifactId>apollo-client</artifactId>
+                <version>${apollo.client.version}</version>
+            </dependency>
 
             <dependency>
                 <groupId>org.aspectj</groupId>

qg-apollo-starter/pom.xml

+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <parent>
+        <artifactId>commons-parent</artifactId>
+        <groupId>cn.quantgroup</groupId>
+        <version>${revision}</version>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+    <artifactId>qg-apollo-starter</artifactId>
+    <dependencies>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>com.ctrip.framework.apollo</groupId>
+            <artifactId>apollo-client</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-autoconfigure</artifactId>
+            <optional>true</optional>
+        </dependency>
+    </dependencies>
+</project>
\ No newline at end of file

qg-apollo-starter/src/main/java/cn/quantgroup/apollopatch/QGApolloInjector.java

+package cn.quantgroup.apollopatch;
+
+import com.ctrip.framework.apollo.exceptions.ApolloConfigException;
+import com.ctrip.framework.apollo.internals.ConfigManager;
+import com.ctrip.framework.apollo.internals.ConfigServiceLocator;
+import com.ctrip.framework.apollo.internals.Injector;
+import com.ctrip.framework.apollo.internals.RemoteConfigLongPollService;
+import com.ctrip.framework.apollo.spi.*;
+import com.ctrip.framework.apollo.tracer.Tracer;
+import com.ctrip.framework.apollo.util.ConfigUtil;
+import com.ctrip.framework.apollo.util.http.HttpUtil;
+import com.ctrip.framework.apollo.util.yaml.YamlParser;
+import com.google.inject.AbstractModule;
+import com.google.inject.Guice;
+import com.google.inject.Singleton;
+import lombok.extern.slf4j.Slf4j;
+
+@Slf4j
+public class QGApolloInjector implements Injector {
+
+    private com.google.inject.Injector m_injector;
+
+    public QGApolloInjector() {
+        try {
+            m_injector = Guice.createInjector(new QGApolloInjector.ApolloModule());
+        } catch (Throwable ex) {
+            ApolloConfigException exception = new ApolloConfigException("Unable to initialize Guice Injector!", ex);
+            Tracer.logError(exception);
+            throw exception;
+        }
+    }
+
+    @Override
+    public <T> T getInstance(Class<T> clazz) {
+        try {
+            return m_injector.getInstance(clazz);
+        } catch (Throwable ex) {
+            Tracer.logError(ex);
+            throw new ApolloConfigException(
+                    String.format("Unable to load instance for %s!", clazz.getName()), ex);
+        }
+    }
+
+    @Override
+    public <T> T getInstance(Class<T> clazz, String name) {
+        //Guice does not support get instance by type and name
+        return null;
+    }
+
+    private static class ApolloModule extends AbstractModule {
+        @Override
+        protected void configure() {
+            bind(ConfigManager.class).to(QGConfigManager.class).in(Singleton.class);
+            bind(ConfigFactoryManager.class).to(DefaultConfigFactoryManager.class).in(Singleton.class);
+            bind(ConfigRegistry.class).to(DefaultConfigRegistry.class).in(Singleton.class);
+            bind(ConfigFactory.class).to(DefaultConfigFactory.class).in(Singleton.class);
+            bind(ConfigUtil.class).in(Singleton.class);
+            bind(HttpUtil.class).in(Singleton.class);
+            bind(ConfigServiceLocator.class).in(Singleton.class);
+            bind(RemoteConfigLongPollService.class).in(Singleton.class);
+            bind(YamlParser.class).in(Singleton.class);
+        }
+    }
+}

qg-apollo-starter/src/main/java/cn/quantgroup/apollopatch/QGConfigManager.java

+package cn.quantgroup.apollopatch;
+
+import com.ctrip.framework.apollo.build.ApolloInjector;
+import com.ctrip.framework.apollo.core.enums.Env;
+import com.ctrip.framework.apollo.internals.DefaultConfigManager;
+import com.ctrip.framework.apollo.util.http.HttpRequest;
+import com.ctrip.framework.apollo.util.http.HttpResponse;
+import com.ctrip.framework.apollo.util.http.HttpUtil;
+import com.ctrip.framework.foundation.Foundation;
+import lombok.ToString;
+import lombok.extern.slf4j.Slf4j;
+
+import java.util.Map;
+import java.util.Properties;
+
+@Slf4j
+public class QGConfigManager extends DefaultConfigManager {
+    public QGConfigManager() {
+        super();
+        boolean isPro = Env.PRO.name().equalsIgnoreCase(Foundation.server().getEnvType());
+        //如果是生产环境, 我也啥都不干. 太吓人了.
+        if (isPro) {
+            log.info("哇, 生产环境. 配置中心静悄悄. 什么都不敢做.");
+            return;
+        }
+
+        String namespace = System.getProperty("NAMESPACE");
+        if (namespace == null) {
+            log.info("你好像没有配置 NAMESPACE 哦?你不打算连接到 kubernetes 内部么?");
+            return;
+        }
+
+        String eosHost = System.getProperty("eos_server_host", "http://eos.quantgroups.com/");
+        HttpUtil httpUtil = ApolloInjector.getInstance(HttpUtil.class);
+        HttpRequest httpRequest = new HttpRequest(eosHost + "api/apollo/env_vars?namespace=" + namespace);
+
+        HttpResponse<KubeEnvInfo> mapHttpResponse = httpUtil.doGet(httpRequest, KubeEnvInfo.class);
+        KubeEnvInfo body = mapHttpResponse.getBody();
+        if (body != null && body.success) {
+            log.info("客官请放心, kubernetes 的环境变量已经注入, 您可以放心的在 kubernetes 之外启动你的服务了");
+            Properties properties = System.getProperties();
+            properties.putAll(body.details);
+            return;
+        }
+        log.error("额... 看起来 eos server 有点问题, 返回了false, serverInfo:{} ,body:{}", eosHost, body);
+    }
+
+
+    @ToString
+    private static class KubeEnvInfo {
+        boolean success;
+        Map<String, String> details;
+    }
+}

qg-apollo-starter/src/main/java/cn/quantgroup/apollopatch/QGMetaServerProvider.java

+package cn.quantgroup.apollopatch;
+
+import com.ctrip.framework.apollo.core.enums.Env;
+import com.ctrip.framework.apollo.internals.DefaultMetaServerProvider;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class QGMetaServerProvider extends DefaultMetaServerProvider {
+
+    private Map<Env, String> envMeta = new HashMap<>();
+
+    {
+        envMeta.put(Env.DEV, "http://apollo-dev.quantgroups.com");
+        envMeta.put(Env.PRO, "http://apollo-pro.quantgroups.com");
+    }
+
+    @Override
+    public String getMetaServerAddress(Env targetEnv) {
+        String metaServerAddress = super.getMetaServerAddress(targetEnv);
+        return metaServerAddress == null ? envMeta.get(targetEnv) : metaServerAddress;
+    }
+
+}

qg-apollo-starter/src/main/java/cn/quantgroup/apollopatch/spring/QGSpringApplicationRunListener.java

+package cn.quantgroup.apollopatch.spring;
+
+import com.ctrip.framework.foundation.Foundation;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.SpringApplicationRunListener;
+
+public class QGSpringApplicationRunListener implements SpringApplicationRunListener {
+
+    private final SpringApplication application;
+
+    private final String[] args;
+
+    public QGSpringApplicationRunListener(SpringApplication application, String[] args) {
+        this.application = application;
+        this.args = args;
+
+        System.setProperty("apollo.bootstrap.enabled", "true");
+        String property = Foundation.app().getProperty("namespace", "application");
+        System.setProperty("apollo.bootstrap.namespaces", property);
+        System.setProperty("apollo.bootstrap.eagerLoad.enabled", "true");
+    }
+}

qg-apollo-starter/src/main/resources/META-INF/services/com.ctrip.framework.apollo.core.spi.MetaServerProvider

+cn.quantgroup.apollopatch.QGMetaServerProvider
\ No newline at end of file

qg-apollo-starter/src/main/resources/META-INF/services/com.ctrip.framework.apollo.internals.Injector

+cn.quantgroup.apollopatch.QGApolloInjector
\ No newline at end of file

qg-apollo-starter/src/main/resources/META-INF/spring.factories

+org.springframework.boot.SpringApplicationRunListener=\
+cn.quantgroup.apollopatch.spring.QGSpringApplicationRunListener
\ No newline at end of file