添加灰度热开关

Author: saleson

spring-cloud-gray-client/src/main/java/cn/springcloud/gray/DefaultGrayManager.java

@@ -52,6 +52,12 @@ public void shutdown() {
         updateTimer.cancel();
     }
 
+
+    @Override
+    public boolean hasGray(String serviceId) {
+        return GrayClientHolder.getGraySwitcher().judge() && super.hasGray(serviceId);
+    }
+
     public void openForWork() {
         if (getGrayInformationClient() != null) {
             log.info("拉取灰度列表");

spring-cloud-gray-client/src/main/java/cn/springcloud/gray/GrayClientHolder.java

@@ -1,5 +1,6 @@
 package cn.springcloud.gray;
 
+import cn.springcloud.gray.client.switcher.GraySwitcher;
 import cn.springcloud.gray.request.LocalStorageLifeCycle;
 import cn.springcloud.gray.request.RequestLocalStorage;
 import cn.springcloud.gray.servernode.ServerExplainer;
@@ -12,6 +13,7 @@ public class GrayClientHolder {
     private static LocalStorageLifeCycle localStorageLifeCycle;
     private static ServerExplainer<?> serverExplainer;
     private static ServerListProcessor<?> serverListProcessor;
+    private static GraySwitcher graySwitcher = new GraySwitcher.DefaultGraySwitcher();
 
     public static GrayManager getGrayManager() {
         return grayManager;
@@ -52,4 +54,12 @@ public static void setServerListProcessor(ServerListProcessor<?> serverListProce
     public static <SERVER> ServerListProcessor<SERVER> getServereListProcessor() {
         return (ServerListProcessor<SERVER>) serverListProcessor;
     }
+
+    public static GraySwitcher getGraySwitcher() {
+        return graySwitcher;
+    }
+
+    public static void setGraySwitcher(GraySwitcher graySwitcher) {
+        GrayClientHolder.graySwitcher = graySwitcher;
+    }
 }

spring-cloud-gray-client/src/main/java/cn/springcloud/gray/GrayClientInitializer.java

@@ -1,5 +1,6 @@
 package cn.springcloud.gray;
 
+import cn.springcloud.gray.client.switcher.GraySwitcher;
 import cn.springcloud.gray.request.LocalStorageLifeCycle;
 import cn.springcloud.gray.request.RequestLocalStorage;
 import cn.springcloud.gray.servernode.ServerExplainer;
@@ -24,6 +25,8 @@ public void afterPropertiesSet() throws Exception {
         GrayClientHolder.setServerExplainer(getBean("serverExplainer", ServerExplainer.class));
         GrayClientHolder.setServerListProcessor(
                 getBean("serverListProcessor", ServerListProcessor.class, new ServerListProcessor.Default()));
+        GrayClientHolder.setGraySwitcher(getBean(
+                "graySwitcher", GraySwitcher.class, new GraySwitcher.DefaultGraySwitcher()));
 
         initGrayManagerRequestInterceptors();
 

spring-cloud-gray-client/src/main/java/cn/springcloud/gray/client/config/GrayClientAutoConfiguration.java

@@ -4,6 +4,8 @@
 import cn.springcloud.gray.cache.CaffeineCache;
 import cn.springcloud.gray.client.GrayClientEnrollInitializingDestroyBean;
 import cn.springcloud.gray.client.config.properties.*;
+import cn.springcloud.gray.client.switcher.EnbGraySwitcher;
+import cn.springcloud.gray.client.switcher.GraySwitcher;
 import cn.springcloud.gray.communication.InformationClient;
 import cn.springcloud.gray.decision.GrayDecision;
 import cn.springcloud.gray.decision.GrayDecisionFactoryKeeper;
@@ -24,7 +26,8 @@
 
 @Configuration
 @EnableConfigurationProperties(
-        {GrayClientProperties.class,
+        {GrayProperties.class,
+                GrayClientProperties.class,
                 GrayServerProperties.class,
                 GrayRequestProperties.class,
                 GrayLoadProperties.class,
@@ -38,6 +41,8 @@ public class GrayClientAutoConfiguration {
 
     @Autowired
     private GrayClientProperties grayClientProperties;
+    @Autowired
+    private GrayProperties grayProperties;
 
 
     @Bean
@@ -63,6 +68,12 @@ public GrayManager grayManager(
     }
 
 
+    @Bean
+    @ConditionalOnMissingBean
+    public GraySwitcher graySwitcher() {
+        return new EnbGraySwitcher(grayProperties);
+    }
+
     @Bean
     @ConditionalOnProperty(value = "gray.client.instance.grayEnroll")
     public GrayClientEnrollInitializingDestroyBean grayClientEnrollInitializingDestroyBean(

spring-cloud-gray-client/src/main/java/cn/springcloud/gray/client/config/properties/GrayProperties.java

@@ -0,0 +1,18 @@
+package cn.springcloud.gray.client.config.properties;
+
+import org.springframework.boot.context.properties.ConfigurationProperties;
+
+@ConfigurationProperties("gray")
+public class GrayProperties {
+
+    private boolean enabled;
+
+
+    public boolean isEnabled() {
+        return enabled;
+    }
+
+    public void setEnabled(boolean enabled) {
+        this.enabled = enabled;
+    }
+}

spring-cloud-gray-client/src/main/java/cn/springcloud/gray/client/switcher/EnbGraySwitcher.java

@@ -0,0 +1,17 @@
+package cn.springcloud.gray.client.switcher;
+
+import cn.springcloud.gray.client.config.properties.GrayProperties;
+
+public class EnbGraySwitcher implements GraySwitcher {
+
+    private GrayProperties grayProperties;
+
+    public EnbGraySwitcher(GrayProperties grayProperties) {
+        this.grayProperties = grayProperties;
+    }
+
+    @Override
+    public boolean judge() {
+        return grayProperties.isEnabled();
+    }
+}

spring-cloud-gray-client/src/main/java/cn/springcloud/gray/client/switcher/GraySwitcher.java

@@ -0,0 +1,15 @@
+package cn.springcloud.gray.client.switcher;
+
+public interface GraySwitcher {
+
+    boolean judge();
+
+
+    public static class DefaultGraySwitcher implements GraySwitcher {
+
+        @Override
+        public boolean judge() {
+            return true;
+        }
+    }
+}