feat: 添加加载静态资源功能

Author: meethigher

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

@@ -7,6 +7,7 @@
 import io.vertx.ext.web.Route;
 import io.vertx.ext.web.Router;
 import io.vertx.ext.web.RoutingContext;
+import io.vertx.ext.web.handler.StaticHandler;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -162,6 +163,11 @@ public class ReverseHttpProxy {
      */
     protected static final String INTERNAL_STATUS_CODE = "INTERNAL_STATUS_CODE";
 
+    /**
+     * 静态资源前缀
+     */
+    protected static final String STATIC = "static:";
+
     protected static final char[] ID_CHARACTERS = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ".toCharArray();
 
 
@@ -329,16 +335,20 @@ public void stop() {
     }
 
     public ReverseHttpProxy addRoute(ProxyRoute proxyRoute) {
-        return addRoute(proxyRoute, null);
+        return addRoute(proxyRoute, null, true);
     }
 
+    public ReverseHttpProxy addRoute(ProxyRoute proxyRoute, Integer order) {
+        return addRoute(proxyRoute, order, true);
+    }
 
     /**
      * order越小，优先级越高
      */
     public ReverseHttpProxy addRoute(
             ProxyRoute proxyRoute,
-            Integer order
+            Integer order,
+            boolean printLog
     ) {
         Route route = router.route(proxyRoute.getSourceUrl()).setName(proxyRoute.getName());
         if (order != null) {
@@ -348,8 +358,20 @@ public ReverseHttpProxy addRoute(
         for (String key : map.keySet()) {
             setRouteMetadata(route, key, map.get(key));
         }
-        route.handler(routingContextHandler(httpClient));
-        jsonLog(proxyRoute);
+        String targetUrl = proxyRoute.getTargetUrl();
+        if (targetUrl.startsWith(STATIC)) {
+            String staticPath = targetUrl.replace(STATIC, "");
+            StaticHandler staticHandler = StaticHandler.create(staticPath)
+                    .setDirectoryListing(false)
+                    .setAlwaysAsyncFS(true)
+                    .setIndexPage("index.html");
+            route.handler(staticHandler);
+        } else {
+            route.handler(routingContextHandler(httpClient));
+        }
+        if (printLog) {
+            jsonLog(proxyRoute);
+        }
         return this;
     }
 

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

@@ -43,6 +43,24 @@ public void testAllMethod() throws Exception {
         proxy.stop();
     }
 
+    @Test
+    public void testStatic() throws Exception {
+        ReverseHttpProxy proxy = ReverseHttpProxy.create(
+                vertx
+        );
+        proxy.port(8080);
+        proxy.start();
+        ProxyRoute proxyRoute = new ProxyRoute();
+        proxyRoute.setHttpKeepAlive(false);
+//        proxyRoute.setSourceUrl("/*");
+//        proxyRoute.setTargetUrl("static:D:/Desktop");
+        proxyRoute.setSourceUrl("/blog/*");
+        proxyRoute.setTargetUrl("static:D:/3Develop/www/hexoBlog/blog/public");
+        proxy.addRoute(proxyRoute);
+        TimeUnit.HOURS.sleep(1);
+        proxy.stop();
+    }
+
 
     /**
      * 用于测试省略端口时的反代场景