ll

liukangdong

2025-03-03 3aa3e034d30b201dde95d9bc33e5b1250dd8f256

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package com.doumee.core.servlet;
 
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import java.io.*;
import java.nio.charset.Charset;
 
/**
 * 包含副本的输入流
 * @author Eva.Caesar Liu
 * @date 2022/03/15 09:54
 */
public class ServletDuplicateInputStream extends ServletInputStream {
 
    private ServletInputStream stream;
 
    private ByteArrayInputStream duplicate;
 
    private String body;
 
    public ServletDuplicateInputStream (ServletInputStream servletInputStream) {
        this.stream = servletInputStream;
        this.body = this.getBodyString();
        this.duplicate = new ByteArrayInputStream(body.getBytes(Charset.forName("UTF-8")));
    }
 
    @Override
    public int read() {
        return duplicate.read();
    }
 
    @Override
    public boolean isFinished() {
        return stream.isFinished();
    }
 
    @Override
    public boolean isReady() {
        return stream.isReady();
    }
 
    @Override
    public void setReadListener(ReadListener readListener) {
        stream.setReadListener(readListener);
    }
 
    /**
     * 获取请求体参数字符串
     */
    private String getBodyString() {
        StringBuilder sb = new StringBuilder();
        InputStream inputStream = null;
        BufferedReader reader = null;
        try {
            inputStream = cloneInputStream(this.stream);
            reader = new BufferedReader(new InputStreamReader(inputStream, Charset.forName("UTF-8")));
            String line;
            while ((line = reader.readLine()) != null) {
                sb.append(line);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (inputStream != null) {
                try {
                    inputStream.close();
                }
                catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (reader != null) {
                try {
                    reader.close();
                }
                catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return sb.toString();
    }
 
    /**
     * 复制输入流
     */
    private InputStream cloneInputStream(ServletInputStream inputStream) {
        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
        byte[] buffer = new byte[1024];
        int len;
        try {
            while ((len = inputStream.read(buffer)) > -1) {
                byteArrayOutputStream.write(buffer, 0, len);
            }
            byteArrayOutputStream.flush();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return new ByteArrayInputStream(byteArrayOutputStream.toByteArray());
    }
 
    /**
     * 获取请求体内容
     */
    public String getBody () {
        return body;
    }
}