update 租赁系统11-14 一订单对应多个卖家需求完成
This commit is contained in:
@@ -0,0 +1,64 @@
|
||||
package com.m2pool.common.security.interceptor;
|
||||
|
||||
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.InputStreamReader;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
public class AuthRequestWrapper extends HttpServletRequestWrapper {
|
||||
|
||||
private String body;
|
||||
|
||||
public AuthRequestWrapper(HttpServletRequest request) {
|
||||
super(request);
|
||||
try(BufferedReader reader = request.getReader()){
|
||||
body= reader.lines().collect(Collectors.joining());
|
||||
}catch (Exception e){
|
||||
System.out.println("!!-- read request from requestbody error"+e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public BufferedReader getReader() {
|
||||
return new BufferedReader(new InputStreamReader(this.getInputStream()));
|
||||
}
|
||||
|
||||
@Override
|
||||
public ServletInputStream getInputStream() {
|
||||
final ByteArrayInputStream byteArrayIns = new ByteArrayInputStream(body.getBytes());
|
||||
ServletInputStream servletIns = new ServletInputStream() {
|
||||
@Override
|
||||
public boolean isFinished() {
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isReady() {
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setReadListener(ReadListener readListener) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public int read() {
|
||||
return byteArrayIns.read();
|
||||
}
|
||||
};
|
||||
return servletIns;
|
||||
}
|
||||
|
||||
public String getBody() {
|
||||
return body;
|
||||
}
|
||||
|
||||
public void setBody(String body) {
|
||||
this.body = body;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,66 @@
|
||||
package com.m2pool.common.security.interceptor;
|
||||
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.web.servlet.HandlerInterceptor;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.BufferedReader;
|
||||
import java.io.IOException;
|
||||
|
||||
@Component
|
||||
public class CoinInterceptor implements HandlerInterceptor {
|
||||
|
||||
private final ObjectMapper objectMapper = new ObjectMapper();
|
||||
|
||||
@Override
|
||||
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
|
||||
System.out.println("调用Coin拦截器1:" + request);
|
||||
|
||||
// 仅处理 POST、PUT 等有请求体的请求方法
|
||||
if ("POST".equalsIgnoreCase(request.getMethod()) || "PUT".equalsIgnoreCase(request.getMethod())) {
|
||||
try {
|
||||
// 读取请求体内容
|
||||
if (request instanceof AuthRequestWrapper){
|
||||
//BufferedReader reader = request.getReader();
|
||||
//System.out.println("调用Coin拦截器2:" + reader);
|
||||
//
|
||||
//if (!requestBody.isEmpty()) {
|
||||
// // 反序列化请求体为 RequestObject 对象
|
||||
// RequestObject requestObject = objectMapper.readValue(requestBody, RequestObject.class);
|
||||
// System.out.println("调用Coin拦截器3:" + requestObject.toString());
|
||||
// String coin = requestObject.getCoin();
|
||||
// System.out.println("调用Coin拦截器4:" + coin);
|
||||
// if ("enx".equals(coin)) {
|
||||
// response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
|
||||
// response.getWriter().write("enx币种已下架");
|
||||
// return false;
|
||||
// }
|
||||
//}
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
// 处理反序列化异常
|
||||
response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
|
||||
response.getWriter().write("请求体解析失败");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 读取请求体内容
|
||||
*/
|
||||
private String readRequestBody(HttpServletRequest request) throws IOException {
|
||||
StringBuilder stringBuilder = new StringBuilder();
|
||||
try (BufferedReader reader = request.getReader()) {
|
||||
String line;
|
||||
while ((line = reader.readLine()) != null) {
|
||||
stringBuilder.append(line);
|
||||
}
|
||||
}
|
||||
return stringBuilder.toString();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,27 @@
|
||||
package com.m2pool.common.security.interceptor;
|
||||
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.web.filter.OncePerRequestFilter;
|
||||
import org.springframework.web.util.ContentCachingRequestWrapper;
|
||||
|
||||
import javax.servlet.FilterChain;
|
||||
import javax.servlet.ServletException;
|
||||
import javax.servlet.annotation.WebFilter;
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.IOException;
|
||||
import java.io.BufferedReader;
|
||||
import java.io.InputStreamReader;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
@Component
|
||||
@WebFilter("/*")
|
||||
public class RequestBodyFilter extends OncePerRequestFilter {
|
||||
|
||||
@Override
|
||||
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
|
||||
|
||||
request = new AuthRequestWrapper(request);
|
||||
filterChain.doFilter(request,response);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,16 @@
|
||||
package com.m2pool.common.security.interceptor;
|
||||
|
||||
import com.fasterxml.jackson.annotation.JsonProperty;
|
||||
|
||||
public class RequestObject {
|
||||
private String coin;
|
||||
|
||||
// 添加 getter 和 setter 方法
|
||||
public String getCoin() {
|
||||
return coin;
|
||||
}
|
||||
|
||||
public void setCoin(String coin) {
|
||||
this.coin = coin;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user