Làm cách nào để lấy mã trạng thái HTTP ra khỏi ServletResponse trong ServletFilter?

Question

Tôi đang cố gắng báo cáo về mọi mã trạng thái HTTP được trả về từ webapp của tôi. Tuy nhiên, mã trạng thái dường như không thể truy cập được thông qua ServletResponse, hoặc thậm chí nếu tôi đưa nó vào một HttpServletResponse. Có cách nào để truy cập vào giá trị này trong ServletFilter không?

Answer 1

Trước tiên, bạn cần phải lưu mã trạng thái vào một nơi có thể truy cập. Các tốt nhất để quấn phản ứng với thực hiện của bạn và giữ nó ở đó:

public class StatusExposingServletResponse extends HttpServletResponseWrapper { 

    private int httpStatus; 

    public StatusExposingServletResponse(HttpServletResponse response) { 
     super(response); 
    } 

    @Override 
    public void sendError(int sc) throws IOException { 
     httpStatus = sc; 
     super.sendError(sc); 
    } 

    @Override 
    public void sendError(int sc, String msg) throws IOException { 
     httpStatus = sc; 
     super.sendError(sc, msg); 
    } 


    @Override 
    public void setStatus(int sc) { 
     httpStatus = sc; 
     super.setStatus(sc); 
    } 

    public int getStatus() { 
     return httpStatus; 
    } 

} 

Để sử dụng wrapper này, bạn cần phải thêm một bộ lọc servlet, là bạn có thể làm báo cáo của bạn:

public class StatusReportingFilter implements Filter { 

    public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain) throws IOException, ServletException { 
     StatusExposingServletResponse response = new StatusExposingServletResponse((HttpServletResponse)res); 
     chain.doFilter(req, response); 
     int status = response.getStatus(); 
     // report 
    } 

    public void init(FilterConfig config) throws ServletException { 
     //empty 
    } 

    public void destroy() { 
     // empty 
    } 

} 

Answer 2

Viết một HttpServletResponseWrapper và ghi đè tất cả các phương thức setStatus(), sendError() và sendRedirect() để ghi lại mọi thứ. Viết một Bộ lọc hoán đổi trình bao bọc của bạn cho đối tượng phản hồi trên mọi yêu cầu.

Answer 3

một điều mất tích từ câu trả lời của David trên là bạn cũng nên ghi đè lên các hình thức khác của sendError:

@Override 
public void sendError(int sc, String msg) throws IOException { 
    httpStatus = sc; 
    super.sendError(sc, msg); 
} 

Answer 4

Kể từ Servlet 3.0, có một HttpServletResponse#getStatus().

Vì vậy, nếu có chỗ để nâng cấp, hãy nâng cấp lên Servlet 3.0 (Tomcat 7, Glassfish 3, JBoss AS 6, v.v.) và bạn không cần trình bao bọc.

chain.doFilter(request, response); 
int status = ((HttpServletResponse) response).getStatus(); 

Answer 5

Cũng cần phải bao gồm một wrapper cho #sendRedirect, và nó sẽ là tốt hơn để khởi tạo trạng thái để '200' chứ không phải '0'

private int httpStatus = SC_OK; 

... 

@Override 
public void sendRedirect(String location) throws IOException { 
    httpStatus = SC_MOVED_TEMPORARILY; 
    super.sendRedirect(location); 
} 

Answer 6

Nếu bạn đang mắc kẹt với một container cũ rồi một giải pháp thay thế để David Rabinowitz có sử dụng mã trạng thái thực tế (trong trường hợp nó thay đổi sau khi nó được thiết lập bằng cách sử dụng wrapper) là:

public class StatusExposingServletResponse extends HttpServletResponseWrapper { 

    public StatusExposingServletResponse(HttpServletResponse response) { 
     super(response); 
    } 

    @Override 
    public void sendError(int sc) throws IOException { 
     super.sendError(sc); 
    } 

    @Override 
    public void sendError(int sc, String msg) throws IOException { 
     super.sendError(sc, msg); 
    } 

    @Override 
    public void setStatus(int sc) { 
     super.setStatus(sc); 
    } 

    public int getStatus() { 
     try { 
      ServletResponse object = super.getResponse(); 

      // call the private method 'getResponse' 
      Method method1 = object.getClass().getMethod("getResponse"); 
      Object servletResponse = method1.invoke(object, new Object[] {}); 

      // call the parents private method 'getResponse' 
      Method method2 = servletResponse.getClass().getMethod("getResponse"); 
      Object parentResponse = method2.invoke(servletResponse, new Object[] {}); 

      // call the parents private method 'getResponse' 
      Method method3 = parentResponse.getClass().getMethod("getStatus"); 
      int httpStatus = (Integer) method3.invoke(parentResponse, new Object[] {}); 

      return httpStatus; 
     } 
     catch (Exception e) { 
      e.printStackTrace(); 
      return HttpServletResponse.SC_ACCEPTED; 
     } 
    } 

    public String getMessage() { 
     try { 
      ServletResponse object = super.getResponse(); 

      // call the private method 'getResponse' 
      Method method1 = object.getClass().getMethod("getResponse"); 
      Object servletResponse = method1.invoke(object, new Object[] {}); 

      // call the parents private method 'getResponse' 
      Method method2 = servletResponse.getClass().getMethod("getResponse"); 
      Object parentResponse = method2.invoke(servletResponse, new Object[] {}); 

      // call the parents private method 'getResponse' 
      Method method3 = parentResponse.getClass().getMethod("getReason"); 
      String httpStatusMessage = (String) method3.invoke(parentResponse, new Object[] {}); 

      if (httpStatusMessage == null) { 
       int status = getStatus(); 
       java.lang.reflect.Field[] fields = HttpServletResponse.class.getFields(); 

       for (java.lang.reflect.Field field : fields) { 
        if (status == field.getInt(servletResponse)) { 
         httpStatusMessage = field.getName(); 
         httpStatusMessage = httpStatusMessage.replace("SC_", ""); 
         if (!"OK".equals(httpStatusMessage)) { 
          httpStatusMessage = httpStatusMessage.toLowerCase(); 
          httpStatusMessage = httpStatusMessage.replace("_", " "); 
          httpStatusMessage = capitalizeFirstLetters(httpStatusMessage); 
         } 

         break; 
        } 
       } 
      } 

      return httpStatusMessage; 
     } 
     catch (Exception e) { 
      e.printStackTrace(); 
      return ""; 
     } 
    } 

    private static String capitalizeFirstLetters(String s) { 

     for (int i = 0; i < s.length(); i++) { 
      if (i == 0) { 
       // Capitalize the first letter of the string. 
       s = String.format("%s%s", Character.toUpperCase(s.charAt(0)), s.substring(1)); 
      } 

      if (!Character.isLetterOrDigit(s.charAt(i))) { 
       if (i + 1 < s.length()) { 
        s = String.format("%s%s%s", s.subSequence(0, i + 1), 
          Character.toUpperCase(s.charAt(i + 1)), 
          s.substring(i + 2)); 
       } 
      } 
     } 

     return s; 

    } 

    @Override 
    public String toString() { 
     return this.getMessage() + " " + this.getStatus(); 
    } 

} 

Cảnh báo: rất nhiều giả định của hệ thống phân cấp lớp khi sử dụng SNE phản ánh và quan tâm đến các giá trị dữ liệu riêng tư.

Answer 7

Ngoài câu trả lời của David, bạn cũng sẽ muốn ghi đè lên các phương pháp thiết lập lại:

@Override 
public void reset() { 
    super.reset(); 
    this.httpStatus = SC_OK; 
} 

... cũng như phản đối setStatus (int, String)

@Override 
public void setStatus(int status, String string) { 
    super.setStatus(status, string); 
    this.httpStatus = status; 
}