[개발일지] 취미 커뮤니티 - 사용자 프로필 설정 및 변경

zwon·2023년 10월 27일

목록 보기

23/23

사용자 프로필 설정 기능이다.
롬복 어노테이션들은 생략하고 포스팅하고자 한다.

User

public class User extends BaseTimeEntity {
  ...
  @OneToOne(mappedBy = "user", fetch = FetchType.LAZY, cascade = CascadeType.ALL)
  private Profile profile;

  public void updateProfile(Profile newProfile) {
    this.profile = newProfile;
    newProfile.setUser(this);
  }
  ...

회원 탈퇴 시 프로필 사진도 삭제되게 CascadeType.ALL로 설정하였다.

Profile

public class Profile {

  @Id
  @GeneratedValue(strategy = GenerationType.IDENTITY)
  private Long id;

  private String originFilename; // 업로드된 파일 이름
  private String storeFilename; // 저장한 파일명 ex) 132-5sdf-23451-1as.png -> UUID로 식별자 생성 + 확장자

  @OneToOne(fetch = FetchType.EAGER)
  @JoinColumn(name = "user_id")
  private User user;

  public void setUser(User user) {
    this.user = user;
  }
}

ImgController

@Controller
@RequiredArgsConstructor
public class ImgController {
  private final FileStore fileStore;
  private final FileService fileService;


  @PostMapping("/profile/update") // 프로필 사진 변경과 삭제
  public String profileUpdate(@ModelAttribute ProfileUpdateDto profileUpdateDto,
                              HttpSession session) throws IOException {
    User loginUser = (User) session.getAttribute("loginUser");

    if (profileUpdateDto.getProfile().isEmpty()) { 
      fileService.deleteBeforeProfile(loginUser.getNickname());
    } else {
      fileService.transferProfile(profileUpdateDto.getProfile(), loginUser.getNickname());
    }

    return "redirect:/my-page";
  }
  // 프로필 이미지 보여주기
  @ResponseBody
  @GetMapping("/profile/{filename}")
  public Resource downloadProfile(@PathVariable String filename)
      throws MalformedURLException {
    return new UrlResource("file:" + fileStore.getProfileFullPath(filename));
  }
}

FileStore

첨부파일 업로드 포스팅과 별 차이 없다.
추가된 부분만 정리하겠다.

  // 프로필 이미지
  @Value("${profile.dir}")
  private String profileDir;

  // 파일 저장 경로 Full
  public String getProfileFullPath(String filename){
    return profileDir + filename;
  }

  public Profile profileImgStore(MultipartFile multipartFile) throws IOException {
    if (multipartFile.isEmpty()){
      return null;
    }
    String originalFileName = multipartFile.getOriginalFilename();
    String ext = extractedExt(originalFileName);
    String storeFileName = createStoreFileName(ext);
    multipartFile.transferTo(new File(getProfileFullPath(storeFileName)));

    return Profile.builder()
        .originFilename(originalFileName)
        .storeFilename(storeFileName)
        .build();
  }

첨부파일 구현과 똑같고 약간 다른거는 파일 경로 가져오는거나(따로 폴더를 생성함) ,, 말고는 없다.

FileService

@Service
@RequiredArgsConstructor
@Slf4j
public class FileService {
  private final FileStore fileStore;
  private final FileRepository fileRepository; 
  private final UserRepository userRepository;
  private final ProfileRepository profileRepository;


  // 프로필 이미지 변경
  @Transactional
  public Profile transferProfile(MultipartFile profile, String nickname) throws IOException {
    User user = userRepository.findUserByNickname(nickname).orElseThrow(IllegalArgumentException::new);
    Profile profileImg = fileStore.profileImgStore(profile);
    user.updateProfile(profileImg);
    return profileImg;
  }

  public void deleteBeforeProfile(String nickname) {
    User user = userRepository.findUserByNickname(nickname).orElseThrow(IllegalArgumentException::new);

    if (user.getProfile() != null) {
      Profile profile = user.getProfile();

      File imgFileOnDisk = new File(fileStore.getProfileFullPath(profile.getStoreFilename()));
      if (imgFileOnDisk.exists()) {
        imgFileOnDisk.delete();
      }
      user.updateProfile(null);
      profileRepository.delete(profile);
    }
  }

}

이 부분도 첨부파일과는 별 차이 없다.

마이페이지

<img id="profile" th:src="|/profile/profile.png|" alt="프로필 사진" th:if="${loginUser.profile == null}">
<img th:src="|/profile/${loginUser.profile.getStoreFilename()}|" th:if="${loginUser.profile != null}"/>