package common.util; import common.vo.CheckMessage; import common.vo.CommonFile; import common.vo.CommonFile.FileStatus; import common.vo.SystemCode.FileUploadPath; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import javax.servlet.http.HttpServletRequest; import java.io.File; import java.text.DecimalFormat; import java.time.LocalDate; import java.util.*; /** * @author 최정우 * @since 2019.11.13 * * File과 관련된 기능을 정의 해놓은 Util입니다. * 업로드, 다운로드, 파일읽기, 파일쓰기 * * 파일 정보만 가지고오기 -> multipartFileToCommonFile (multipartFileCheck기능 포함) * 파일 확장자 체크 -> multipartFileCheck * 특정 디렉토리에 파일 생성 -> fileCreate * * ※주요 활용 메소드 : fileUpload -> 파일 업로드 및 업로드 정보 반환 (fileCreate기능 및 multipartFileCheck 까지 포함되어 있음) */ public class FileUtil { /** * @author 최정우 * @since 2019.11.13 * * 파일 기본 정보 (여러개 파일) */ public static List<CommonFile> multipartFileToCommonFile (HttpServletRequest request) throws Exception { return multipartFileToCommonFile(request, null); } public static List<CommonFile> multipartFileToCommonFile (HttpServletRequest request, List<String> extensions) throws Exception { List<CommonFile> fileuploads = new ArrayList<CommonFile>(); MultipartHttpServletRequest multipart = (MultipartHttpServletRequest) request; Iterator<String> itr = multipart.getFileNames(); while (itr.hasNext()) { fileuploads.add(multipartFileToCommonFile(multipart.getFile(itr.next()), extensions)); } return fileuploads; } /** * @author 최정우 * @since 2019.11.13 * * 파일 기본 정보 (한개 파일) */ public static CommonFile multipartFileToCommonFile (MultipartFile multipartFile) throws Exception { return multipartFileToCommonFile(multipartFile, null); } public static CommonFile multipartFileToCommonFile (MultipartFile multipartFile, List<String> extensions) throws Exception { CommonFile commonFile = new CommonFile(); commonFile.setFileOriginName(getFileNameExceptExtension(multipartFile.getOriginalFilename())); commonFile.setFileExtension(getFileExtension(multipartFile.getOriginalFilename())); commonFile.setFileSize(multipartFile.getSize()); commonFile.setCheckMessage(multipartFileCheck(multipartFile, extensions)); return commonFile; } /** * @author 최정우 * @since 2019.11.13 * * 업로드 파일 체크 (여러개 파일 - extensions) */ public static Map<String, CheckMessage> multipartFileCheck (HttpServletRequest request, List<String> extensions) throws Exception { Map<String, CheckMessage> info = new HashMap<String, CheckMessage>(); MultipartHttpServletRequest multipart = (MultipartHttpServletRequest) request; Iterator<String> itr = multipart.getFileNames(); while (itr.hasNext()) { MultipartFile multipartFile = multipart.getFile(itr.next()); info.put(multipartFile.getName(), multipartFileCheck (multipartFile, extensions)); } return info; } /** * @author 최정우 * @since 2019.11.13 * * 업로드 파일 체크 (한개 파일 - extensions) */ public static CheckMessage multipartFileCheck (MultipartFile multipartFile, List<String> extensions) throws Exception { String orginFileName = multipartFile.getOriginalFilename(); // System.out.println("multipartFile name : " + multipartFile.getName()); //원 파일명이 없는 경우 처리(첨부가 되지 않은 input file type) if (orginFileName.equals("")) { return new CheckMessage(false, "파일명이 존재하지 않습니다"); } //파일 확장자 존재 유무 int index = orginFileName.lastIndexOf("."); if (index == -1) { return new CheckMessage(false, "파일에 확장자가 존재하지 않습니다"); } //파일 확장자 체크 String fileExt = orginFileName.substring(index + 1); //파일 확장자 체크 후, 컴펌 유무 boolean isConfirm = false; if (extensions != null && extensions.size() > 0) {//체크할 확장자가 있으면 -> 체킹 for (int i = 0; i < extensions.size(); i++) { if (extensions.get(i).toUpperCase().equals(fileExt.toUpperCase())) { isConfirm = true; break; } } } else {//체크할 확장자가 없으면 -> 컴펌 isConfirm = true; } if (isConfirm == false) { return new CheckMessage(false, "." + fileExt + " 파일은(는) 업로드가 불가능한 확장자입니다"); } return new CheckMessage(true, "업로드 가능한 파일 입니다"); } /** * @author 최정우 * @since 2019.11.17 * * 업로드 파일 정보 조회 (여러개 파일) */ public static List<CommonFile> fileUpload (HttpServletRequest request) throws Exception { return fileUpload(request, null); } public static List<CommonFile> fileUpload (HttpServletRequest request, List<String> extensions) throws Exception { List<CommonFile> fileUploads = new ArrayList<CommonFile>(); MultipartHttpServletRequest multipart = (MultipartHttpServletRequest) request; Iterator<String> multiFiles = multipart.getFileNames(); while (multiFiles.hasNext()) { MultipartFile multipartFile = multipart.getFile(multiFiles.next()); fileUploads.add(fileUpload(multipartFile, request, extensions)); } return fileUploads; } /** * @author 최정우 * @since 2019.11.17 * * 업로드 파일 정보 조회 (한개 파일) */ public static CommonFile fileUpload (MultipartFile multipartFile, HttpServletRequest request) throws Exception { return fileUpload(multipartFile, request, null); } public static CommonFile fileUpload (MultipartFile multipartFile, HttpServletRequest request, List<String> extensions) throws Exception { long startTime = System.currentTimeMillis();/*삭제 요망*/ //파일 정보 객체 CommonFile fileUpload = new CommonFile(); try { //프로젝트 내부 경로 String relativePath = FileUploadPath.MAIN_RELATIVE_PATH.getOSFileUploadPath(); //프로젝트 내부 경로 = 프로젝트 내부 경로 + /년도 relativePath += File.separator + LocalDate.now().getYear(); //프로젝트 내부 경로 = 프로젝트 내부 경로 + /년도 + /월 relativePath += File.separator + LocalDate.now().getMonthValue(); //절대 경로(프로젝트 내부 경로의 절대경로) String absolutePath = request.getSession().getServletContext().getRealPath(relativePath); //확장자를 제거한 원본 파일명 String originName = getFileNameExceptExtension(multipartFile.getOriginalFilename()); //새로 생성될 파일명 String maskName = StringUtil.getCreateKey("bigdata"); //원본 파일의 확장자 String extension = getFileExtension(multipartFile.getOriginalFilename()); //원본 파일의 파일크기(byte단위) long size = multipartFile.getSize(); //파일 정보 셋팅 (시작) fileUpload.setFileRelativePath(relativePath); fileUpload.setFileAbsolutePath(absolutePath); fileUpload.setFileOriginName(originName); fileUpload.setFileMaskName(maskName); fileUpload.setFileExtension(extension); fileUpload.setFileSize(size); fileUpload.setFileStatus(FileStatus.BEFORE_INSERT); fileUpload.setCheckMessage(new CheckMessage(true, "파일 정보 조회 성공")); //파일 정보 셋팅 (끝) } catch (Exception e) { fileUpload.setCheckMessage(new CheckMessage(false, "파일 정보 조회 에러", e.getMessage())); } //실제 디렉토리에 업로드 파일 생성 if (extensions != null && extensions.size() > 0) {//extensions가 존재하면 -> 파일 검사 -> 통과 시, 파일 생성 //파일 확장자 검사 fileUpload.setCheckMessage(multipartFileCheck(multipartFile, extensions)); if (fileUpload.getCheckMessage().getIsSuccess() == true) {//확장자 검사 통과 -> 성공 -> 파일 생성 fileCreate(fileUpload, multipartFile); long endTime = System.currentTimeMillis();/*삭제 요망*/ long totalTime = endTime - startTime;/*삭제 요망*/ System.out.println("업로드 경로 : " + fileUpload.getFileAbsolutePath() + ", 파일명 : " + multipartFile.getOriginalFilename() + "(" + fileUpload.getFileMaskName() + ") 파일 업로드 소요시간 : " + totalTime / 1000.0 + "초");/*삭제 요망*/ } else {//확장자 검사 통과 -> 실패 System.out.println(multipartFile.getOriginalFilename() + " 파일 확장자 검사 통과 실패");/*삭제 요망*/ } } else {//extensions가 존재하지 않으면 -> 파일 생성 fileCreate(fileUpload, multipartFile); long endTime = System.currentTimeMillis();/*삭제 요망*/ long totalTime = endTime - startTime;/*삭제 요망*/ System.out.println("업로드 경로 : " + fileUpload.getFileAbsolutePath() + ", 파일명 : " + multipartFile.getOriginalFilename() + "(" + fileUpload.getFileMaskName() + ") 파일 업로드 소요시간 : " + totalTime / 1000.0 + "초");/*삭제 요망*/ } return fileUpload; } /** * @author 최정우 * @since 2019.11.17 * * 실제 디렉토리에 업로드 파일 생성 */ public static void fileCreate (CommonFile fileupload, MultipartFile multipartFile) throws Exception { try { //디렉토리 만들기. File dir = new File(fileupload.getFileAbsolutePath()); boolean isExistDir = true; if (dir.exists() == false) { isExistDir = dir.mkdirs(); } //디렉토리가 존재할 때 if (isExistDir == true) { //디렉토리에 쓰기 권한이 있을 때 if (dir.canWrite() == true) { //디렉토리 경로에 파일 생성 multipartFile.transferTo(new File(fileupload.getFileAbsolutePath(), fileupload.getFileMaskName() + "." + fileupload.getFileExtension())); fileupload.setCheckMessage(new CheckMessage(true, "파일 활용 가능")); } else { fileupload.setCheckMessage(new CheckMessage(false, "쓰기 권한이 없습니다")); } } else { fileupload.setCheckMessage(new CheckMessage(false, "디렉토리 생성 불가")); } } catch (Exception e) { fileupload.setCheckMessage(new CheckMessage(false, "파일 업르드 에러", e.getMessage())); } } /** * @author 최정우 * @since 2019.11.17 * * 실제 디렉토리에 업로드 파일 생성 */ public static void fileRemove (String fileFullPath) throws Exception { //commonFile.getFileAbsolutePath() + File.separator + commonFile.getFileMaskName() + "." + commonFile.getFileExtension() File file = new File(fileFullPath); if (file.exists()) { file.delete(); } } /** * @author 최정우 * @since 2019.11.17 * * 파일 확장자 얻기 */ public static String getFileExtension (String fileName) { if (StringUtil.isEmpty(fileName) == true) { return null; } else { int index = StringUtil.lastIndexOf(fileName, "."); if (index > -1) { return StringUtil.lowerCase(fileName.substring(index + 1)); } else { return null; } } } /** * @author 최정우 * @since 2019.11.17 * * 확장자를 제거한 원본 파일명 얻기 */ public static String getFileNameExceptExtension (String fileName) { if (StringUtil.isEmpty(fileName) == true) { return fileName; } else { int index = StringUtil.lastIndexOf(fileName, "."); if (index > -1) { return fileName.substring(0, index); } else { return fileName; } } } /** * @author 최정우 * @since 2019.11.17 * * File의 용량(byte)을 각 단위에 맞게 변형 */ public static String convertFileSize (String bytes) { String retFormat = "0"; Double size = Double.parseDouble(bytes); String[] s = { "bytes", "KB", "MB", "GB", "TB", "PB" }; if (bytes != null && bytes.length() > 0 && !bytes.equals("0") ) { int idx = (int) Math.floor(Math.log(size) / Math.log(1024)); DecimalFormat df = new DecimalFormat("#,###.##"); double ret = ((size / Math.pow(1024, Math.floor(idx)))); retFormat = df.format(ret) + " " + s[idx]; } else { retFormat += " " + s[0]; } return retFormat; } }