限流降级

Author: robberphex

mse-simple-demo/B/src/main/java/com/alibabacloud/mse/demo/b/BController.java

@@ -83,7 +83,7 @@ public String params(HttpServletRequest request,@PathVariable("hot") String hot)
 
     @GetMapping("/isolate")
     public String isolate(HttpServletRequest request) throws ExecutionException, InterruptedException {
-        long sleepTime = 20 + RANDOM.nextInt(5);
+        long sleepTime = 500 + RANDOM.nextInt(5);
         silentSleep(sleepTime);
         return "B" + serviceTag + "[" + inetUtils.findFirstNonLoopbackAddress().getHostAddress() + "]" + " -> " + sleepTime;
     }

mse-simple-demo/gateway/src/main/java/com/alibabacloud/mse/demo/DemoController.java

@@ -15,6 +15,7 @@
 
 import javax.annotation.PostConstruct;
 import javax.servlet.http.HttpServletRequest;
+import java.time.LocalDateTime;
 import java.util.concurrent.Executors;
 import java.util.concurrent.ScheduledExecutorService;
 import java.util.concurrent.ThreadFactory;
@@ -36,9 +37,8 @@ public class DemoController {
     @Value("${background.color:white}")
     private String backgroundColor;
 
-    private static final ScheduledExecutorService FLOW_EXECUTOR = Executors.newScheduledThreadPool(6,
+    private static final ScheduledExecutorService FLOW_EXECUTOR = Executors.newScheduledThreadPool(16,
             new ThreadFactory() {
-
                 @Override
                 public Thread newThread(Runnable r) {
                     Thread thread = new Thread(r);
@@ -106,12 +106,10 @@ public void run() {
                 }
             }, 100, 10 * 1000000 / qps, TimeUnit.MICROSECONDS);
 
-
-
+            // 限流降级的流量发起
             FLOW_EXECUTOR.scheduleAtFixedRate(new Runnable() {
                 @Override
                 public void run() {
-
                     try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
                         HttpGet httpGet = new HttpGet("http://localhost:20000/A/flow");
                         httpClient.execute(httpGet);
@@ -127,34 +125,6 @@ public void run() {
                     } catch (Exception ignore) {
                     }
 
-                    try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
-                        HttpGet httpGet = new HttpGet("http://localhost:20000/A/isolate");
-                        httpClient.execute(httpGet);
-
-                    } catch (Exception ignore) {
-                    }
-                    try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
-                        HttpGet httpGet = new HttpGet("http://localhost:20000/A/isolate");
-                        httpGet.addHeader("x-mse-tag", "gray");
-                        httpClient.execute(httpGet);
-
-                    } catch (Exception ignore) {
-                    }
-
-                    try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
-                        HttpGet httpGet = new HttpGet("http://localhost:20000/A/params/hot");
-                        httpClient.execute(httpGet);
-
-                    } catch (Exception ignore) {
-                    }
-                    try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
-                        HttpGet httpGet = new HttpGet("http://localhost:20000/A/params/hot");
-                        httpGet.addHeader("x-mse-tag", "gray");
-                        httpClient.execute(httpGet);
-
-                    } catch (Exception ignore) {
-                    }
-
                     try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
                         HttpGet httpGet = new HttpGet("http://localhost:20000/A/dubbo-flow");
                         httpClient.execute(httpGet);
@@ -183,20 +153,6 @@ public void run() {
                     } catch (Exception ignore) {
                     }
 
-                    try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
-                        HttpGet httpGet = new HttpGet("http://localhost:20000/A/dubbo-params/hot");
-                        httpClient.execute(httpGet);
-
-                    } catch (Exception ignore) {
-                    }
-                    try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
-                        HttpGet httpGet = new HttpGet("http://localhost:20000/A/dubbo-params/hot");
-                        httpGet.addHeader("x-mse-tag", "gray");
-                        httpClient.execute(httpGet);
-
-                    } catch (Exception ignore) {
-                    }
-
 //
 //                    try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
 //                        HttpGet httpGet = new HttpGet("http://localhost:20000/B/flow-c");
@@ -239,12 +195,37 @@ public void run() {
 //
 //                    } catch (Exception ignore) {
 //                    }
-
-
                 }
             }, 100, 1000000 / qps, TimeUnit.MICROSECONDS);
 
-
+            // region 热点限流
+            FLOW_EXECUTOR.scheduleAtFixedRate(new Runnable() {
+                @Override
+                public void run() {
+                    try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
+                        HttpGet httpGet = new HttpGet("http://localhost:20000/A/dubbo-params/hot");
+                        httpClient.execute(httpGet);
+                    } catch (Exception ignore) {
+                    }
+                }
+            }, 100, 1000000 / qps, TimeUnit.MICROSECONDS);
+            // endregion 热点限流
+
+            // region 隔离规则
+            for (int i = 0; i < 8; i++) {
+                int finalI = i;
+                FLOW_EXECUTOR.scheduleAtFixedRate(new Runnable() {
+                    @Override
+                    public void run() {
+                        try (CloseableHttpClient httpClient = HttpClientBuilder.create().build()) {
+                            HttpGet httpGet = new HttpGet("http://localhost:20000/A/isolate?i_id=" + finalI);
+                            httpClient.execute(httpGet);
+                        } catch (Exception ignore) {
+                        }
+                    }
+                }, 100, 1000000 / qps, TimeUnit.MICROSECONDS);
+            }
+            // endregion 隔离规则
         }