MenuImgService.java

package com.ourmenu.backend.domain.menu.application;

import com.ourmenu.backend.domain.menu.application.converter.DefaultImgConverterService;
import com.ourmenu.backend.domain.menu.dao.MenuImgRepository;
import com.ourmenu.backend.domain.menu.domain.MenuImg;
import java.util.List;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;

@Service
@RequiredArgsConstructor
public class MenuImgService {

    private final MenuImgRepository menuImgRepository;
    private final AwsS3Service awsS3Service;
    private final DefaultImgConverterService defaultImgConverterService;

    @Transactional
    public List<MenuImg> saveMenuImgs(Long menuId, List<MultipartFile> multipartFiles) {
        List<MenuImg> menuImgs = multipartFiles.stream()
                .map(awsS3Service::uploadFileAsync)
                .map(imgUrl -> this.saveMenuImg(menuId, imgUrl))
                .toList();
        if (menuImgs.size() == 0) {
            return List.of(createDefaultMenuImg());
        }
        return menuImgs;
    }

    /**
     * 메뉴 이미지 및 menuImg 삭제
     *
     * @param menuId
     */
    @Transactional
    public void deleteMenuImgs(Long menuId) {
        List<MenuImg> menuImgs = menuImgRepository.findAllByMenuId(menuId);
        menuImgs.forEach(menuImg -> awsS3Service.deleteFileAsync(menuImg.getImgUrl()));
        menuImgRepository.deleteAllByMenuId(menuId);
    }

    /**
     * 메뉴 이미지 하나 조회 하나도 없다면 기본 메뉴
     *
     * @param menuId
     * @return
     */
    @Transactional
    public String findUniqueImg(Long menuId) {
        return menuImgRepository.findAllByMenuId(menuId).stream()
                .map(MenuImg::getImgUrl)
                .findFirst()
                .orElse(defaultImgConverterService.getDefaultMenuImgUrl());
    }

    /**
     * 메뉴 이미지 조회 없으면 기본 이미지 반환
     *
     * @param menuId
     * @return
     */
    @Transactional
    public List<String> findImgUrls(Long menuId) {
        List<String> menuImgs = menuImgRepository.findAllByMenuId(menuId).stream()
                .map(MenuImg::getImgUrl)
                .toList();
        if (menuImgs.size() == 0) {
            return List.of(createDefaultMenuImg().getImgUrl());
        }
        return menuImgs;
    }

    private MenuImg saveMenuImg(Long menuId, String imgUrl) {
        MenuImg menuImg = MenuImg.builder()
                .menuId(menuId)
                .imgUrl(imgUrl)
                .build();
        return menuImgRepository.save(menuImg);
    }

    /**
     * 기본 이미지 생성
     *
     * @return
     */
    private MenuImg createDefaultMenuImg() {
        return MenuImg.builder()
                .menuId(null)
                .imgUrl(defaultImgConverterService.getDefaultMenuImgUrl())
                .build();
    }
}