contextHolder = new ThreadLocal<>();
@Override
public void clearContext() {
contextHolder.remove(); // 메모리 누수 방지 및 스레드 풀 오염 방지
}
@Override
public SecurityContext getContext() {
SecurityContext ctx = contextHolder.get(); // 현재 스레드 맵에서 조회
if (ctx == null) {
ctx = createEmptyContext();
contextHolder.set(ctx);
}
return ctx;
}
@Override
public void setContext(SecurityContext context) {
Assert.notNull(context, "Only non-null SecurityContext instances are permitted");
contextHolder.set(context); // 현재 스레드 맵에 저장
}
// ... 생략
}
```
{% endtab %}
{% endtabs %}
#### 실전 사이드 프로젝트 코드를 탐구하기
아래 사이드 프로젝트의 예제는 코틀린으로 구성 되어있다.
> SecurityContextHolder 를 감싼 스레드 로컬의 get() 은 어떻게 쓰이고 있을까?
>
> * 유저 프로필을 조회하는 리졸버 컴포넌트이다.
@Component
class ProfileArgumentResolver(
private val profileService: ProfileService,
) : HandlerMethodArgumentResolver {
override fun supportsParameter(parameter: MethodParameter): Boolean {
return parameter.hasParameterAnnotation(UserProfile::class.java)
}
override fun resolveArgument(
parameter: MethodParameter,
mavContainer: ModelAndViewContainer?,
webRequest: NativeWebRequest,
binderFactory: WebDataBinderFactory?
): Any? {
val user = SecurityContextHolder.getContext().authentication.principal as User
return profileService.getProfileByUserId(user.id).id
}
}
> SecurityContextHolder 를 감싼 스레드 로컬의 set() 은 어떻게 쓰이고 있을까?
>
> * jwt 인증 필터 구성 정보이다.
class JwtAuthenticationFilter(
private val tokenService: TokenService,
private val authService: AuthService,
private val handlerExceptionResolver: HandlerExceptionResolver,
private val permitAllMatchers: List<RequestMatcher>,
private val allowEndpoints: ApiAllowEndpoints,
) : OncePerRequestFilter() {
override fun doFilterInternal(
request: HttpServletRequest,
response: HttpServletResponse,
filterChain: FilterChain
) {
try {
if (isPermitAllRequest(request) || doesNotAllowEndpoint(request)) {
filterChain.doFilter(request, response)
return
}
val accessToken: String = resolveToken(request) ?: throw JwtAuthenticationException("액세스 토큰을 찾을 수 없습니다.")
if (tokenService.validate(accessToken)) {
val memberId: UUID = tokenService.getSubject(accessToken)
val user: User = authService.getUserById(memberId)
val authenticationToken: Authentication = PreAuthenticatedAuthenticationToken(user, null, ArrayList())
SecurityContextHolder.getContext().authentication = authenticationToken
filterChain.doFilter(request, response)
return
}
throw JwtAuthenticationException("잘못된 토큰 정보입니다.")
} catch (exception: Exception) {
SecurityContextHolder.clearContext()
handlerExceptionResolver.resolveException(request, response, null, exception)
}
}
이렇게 적용된 코드는 사용자의 요청을 처리하기 까지 계속 유저 정보를 다른 컴포넌트와 공유하여 사용할 수 있게된다.
만약, 스레드 로컬을 사용하지 못했다면 사용자 인증을 거친 후 프로필 조회, 게시글 조회 등 유저 인증 정보가 필요한 모든 컴포넌트 마다 우리는 인자로 유저 인증 정보를 계속 전달하여 사용해야한다.
### 정리하기
***
* **동시성 문제 해결**: `ThreadLocal`은 스레드마다 고유한 저장소를 생성하므로, 멀티 스레드 환경에서도 다른 스레드의 영향을 받지 않고 안전하게 데이터를 관리할 수 있다.
* **비즈니스 로직 단순화**: 객체 간 협력 관계에서 불필요한 파라미터를 획기적으로 줄여주며, 스레드가 작업을 마칠 때까지 데이터를 전역적으로 공유할 수 있다.
* **리소스 정리**: 스레드 풀을 사용하는 현대 웹 애플리케이션 환경에서는 스레드가 재사용된다. 따라서 데이터 오염을 막기 위해 사용 후에는 반드시 `remove()` 메서드로 저장된 값을 지워야 한다.
참고 자료
*
*
*
