优化拦截器

wushuo894 · Sep 25, 2024 · 5e231b8 · 5e231b8
1 parent d41d633
commit 5e231b8
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/src/main/java/com/emby/util/ServerUtil.java b/src/main/java/com/emby/util/ServerUtil.java
@@ -54,11 +54,12 @@ public static SimpleServer create(String... args) {
             }
             Object action = ReflectUtil.newInstanceIfPossible(aClass);
             String urlPath = "/api" + path.value();
-            server.addFilter(new HttpFilter() {
-                @Override
-                public void doFilter(HttpServerRequest req, HttpServerResponse res, Filter.Chain chain) throws IOException {
-                    Config config = ConfigUtil.CONFIG;
-                    Boolean isInnerIP = config.getIsInnerIP();
+            server.addFilter((req, res, chain) -> {
+                REQUEST.set(req);
+                RESPONSE.set(res);
+                Config config = ConfigUtil.CONFIG;
+                Boolean isInnerIP = config.getIsInnerIP();
+                try {
                     String ip = getIp();
                     if (isInnerIP) {
                         if (!PatternPool.IPV4.matcher(ip).matches()) {
@@ -71,6 +72,9 @@ public void doFilter(HttpServerRequest req, HttpServerResponse res, Filter.Chain
                         }
                     }
                     chain.doFilter(req.getHttpExchange());
+                } finally {
+                    REQUEST.remove();
+                    RESPONSE.remove();
                 }
             });
             server.addAction(urlPath, new BaseAction() {
@@ -79,8 +83,6 @@ public void doFilter(HttpServerRequest req, HttpServerResponse res, Filter.Chain
                 @Override
                 public void doAction(HttpServerRequest req, HttpServerResponse res) {
                     try {
-                        REQUEST.set(req);
-                        RESPONSE.set(res);
                         BaseAction baseAction = (BaseAction) action;
                         baseAction.doAction(req, res);
                     } catch (Exception e) {
@@ -91,9 +93,6 @@ public void doAction(HttpServerRequest req, HttpServerResponse res) {
                             log.error("{} {}", urlPath, e.getMessage());
                             log.debug(e);
                         }
-                    } finally {
-                        REQUEST.remove();
-                        RESPONSE.remove();
                     }
                 }
             });