使用AOP在切面逻辑中无法获取到requesetBody

使用场景:在接口处理之前,我们需要拿到请求参数,对参数进行校验。注意,这里需要拿到的是原始的请求信息!

一般的获取方式

ServletInputStream inputStream = request.getInputStream();
StringBuilder stringBuilder = new StringBuilder();
try (BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream, request.getCharacterEncoding()))) {
    String line;
    while ((line = reader.readLine()) != null) {
        stringBuilder.append(line);
    }
} catch (IOException e) {
    e.printStackTrace();
}
String requestBody = stringBuilder.toString();

获取requestBody的代码大同小异,可自行搜索。代码没有问题,但是无法获取到requestBody,就要考虑接口方法是否使用了@RequestBody。

在Spring MVC框架中,当你使用@RequestBody注解时,Spring会在将请求转发到控制器方法之前,从HTTP请求体中读取数据并将其反序列化为Java对象。这一处理过程是由HttpMessageConverter完成的,它负责将请求体中的JSON、XML或其他格式的数据转换为Java对象。

请求体只能被读取一次:HTTP请求体是一个输入流,它只能被读取一次。一旦HttpMessageConverter读取了请求体,输入流就被“消耗”掉了,无法再次被读取。因此,当切面尝试访问请求体时,它发现输入流已经为空或不可用。

解决方法:

import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;

public class CachedBodyHttpServletRequestWrapper extends HttpServletRequestWrapper {

    private final byte[] cachedBody;

    public CachedBodyHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        InputStream inputStream = request.getInputStream();
        byte[] buffer = new byte[1024];
        int bytesRead;
        while ((bytesRead = inputStream.read(buffer)) != -1) {
            outputStream.write(buffer, 0, bytesRead);
        }
        cachedBody = outputStream.toByteArray();
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(cachedBody);
        return new ServletInputStream() {
            @Override
            public boolean isFinished() {
                return byteArrayInputStream.available() == 0;
            }

            @Override
            public boolean isReady() {
                return true;
            }

            @Override
            public void setReadListener(ReadListener readListener) {
            }

            @Override
            public int read() throws IOException {
                return byteArrayInputStream.read();
            }
        };
    }

    public String getCachedBody() {
        return new String(cachedBody, StandardCharsets.UTF_8);
    }
}
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;

public class CachedBodyFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
    }

    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
            throws IOException, ServletException {
        if (request instanceof HttpServletRequest) {
            HttpServletRequest httpRequest = (HttpServletRequest) request;
            CachedBodyHttpServletRequestWrapper cachedBodyRequestWrapper = new CachedBodyHttpServletRequestWrapper(httpRequest);
            chain.doFilter(cachedBodyRequestWrapper, response);
        } else {
            chain.doFilter(request, response);
        }
    }

    @Override
    public void destroy() {
    }
}
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class FilterConfig {

    @Bean
    public FilterRegistrationBean<CachedBodyFilter> loggingFilter(){
        FilterRegistrationBean<CachedBodyFilter> registrationBean 
          = new FilterRegistrationBean<>();

        registrationBean.setFilter(new CachedBodyFilter());
        registrationBean.addUrlPatterns("/*");

        return registrationBean;    
    }
}

然后在切面里利用CachedBodyHttpServletRequestWrapper 去获取requestBody就可以了。

### 在AOP中实现获取用户信息的功能 在面向切面编程(AOP)中,可以通过拦截方法调用的方式,在切面获取用户信息。通常情况下,用户信息可能存储在当前线程的上下文中(如 `ThreadLocal`),或者通过请求参数传递。以下是实现这一功能的具体方法和代码示例。 #### 1. 使用 `ThreadLocal` 存储用户信息 在实际应用中,用户信息(如用户名、用户ID等)可以在用户登录时存储到 `ThreadLocal` 中,然后在 AOP 切面中直接获取这些信息[^1]。 ```java public class UserContext { private static final ThreadLocal<String> currentUser = new ThreadLocal<>(); public static void setCurrentUser(String user) { currentUser.set(user); } public static String getCurrentUser() { return currentUser.get(); } public static void clear() { currentUser.remove(); } } ``` 在 AOP 切面中,可以调用 `UserContext.getCurrentUser()` 方法来获取当前用户信息。 #### 2. 在 AOP 切面获取用户信息 以下是一个简单的 AOP 切面示例,展示了如何在切面获取用户信息并进行日志记录[^3]。 ```java import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.annotation.Around; import org.aspectj.lang.annotation.Aspect; import org.springframework.stereotype.Component; @Aspect @Component public class UserLoggingAspect { @Around("execution(* com.example.service.*.*(..))") public Object logUserInformation(ProceedingJoinPoint joinPoint) throws Throwable { // 获取当前用户信息 String currentUser = UserContext.getCurrentUser(); if (currentUser != null) { System.out.println("Current User: " + currentUser); } else { System.out.println("No user information available."); } // 继续执行目标方法 Object result = joinPoint.proceed(); return result; } } ``` #### 3. 动态代理与 Spring AOP 的结合 Spring AOP 底层使用动态代理机制实现切面功能[^4]。因此,开发者无需关心底层细节,只需定义切点表达式和通知逻辑即可完成功能增强。 #### 4. 注意事项 - 如果用户信息存储在 HTTP 请求中(如 `HttpServletRequest`),可以通过依赖注入 `RequestContextHolder` 来获取当前请求对象[^2]。 - 确保在适当的地方清理 `ThreadLocal` 数据,以避免内存泄漏问题。 ```java import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.ServletRequestAttributes; import javax.servlet.http.HttpServletRequest; public class WebUserContext { public static String getCurrentUserFromRequest() { HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest(); return request.getHeader("X-User-Id"); // 假设用户信息存储在请求头中 } } ``` ### 示例代码整合 将上述逻辑整合后,完整的 AOP 切面代码如下: ```java @Aspect @Component public class UserLoggingAspect { @Around("execution(* com.example.service.*.*(..))") public Object logUserInformation(ProceedingJoinPoint joinPoint) throws Throwable { String currentUser = UserContext.getCurrentUser(); // 或者使用 WebUserContext.getCurrentUserFromRequest() if (currentUser != null) { System.out.println("Current User: " + currentUser); } else { System.out.println("No user information available."); } Object[] args = joinPoint.getArgs(); System.out.println("Method Arguments: " + Arrays.toString(args)); Object result = joinPoint.proceed(); return result; } } ``` --- ###
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值