当前位置: 首页> 教育> 就业 > Aop切面技术之存储用户信息

Aop切面技术之存储用户信息

时间:2025/7/11 15:28:23来源:https://blog.csdn.net/weixin_42039228/article/details/141133891 浏览次数:0次

一、背景

在我们日常项目中,常常会将用户的认证信息存入缓存中,方便我们在程序执行中,进行获取用户信息。本篇文章主要是介绍使用自定义注解和AOP切面技术进行实现,这也也是非常容易使用的。

二、代码详解

2.1 自定义注解

定义一个注解,用于标记需要自动注入当前用户信息的方法。

import java.lang.annotation.ElementType;  
import java.lang.annotation.Retention;  
import java.lang.annotation.RetentionPolicy;  
import java.lang.annotation.Target;  @Target(ElementType.METHOD)  
@Retention(RetentionPolicy.RUNTIME)  
public @interface InjectCurrentUser {  
}
2.2 创建切面

创建一个切面来拦截所有标记了@InjectCurrentUser注解的方法,并在调用它们之前获取当前用户信息,然后作为参数传递给这些方法(注意:Java的方法签名在编译时是固定的,不能直接向现有方法添加参数。因此,这里假设我们修改方法签名以接受用户信息作为参数,或者使用其他方式如ThreadLocal)。

@Aspect  
@Component  
public class UserAspect {  @Before("@annotation(InjectCurrentUser)")  public void beforeMethod(JoinPoint joinPoint) {  Authentication authentication = SecurityContextHolder.getContext().getAuthentication();  if (authentication != null && authentication.getPrincipal() instanceof UserDetails) {  UserDetails userDetails = (UserDetails) authentication.getPrincipal();  UserContextHolder.setCurrentUser(userDetails);  // 如果需要,可以在这里添加日志或其他逻辑  System.out.println("Setting current user: " + userDetails.getUsername());  }  }  // 可以添加一个@After或@AfterReturning切面来清除ThreadLocal中的用户信息  // 这样做是为了避免内存泄漏,尤其是在长时间运行的线程或线程池中  @After("@annotation(InjectCurrentUser)")  public void afterMethod(JoinPoint joinPoint) {  UserContextHolder.clearCurrentUser();  }  
}
2.3 创建ThreadLocal存储用户信息

定义一个ThreadLocal来存储当前用户的信息(比如UserDetails

public class UserContextHolder {  private static final ThreadLocal<UserDetails> currentUser = new ThreadLocal<>();  public static void setCurrentUser(UserDetails userDetails) {  currentUser.set(userDetails);  }  public static UserDetails getCurrentUser() {  return currentUser.get();  }  // 清除ThreadLocal中的用户信息,避免内存泄漏  public static void clearCurrentUser() {  currentUser.remove();  }  
}
2.4 业务逻辑中获取用户信息

业务逻辑中,通过UserContextHolder.getCurrentUser()来获取当前用户的信息

public class SomeService {  public void someBusinessMethod() {  UserDetails user = UserContextHolder.getCurrentUser();  if (user != null) {  // 使用用户信息进行业务逻辑处理  System.out.println("Doing something with user: " + user.getUsername());  }  }  
}

三、总结

内存泄漏ThreadLocal可能会导致内存泄漏,特别是当使用线程池时,因为线程可能会被重用,而ThreadLocal中的值可能不会被自动清除。因此,在不再需要时显式清除ThreadLocal中的值是一个好习惯。
依赖注入:尽管在这个例子中我们使用了AOP来设置ThreadLocal,但在某些情况下,你可能还想通过依赖注入来传递用户信息,特别是当你需要在多个组件或服务之间共享用户信息时。然而,对于跨线程或跨方法调用的情况,ThreadLocal通常是一个更好的选择。

关键字:Aop切面技术之存储用户信息

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: