[Spring JPA] 1-11. 회원 가입 : 현재 인증된 사용자 정보 참조

package com.studyolle.account.controller;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

import org.springframework.security.core.annotation.AuthenticationPrincipal;

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.PARAMETER)
@AuthenticationPrincipal(expression = "#this == 'anonymousUser' ? null : account")
public @interface CurrentUser {

}

 

먼저 CurrentUser라는 어노테이션을 사용하기 위해 위와 같은 @interface 객체를 생성해 줍니다.

 

 

package com.studyolle.account.controller;

import java.util.List;

import org.springframework.security.core.authority.SimpleGrantedAuthority;
import org.springframework.security.core.userdetails.User;

import com.studyolle.repository.dto.Account;

import lombok.Getter;

@Getter
public class UserAccount extends User{
	
	private Account account;
	
	public UserAccount(Account account) {
		super(account.getNickname(), account.getPassword(),List.of(new SimpleGrantedAuthority("ROLE_USER")));
		this.account=account;
	}

}

 

그다음 위와 같이 Spring Security에서 제공하는 user를 이용을 해서 위와 같이 작성을 해줍니다.

 

    public void login(Account account) {
        UsernamePasswordAuthenticationToken token = new UsernamePasswordAuthenticationToken(
                new UserAccount(account),
                account.getPassword(),
                List.of(new SimpleGrantedAuthority("ROLE_USER")));
        SecurityContextHolder.getContext().setAuthentication(token);
    }

 

이전에 작성했던 AccountService에서 login을 위와 같이 변경해 줍니다.

 

public class AccountService implements UserDetailsService

 

그리고 AccountService는 Spring Security에서 제공하는 UserDetailsService를 상속받아서 정의해 줘야 합니다.

 

    @Override
    public UserDetails loadUserByUsername(String emailOrNickname) throws UsernameNotFoundException {
        Account account = accountRepository.findByEmail(emailOrNickname);
        if (account == null) {
            account = accountRepository.findByNickname(emailOrNickname);
        }

        if (account == null) {
            throw new UsernameNotFoundException(emailOrNickname);
        }

        return new UserAccount(account);
    }

위의 코드를 서비스에 넣어주어 Spring Security에 로그인을 할 수 있도록 해줍니다.

 

 

위와 같이 로그인을 하게되면

 

 

위와 같이 닉네임이 나오는 것을 확인할 수 있습니다.