fix(server): support put/delete & when 512 length then log cut

apache · Jun 15, 2024 · 071943c · 071943c
1 parent 868adcf
commit 071943c
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/hugegraph-server/hugegraph-api/src/main/java/org/apache/hugegraph/api/filter/PathFilter.java b/hugegraph-server/hugegraph-api/src/main/java/org/apache/hugegraph/api/filter/PathFilter.java
@@ -39,25 +39,30 @@ public class PathFilter implements ContainerRequestFilter {
 
     public static final String REQUEST_TIME = "request_time";
     public static final String REQUEST_PARAMS_JSON = "request_params_json";
+    public static final int MAX_SLOW_LOG_BODY_LENGTH = 512;
 
     @Override
     public void filter(ContainerRequestContext context) throws IOException {
         long startTime = System.currentTimeMillis();
 
         context.setProperty(REQUEST_TIME, startTime);
 
-        recordRequestJson(context);
+        collectRequestParams(context);
     }
 
-    private void recordRequestJson(ContainerRequestContext context) throws IOException {
+    private void collectRequestParams(ContainerRequestContext context) throws IOException {
         String method = context.getMethod();
-        if (method.equals(HttpMethod.POST)) {
+        if (method.equals(HttpMethod.POST) || method.equals(HttpMethod.PUT) ||
+            method.equals(HttpMethod.DELETE)) {
             BufferedInputStream bufferedStream = new BufferedInputStream(context.getEntityStream());
 
             bufferedStream.mark(Integer.MAX_VALUE);
+            String body = IOUtils.toString(bufferedStream,
+                                           Charsets.toCharset(CHARSET));
+            body = body.length() > MAX_SLOW_LOG_BODY_LENGTH ?
+                   body.substring(0, MAX_SLOW_LOG_BODY_LENGTH) : body;
 
-            context.setProperty(REQUEST_PARAMS_JSON,
-                                IOUtils.toString(bufferedStream, Charsets.toCharset(CHARSET)));
+            context.setProperty(REQUEST_PARAMS_JSON, body);
 
             bufferedStream.reset();