게시판 기능에 파일업로드도 추가해보도록 한다.
@Autowired
FileService fileService;
private static String UPLOAD_FOLDER = "D:/logs/";
private static final long MAX_FILE_SIZE = 10 * 1024 * 1024; // 10MB
@PostMapping("/Modify")
public ModelAndView modifyPost(HttpServletRequest request, HttpServletResponse response, @RequestParam("file") MultipartFile file) throws UnsupportedEncodingException {
response.setCharacterEncoding("UTF-8");
response.setContentType("text/html; charset=UTF-8");
request.setCharacterEncoding("UTF-8");
String num = request.getParameter("num");
Post post = postService.getPost(num);
post.setAuthor(request.getParameter("author"));
post.setContents(request.getParameter("contents"));
post.setTitle(request.getParameter("title"));
post.setNum(request.getParameter("num"));
if (!file.isEmpty()){
try{
FileInfo fileInfo = new FileInfo();
String originalFilename = file.getOriginalFilename();
// String fileExt = FilenameUtils.getExtension(file.getOriginalFilename());
String fileExt = originalFilename.contains(".")
? originalFilename.substring(originalFilename.lastIndexOf('.') + 1)
: "";
//랜덤 이름 생성 (겹치지 않기 위해서)
String storedFilename = UUID.randomUUID().toString() + "." + fileExt;
Path path = Paths.get(UPLOAD_FOLDER, storedFilename);
byte[] bytes = file.getBytes();
//실제 파일 저장
Files.write(path, bytes);
fileInfo.setOriginalFilename(originalFilename);
fileInfo.setStoredFilename(storedFilename);
fileInfo.setFileExt(fileExt);
fileInfo.setFilePath(UPLOAD_FOLDER);
fileInfo.setPostId(post.getNum());
// fileService.insertFile(fileInfo);
postService.modifyPost(post, fileInfo);
}catch (IOException e){
e.printStackTrace();
}
}else{
postService.modifyPost(post);
}
mav.setViewName("redirect:/Content?num=" + post.getNum());
return mav;
}
@PostMapping("/Write")
public ModelAndView writePost(@ModelAttribute Post post, HttpServletResponse response, @RequestParam("file") MultipartFile file) throws UnsupportedEncodingException{
System.out.println("test");
response.setCharacterEncoding("UTF-8");
response.setContentType("text/html; charset=UTF-8");
System.out.println(file + "dd " + !file.isEmpty());
if (!file.isEmpty()){
try{
FileInfo fileInfo = new FileInfo();
String originalFilename = file.getOriginalFilename();
String fileExt = originalFilename.contains(".")
? originalFilename.substring(originalFilename.lastIndexOf('.') + 1)
: "";
//랜덤 이름 생성 (겹치지 않기 위해서)
String storedFilename = UUID.randomUUID().toString() + "." + fileExt;
Path path = Paths.get(UPLOAD_FOLDER, storedFilename);
byte[] bytes = file.getBytes();
//실제 파일 저장
Files.write(path, bytes);
fileInfo.setOriginalFilename(originalFilename);
fileInfo.setStoredFilename(storedFilename);
fileInfo.setFileExt(fileExt);
fileInfo.setFilePath(UPLOAD_FOLDER);
postService.inserPost(post, fileInfo);
}catch (IOException e){
e.printStackTrace();
}
}else{
postService.inserPost(post);
}
mav.setViewName("redirect:/");
return mav;
}새롭게 추가한 부분은 if 문 file.isEmpty() 부분이다.
파일이 있으면 파일 업로드 로직을 수행하게 된다.
상단에 static 변수와 FileService 도 주입시켜주었다.
File VO(model) 객체도 생성하였는데
package com.example.post.model;
import lombok.Data;
@Data
public class FileInfo {
private String fileId;
private String postId;
private String originalFilename;
private String storedFilename;
private String filePath;
private String fileExt;
}fileId, postId(게시글), original(업로드 시 파일이름), stored(저장시 파일이름), 파일경로, Ext(확장자) 이다.
위 메서드에 보면 storedFileName 을 무작위로 생성하여, fileExt(확장자)와 더해 저장하게 되는데
무작위로 생성하여 굳이 저장시 파일 이름을 만드는 이유는 업로드 파일이름이 겹칠 수 있어 다운로드때 문제가 발생할 수 있기 때문이다.
CREATE TABLE FileInfo (
fileId INT IDENTITY(1,1) PRIMARY KEY,
postId INT NOT NULL,
originalFilename NVARCHAR(255) NOT NULL,
storedFilename NVARCHAR(255) NOT NULL,
filePath NVARCHAR(500) NOT NULL,
fileExt NVARCHAR(10),
FOREIGN KEY (postId) REFERENCES test1(num)
);
테이블 생성은 이렇게 했다.
다운로드 컨트롤러는 내가 따로만들었는데.. 파일컨트롤러로 구분하는게 더 좋은지 아직 방법은 잘 모르겠다.
controller/DownloadController.java
package com.example.post.controller;
import java.io.UnsupportedEncodingException;
import java.net.MalformedURLException;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.nio.file.Path;
import java.nio.file.Paths;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.Resource;
import org.springframework.core.io.UrlResource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import com.example.post.model.FileInfo;
import com.example.post.service.FileService;
@Controller
@RequestMapping("/Download")
public class DownloadController {
private static String UPLOAD_FOLDER = "D:/logs/";
@Autowired FileService fileService;
@GetMapping("/{fileId}")
public ResponseEntity<Resource> downloadFile(@PathVariable("fileId") int fileId) throws MalformedURLException, UnsupportedEncodingException {
FileInfo fileInfo = fileService.selectFileInfo(fileId);
if(fileInfo == null){
return ResponseEntity.notFound().build();
}
Path path = Paths.get(fileInfo.getFilePath(), fileInfo.getStoredFilename());
Resource resource = new UrlResource(path.toUri());
if(!resource.exists()){
return ResponseEntity.notFound().build();
}
// 파일명을 UTF-8로 인코딩
String encodedFileName = URLEncoder.encode(fileInfo.getOriginalFilename(), StandardCharsets.UTF_8.toString()).replaceAll("\\+", "%20");
String contentDisposition = "attachment; filename*=UTF-8''" + encodedFileName;
//응답헤더에 CONTETN_DISPOSITION -> 첨부파일이라고 알려줌, attachment; 다음나오는것 다운로드, .body(resource)->resource 에 파일내용 담아감
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION, contentDisposition)
.body(resource);
}
}
컨트롤러를 다 만들었으니 Service 수정을 해본다.
우선 업로드시에 파일도 추가하여야 하므로 PostService 부터 수정한다.
@Autowired FileDao fileDao;
@Autowired FileService fileService;
@Transactional
public void inserPost(Post post, FileInfo fileInfo) {
postDao.insertPost(post);
fileInfo.setPostId(post.getNum());
fileDao.insertFile(fileInfo);
}
Transactional 처리를 해준다.
@Transactional
public void deletePost(String num) throws IOException {
commentDao.deleteCommentByPostId(num);
List<FileInfo> fileInfoList = fileDao.selectFileByPostId(num);
if(!fileInfoList.isEmpty()){
for(FileInfo fileInfo : fileInfoList){
System.out.println(fileInfo + " \n\n\n\n");
fileService.deleteFile(fileInfo);
}
}
//외래키때문에 맨마지막에 삭제해야함.
postDao.deletePost(num);
}
@Transactional
public void modifyPost(Post post, FileInfo fileInfo) throws IOException {
postDao.modifyPost(post);
String fileId = fileDao.selectFileByPostId(post.getNum()).get(0).getFileId();
System.out.println(fileId);
fileInfo.setFileId(fileId);
fileService.deleteFile(fileInfo);
fileDao.insertFile(fileInfo);
}그외에 수정, 삭제에도 추가해준다.
외래키가 설정되어있으므로, 파일부터 지우고 게시글을 지울수 있도록 한다.
modify 의 경우 파일id 를 얻어온 뒤 로직 수행
service package 에 FileService.java 도 생성해준다.
package com.example.post.service;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.List;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.example.post.dao.FileDao;
import com.example.post.model.FileInfo;
@Service
public class FileService {
@Autowired
FileDao fileDao;
public void insertFile(FileInfo fileInfo) {
fileDao.insertFile(fileInfo);
}
public FileInfo selectFileInfo(int fileId) {
return fileDao.selectFile(fileId);
}
public List<FileInfo> selectFileByPostId(String num) {
return fileDao.selectFileByPostId(num);
}
public void deleteFile(FileInfo fileInfo) throws IOException{
Files.delete(Paths.get(fileInfo.getFilePath() + "" + fileInfo.getStoredFilename()));
fileDao.deleteFile(fileInfo.getFileId());
}
}delete 의 경우 실제 파일도 지워야하므로, Files.delete(path) 를 통해 실제 경로 파일을 삭제하고, DB 경로도 지운다.
이제보니 Files.delete 결과를 제대로 받고나서 Dao 수행하는게 맞는거같다.
dao package 에도 Dao 를 하나더 만들어준다.
FileDao.java
package com.example.post.dao;
import java.util.List;
import org.mybatis.spring.SqlSessionTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;
import com.example.post.model.FileInfo;
@Repository
public class FileDao {
@Autowired
private SqlSessionTemplate sqlSession;
public void insertFile(FileInfo fileInfo) {
sqlSession.insert("FileMapper.insertFile", fileInfo);
}
public FileInfo selectFile(int fileId) {
return sqlSession.selectOne("FileMapper.selectFile", fileId);
}
public List<FileInfo> selectFileByPostId(String num) {
return sqlSession.selectList("FileMapper.selectFileByPostId", Integer.parseInt(num));
}
public void deleteFile(String fileId) {
sqlSession.delete("FileMapper.deleteFile", Integer.parseInt(fileId));
}
}새로운 mapper 를 작성할것이기에 FileMapper.~~~ 로 지정하였다.
아래는 mmapper-file.xml
<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="FileMapper">
<insert id="insertFile" parameterType="com.example.post.model.FileInfo">
INSERT INTO FILEINFO1 (postId, originalFilename, storedFilename, filePath, fileExt)
VALUES (#{postId}, #{originalFilename}, #{storedFilename}, #{filePath}, #{fileExt})
</insert>
<select id="selectFileByPostId" parameterType="int" resultType="com.example.post.model.FileInfo">
SELECT * FROM FILEINFO1 WHERE POSTID = #{num}
</select>
<select id="selectFile" parameterType="int" resultType="com.example.post.model.FileInfo">
SELECT * FROM FILEINFO1 WHERE FILEID = #{num}
</select>
<delete id="deleteFile" parameterType="int">
DELETE FROM FILEINFO1 WHERE fileId = #{fileId}
</delete>
</mapper>파일 관련 CRUD 작업
실제 jsp 에도 수정을 통해 파일 업로드/다운로드 로직을 추가해준다.
<c:when test="${status == 1}">
<form id="postForm" action="${pageContext.request.contextPath}/Modify?num=${post.num}" method="post" accept-charset="utf-8" enctype="multipart/form-data">
<label for="title">글번호:</label>
<input type="text" id="num" name="num" value="${post.num}" readonly><br><br>
<label for="title">제목:</label>
<input type="text" id="title" name="title" value="${post.title}"><br><br>
<label for="author">작성자:</label>
<input type="text" id="author" name="author" value="${post.author}"><br><br>
<label for="file">파일첨부:</label>
<input type="file" id="file" name="file"><br><br>
<label for="contents">내용:</label><br>
<textarea id="contents" name="contents">${post.contents}</textarea><br><br>
<input type="submit" value="수정">
</form>
...중략...
<script>
document.getElementById('postForm').addEventListener('submit', function(event) {
const fileInput = document.getElementById("file");
const maxSize = 10 * 1024 * 1024;
if (fileInput.files.length > 0){
if(fileInput.files[0].size > maxSize){
alert("10MB 이하만 업로드 가능합니다.")
event.preventDefault();
}
}
});
</script>Write.jsp, post.jsp 둘다 파일 업로드 내용을 추가시켜준다.
10MB 이상은 업로드가 불가하기에 eventListener 등록을 통해 미리 제출을 방지한다.