修复 UA 可能为空的问题

PBH-BTN · Oct 7, 2024 · 844d9c3 · 844d9c3
1 parent 03406b0
commit 844d9c3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/com/ghostchu/btn/sparkle/module/tracker/TrackerController.java b/src/main/java/com/ghostchu/btn/sparkle/module/tracker/TrackerController.java
@@ -178,7 +178,7 @@ public byte[] announce() {
                 }});
             }
         }};
-        auditService.log(req, "TRACKER_ANNOUNCE", true, Map.of("hash", infoHash, "user-agent", req.getHeader("User-Agent")));
+        auditService.log(req, "TRACKER_ANNOUNCE", true, Map.of("hash", infoHash, "user-agent", ua(req)));
         return BencodeUtil.INSTANCE.encode(map);
     }
 
@@ -198,7 +198,7 @@ public ResponseEntity<byte[]> scrape() {
         }
         map.put("files", files);
         map.put("external ip", ip(req));
-        auditService.log(req, "TRACKER_SCRAPE", true, Map.of("hash", infoHashes, "user-agent", req.getHeader("User-Agent")));
+        auditService.log(req, "TRACKER_SCRAPE", true, Map.of("hash", infoHashes, "user-agent", ua(req)));
         return ResponseEntity.ok(BencodeUtil.INSTANCE.encode(map));
     }