集成ai部分

ruoyi-common/src/main/java/com/ruoyi/common/utils/EncryptHttpServletRequestWrapper.java

@@ -0,0 +1,58 @@
+package com.ruoyi.common.utils;
+
+import javax.servlet.ReadListener;
+import javax.servlet.ServletInputStream;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletRequestWrapper;
+import java.io.BufferedReader;
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.nio.charset.StandardCharsets;
+
+/**
+ * 请求包装器 - 用于读取和修改请求体
+ */
+public class EncryptHttpServletRequestWrapper extends HttpServletRequestWrapper {
+    private final String body;
+    
+    public EncryptHttpServletRequestWrapper(HttpServletRequest request, String body) {
+        super(request);
+        this.body = body;
+    }
+    
+    @Override
+    public ServletInputStream getInputStream() throws IOException {
+        final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes(StandardCharsets.UTF_8));
+        return new ServletInputStream() {
+            @Override
+            public boolean isFinished() {
+                return false;
+            }
+            
+            @Override
+            public boolean isReady() {
+                return false;
+            }
+            
+            @Override
+            public void setReadListener(ReadListener readListener) {
+            }
+            
+            @Override
+            public int read() throws IOException {
+                return byteArrayInputStream.read();
+            }
+        };
+    }
+    
+    @Override
+    public BufferedReader getReader() throws IOException {
+        return new BufferedReader(new InputStreamReader(this.getInputStream()));
+    }
+    
+    public String getBody() {
+        return this.body;
+    }
+}
+

ruoyi-common/src/main/java/com/ruoyi/common/utils/EncryptHttpServletResponseWrapper.java

@@ -0,0 +1,74 @@
+package com.ruoyi.common.utils;
+
+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.IOException;
+import java.io.OutputStreamWriter;
+import java.io.PrintWriter;
+import java.nio.charset.StandardCharsets;
+
+/**
+ * 响应包装器 - 用于修改响应内容
+ */
+public class EncryptHttpServletResponseWrapper extends HttpServletResponseWrapper {
+    private ByteArrayOutputStream buffer;
+    private ServletOutputStream out;
+    private PrintWriter writer;
+    
+    public EncryptHttpServletResponseWrapper(HttpServletResponse response) {
+        super(response);
+        buffer = new ByteArrayOutputStream();
+        out = new WrappedOutputStream(buffer);
+        writer = new PrintWriter(new OutputStreamWriter(buffer, StandardCharsets.UTF_8));
+    }
+    
+    @Override
+    public ServletOutputStream getOutputStream() throws IOException {
+        return out;
+    }
+    
+    @Override
+    public PrintWriter getWriter() throws IOException {
+        return writer;
+    }
+    
+    @Override
+    public void flushBuffer() throws IOException {
+        if (out != null) {
+            out.flush();
+        }
+        if (writer != null) {
+            writer.flush();
+        }
+    }
+    
+    public byte[] getContent() throws IOException {
+        flushBuffer();
+        return buffer.toByteArray();
+    }
+    
+    private static class WrappedOutputStream extends ServletOutputStream {
+        private ByteArrayOutputStream buffer;
+        
+        public WrappedOutputStream(ByteArrayOutputStream buffer) {
+            this.buffer = buffer;
+        }
+        
+        @Override
+        public void write(int b) throws IOException {
+            buffer.write(b);
+        }
+        
+        @Override
+        public boolean isReady() {
+            return false;
+        }
+        
+        @Override
+        public void setWriteListener(WriteListener writeListener) {
+        }
+    }
+}