perf: 将热点代码中的replace替换为fastReplace。非热点代码replace保持不变

Author: meethigher

src/main/java/top/meethigher/proxy/http/ReverseHttpProxy.java

@@ -14,6 +14,8 @@
 import java.util.*;
 import java.util.concurrent.ThreadLocalRandom;
 
+import static top.meethigher.proxy.http.UrlParser.fastReplace;
+
 /**
  * 基于Vertx实现的HTTP反向代理
  *
@@ -703,7 +705,9 @@ protected String getProxyUrl(RoutingContext ctx, HttpServerRequest serverReq, Ht
 
         // 在vertx中，uri表示hostPort后面带有参数的地址。而这里的uri表示不带有参数的地址。
         final String uri = serverReq.path();
-        final String params = serverReq.uri().replace(uri, "");
+
+        //final String params = serverReq.uri().replace(uri, "");
+        final String params = fastReplace(serverReq.uri(), uri, "");
 
 
         // 若不是多级匹配，则直接代理到目标地址。注意要带上请求参数
@@ -715,7 +719,9 @@ protected String getProxyUrl(RoutingContext ctx, HttpServerRequest serverReq, Ht
         if (matchedUri.endsWith("/")) {
             matchedUri = matchedUri.substring(0, matchedUri.length() - 1);
         }
-        String suffixUri = uri.replace(matchedUri, "");
+
+        //String suffixUri = uri.replace(matchedUri, "");
+        String suffixUri = fastReplace(uri, matchedUri, "");
 
         // 代理路径尾部与用户初始请求保持一致
         if (uri.endsWith("/") && !suffixUri.endsWith("/")) {

src/main/java/top/meethigher/proxy/http/UrlParser.java

@@ -4,6 +4,25 @@
 
 public class UrlParser {
 
+
+    /**
+     * 在压测时，性能比String.replace略优
+     */
+    public static String fastReplace(String text, String search, String replacement) {
+        if (text == null || search == null || replacement == null || search.isEmpty()) {
+            return text;
+        }
+        StringBuilder result = new StringBuilder();
+        int start = 0, index;
+        while ((index = text.indexOf(search, start)) >= 0) {
+            result.append(text, start, index).append(replacement);
+            start = index + search.length();
+        }
+        result.append(text.substring(start));
+        return result.toString();
+    }
+
+
     /**
      * 拼接两个uri。不要求uri以"/"结尾
      */

src/test/java/top/meethigher/proxy/http/UrlParserTest.java

@@ -0,0 +1,104 @@
+package top.meethigher.proxy.http;
+
+import org.junit.Test;
+
+public class UrlParserTest {
+
+
+    @Test
+    public void fastReplaceAgg() {
+
+        String origin = "http://10.0.0.1:8080";
+        String target = "https://meethigher.top";
+
+        // 函数进行预热
+        int preheatNumber = 2000;
+
+        int total = 5000000;
+
+        // 预热
+        for (int i = 0; i < 2; i++) {
+            if (i == 0) {
+                for (int j = 0; j < preheatNumber; j++) {
+                    String replace = origin.replace(origin, target);
+                }
+            } else {
+                for (int j = 0; j < preheatNumber; j++) {
+                    String replace = UrlParser.fastReplace(origin, origin, target);
+                }
+            }
+        }
+
+        // 实际耗时记录
+        for (int i = 0; i < 2; i++) {
+            long start = System.currentTimeMillis();
+            if (i == 0) {
+                for (int j = 0; j < total; j++) {
+                    String replace = origin.replace(origin, target);
+                }
+            } else {
+                for (int j = 0; j < total; j++) {
+                    String replace = UrlParser.fastReplace(origin, origin, target);
+                }
+            }
+            long end = System.currentTimeMillis();
+            System.out.println("方案 " + i + " 耗时: " + (end - start) + " ms");
+        }
+
+    }
+
+
+    @Test
+    public void fastReplace1() {
+
+        String origin = "http://10.0.0.1:8080";
+        String target = "https://meethigher.top";
+
+        // 函数进行预热
+        int preheatNumber = 2000;
+
+        int total = 5000000;
+        for (int i = 0; i < preheatNumber; i++) {
+            String replace = origin.replace(origin, target);
+        }
+
+        long start = System.currentTimeMillis();
+        for (int j = 0; j < total; j++) {
+            String replace = origin.replace(origin, target);
+        }
+        long end = System.currentTimeMillis();
+        System.out.println("耗时: " + (end - start) + " ms");
+
+    }
+
+
+    @Test
+    public void fastReplace2() {
+
+        String origin = "http://10.0.0.1:8080";
+        String target = "https://meethigher.top";
+
+        // 函数进行预热
+        int preheatNumber = 2000;
+
+        int total = 5000000;
+        for (int i = 0; i < preheatNumber; i++) {
+            String replace = UrlParser.fastReplace(origin, origin, target);
+        }
+
+        long start = System.currentTimeMillis();
+        for (int j = 0; j < total; j++) {
+            String replace = UrlParser.fastReplace(origin, origin, target);
+        }
+        long end = System.currentTimeMillis();
+        System.out.println("耗时: " + (end - start) + " ms");
+
+    }
+
+
+    @Test
+    public void fastReplace() {
+        String str = "hello, world";
+        System.out.println(UrlParser.fastReplace(str, ",", "!"));
+    }
+}