[Spring boot] 첨부파일 기능 추가

Spring boot - JPA 활용 게시판 첨부파일 기능 추가하기 

 

FIleStore 콤포넌트 추가 

1. 첨부파일 저장 경로 지정 
2. 첨부파일 이름 변환 (작성자가 업로드한 파일명 → 서버 내부에서 관리하는 파일명)
3. 파일 저장

 

package com.sjy.shopping.attachfile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import com.sjy.shopping.model.entity.UploadFile;

@Component
public class FileStore {

	//루트 경로 불러오기
	private final static String rootPath = System.getProperty("user.dir");
	
	//프로젝트 루트 경로에 있는 files 디렉토리
	private final static String fileDir = rootPath + "/src/main/resources/static/files/";
	
	public static String getPullPath(String filename) { return fileDir + filename;}
	
	public static UploadFile storeFile(MultipartFile multipartFile) throws IOException{
		if(multipartFile.isEmpty()) {
			return null;
		}
		
		String originalFilename = multipartFile.getOriginalFilename();
		//작성자가 업로드한 파일명 -> 서버 내부에서 관리하는 파일명으로 변환
		//파일명 중복되지 않도록 UUID로 정하고 확장자는 그대로!
		String storeFilename = UUID.randomUUID() + "." + extractExt(originalFilename);
		
		//파일을 저장하는 부분 : 파일 경로 + storeFilename에 저장
		multipartFile.transferTo(new File(getPullPath(storeFilename)));
		
		return new UploadFile(originalFilename, storeFilename);
	}
	
	
    // 확장자 추출
    private static String extractExt(String originalFilename) {
        int pos = originalFilename.lastIndexOf(".");
        return originalFilename.substring(pos + 1);
    }
	
}

 

 

첨부파일 Entity 추가 - UploadFile

1. id, 업로드한 파일명, 저장된 파일명을 컬럼으로 가짐
2. posts(게시물) Entity와는 id 컬럼으로 join

 

package com.sjy.shopping.model.entity;

import jakarta.persistence.Column;
import jakarta.persistence.Entity;
import jakarta.persistence.GeneratedValue;
import jakarta.persistence.GenerationType;
import jakarta.persistence.Id;
import jakarta.persistence.JoinColumn;
import jakarta.persistence.ManyToOne;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;

@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
@Entity(name="uploadfile")
public class UploadFile {
	
	@Id
	@GeneratedValue(strategy = GenerationType.IDENTITY)
	@Column(name = "attach_id")
	private Long id;
	
	@Column(name = "upload_filename")
	private String uploadFilename; //작성자가 업로드한 파일명
	
	@Column(name = "store_filename")
	private String storeFilename; //서버 내부에서 관리하는 파일명
	
	@Builder
	public UploadFile(String uploadFilename, String storeFilename) {
		this.uploadFilename = uploadFilename;
		this.storeFilename = storeFilename;
	}
}

 

 

PostController 수정 (글 등록)

1. @RequestPart() MultipartFile
2. 글 등록시 입력받은 첨부파일을 서버에 저장
3. 데이터베이스에는 파일명 저장 

 

	//글 등록
	@PostMapping("/post/add")
	public void post(HttpServletRequest request, HttpSession session, HttpServletResponse response, @RequestPart("attach") MultipartFile file) throws Exception {
		Users user = (Users)session.getAttribute("loginUser");	
		
		//첨부파일 처리
		UploadFile attachFile = null;
		try {
			attachFile = FileStore.storeFile(file);
		} catch (IOException e) {
			e.printStackTrace();
		}
		
		PostReqDto requestDto = new PostReqDto(request.getParameter("title"), request.getParameter("contents"), user, request.getParameter("category_name"), attachFile);
		System.out.println(requestDto.getCategory_name());
		log.info("view to controll with post");
		postService.addPost(requestDto);
		response.sendRedirect("/list");
	}

 

 

 

PostService 수정 

 

	@Transactional
	public Posts addPost(PostReqDto requestDto) {
		Category category = categoryRepository.findByName(requestDto.getCategory_name());
		UploadFile file = attachRepository.save(requestDto.getFile());
		Posts post = new Posts(requestDto.getTitle(), requestDto.getContents(), requestDto.getUser(), category, file);
		return postRepository.save(post);
	}

 

 

 

attachRepository 생성

 

package com.sjy.shopping.repository;

import org.springframework.data.jpa.repository.JpaRepository;

import com.sjy.shopping.model.entity.UploadFile;

public interface AttachRepository extends JpaRepository<UploadFile, Long> {

}

 

 

Front - request

form 태그 내 <input type="file"> 로 파일 전송

 

<div class="form-group">
    <label class="control-label col-sm-2" for="attach">첨부파일:</label>
    <div class="col-sm-10">
        <input type="file" class="form-control" id="attach" name="attach">
    </div>
</div>

 

 

Front - response

 

<tr>
    <td>첨부파일(클릭시 다운)</td>
    <td><a th:if="${post.file}" th:href="|/attach/${post.id}|" th:text="${post.file.uploadFilename}"/></td>
</tr>

 

 

첨부파일 클릭시 다운

PostController에 작성

 

    //첨부파일 다운로드
    @GetMapping("/attach/{id}")
    public ResponseEntity<UrlResource> download(@PathVariable long id) throws MalformedURLException{
    	Posts post = postService.findPost(id);
    	String storeFilename = post.getFile().getStoreFilename();
    	String uploadFilename = post.getFile().getUploadFilename();
    	
    	UrlResource urlResource = new UrlResource("file:" + FileStore.getPullPath(storeFilename));
    	
    	//한글 인코딩
    	String encodedUploadFileName = UriUtils.encode(uploadFilename, StandardCharsets.UTF_8);
    	String contentDisposition = "attachment; filename=\"" + encodedUploadFileName + "\"";
    	
    	//header에 CONTENT_DIPOSITION 설정을 통해 클릭시 다운로드 설정
    	return ResponseEntity.ok()
    			.header(HttpHeaders.CONTENT_DISPOSITION, contentDisposition)
    			.body(urlResource);

    }

 

 

 

 

 

참고자료

[Spring Boot] 파일 업로드