新建德森项目分支

skyouc

2024-12-21 c635d78b479510ebe2556a420948effcd30a0731

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package com.zy.asrs.wms.common.security; 
 
import javax.servlet.ServletOutputStream; 
import javax.servlet.WriteListener; 
import javax.servlet.http.HttpServletResponse; 
import javax.servlet.http.HttpServletResponseWrapper; 
import java.io.ByteArrayOutputStream; 
import java.io.PrintWriter; 
 
public class CachedBodyHttpServletResponse extends HttpServletResponseWrapper { 
    private final ByteArrayOutputStream cachedBody; 
    private final PrintWriter writer; 
 
    public CachedBodyHttpServletResponse(HttpServletResponse response) { 
        super(response); 
        this.cachedBody = new ByteArrayOutputStream(); 
        this.writer = new PrintWriter(cachedBody); 
    } 
 
    @Override 
    public PrintWriter getWriter() { 
        return writer; 
    } 
 
    @Override 
    public ServletOutputStream getOutputStream() { 
        return new ServletOutputStream() { 
            @Override 
            public void write(int b) { 
                cachedBody.write(b); 
            } 
 
            @Override 
            public boolean isReady() { 
                return true; 
            } 
 
            @Override 
            public void setWriteListener(WriteListener writeListener) { 
 
            } 
        }; 
    } 
 
    public byte[] getContent() { 
        writer.flush(); 
        return cachedBody.toByteArray(); 
    } 
}