[Springboot] 34. 파일 업로드(2단계-DTO, Service, JSP 수정)

[Springboot] 34. 파일 업로드(2단계-DTO, Service, JSP 수정)

 

1. signUpDTO.java

package com.tenco.bank.dto;

import org.springframework.web.multipart.MultipartFile;

import com.tenco.bank.repository.model.User;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;

@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
@ToString
public class SignUpDTO {
	
	private String username; 
	private String password; 
	private String fullname;
	private MultipartFile mFile;
	private String originFileName;
	private String uploadFileName;
	
	
	// 2단계 로직 - User Object 반환 
	public User toUser() {
		return User.builder()
				.username(this.username)
				.password(this.password)
				.fullname(this.fullname)
				.originFileName(this.originFileName)
				.uploadFileName(this.uploadFileName)
				.build();
	} 
	
	
}

 

 

2. UserService.java - createUser()

	/**
	 * 회원 등록 서비스 기능
	 * 트랜잭션 처리  
	 * @param dto
	 */
	@Transactional // 트랜잭션 처리는 반드시 습관화 
	public void createUser(SignUpDTO dto) {
		int result = 0; 
		
		System.out.println(dto.getMFile().getOriginalFilename());
		
		if(!dto.getMFile().isEmpty()) {
			// 파일 업로드 로직 구현
			String[] fileNames = uploadFile(dto.getMFile());
			
			dto.setOriginFileName("aaaa.png");
			dto.setUploadFileName("1234_aaaa.png");
		}
		
		
		try {
			// 코드 추가 부분
			// 회원 가입 요청 시, 사용자가 던진 비밀번호 값을 암호화처리 해야함
			String hashpwd = passwordEncoder.encode(dto.getPassword());
			System.out.println("암호화 확인: "+hashpwd);
			//dto.setPassword(hashpwd);
			//result = userRepository.insert(dto.toUser());
		} catch (DataAccessException e) {
			throw new DataDeliveryException("중복 이름을 사용할 수 없습니다.", HttpStatus.INTERNAL_SERVER_ERROR);
		} catch (Exception e) {
			throw new RedirectException("알 수 없는 오류", HttpStatus.SERVICE_UNAVAILABLE);
		}
		if(result != 1) {
			throw new DataDeliveryException("회원가입 실패", HttpStatus.INTERNAL_SERVER_ERROR);
		}
	}

 

 

3. User 클래스 & User.xml

package com.tenco.bank.repository.model;

import java.sql.Timestamp;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;

@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
@ToString
public class User {
	private Integer id; 
	private String username; 
	private String password; 
	private String fullname; 
	private String originFileName;
	private String uploadFileName;
	private Timestamp createdAt;
}

<?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="com.tenco.bank.repository.interfaces.UserRepository">
	
	<!-- 반드시 세미콜론을 제거 해야 한다.   -->
	<!-- id는 매칭되어 있는 인터페이스에 메서드 명과 같아야 한다.  -->	
	<insert id="insert">
		insert into user_tb(username, password, fullname, origin_file_name, upload_file_name ) 
		values( #{username}, #{password}, #{fullname}, #{originFileName}, #{uploadFileName})
	</insert>
	
	<update id="updateById">
		update user_tb set username = #{username}, 
		                   password = #{password}, 
		                   fullname = #{fullname},
		                   where id = #{id} 
	</update>
	
	<delete id="deleteById">
		delete from user_tb where id = #{id}
	</delete>
	
	<select id="findById" resultType="com.tenco.bank.repository.model.User">
		select * from user_tb where id = #{id}
	</select>
	
	<select id="findAll" resultType="com.tenco.bank.repository.model.User">
		select * from user_tb
	</select>
	
	<select id="findByUsernameAndPassword" resultType="com.tenco.bank.repository.model.User" >
		select * from user_tb where username = #{username} and password = #{password}	
	</select>
	
	<select id="findByUsername" resultType="com.tenco.bank.repository.model.User" >
		select * from user_tb where username = #{username}
	</select>
	
</mapper>

select*from user_tb;

alter table user_tb
add column origin_file_name varchar(200);

alter table user_tb
add column upload_file_name varchar(200);

 

 

4. signUp.jsp

<%@ page language="java" contentType="text/html; charset=UTF-8" pageEncoding="UTF-8"%>

<!-- header.jsp  -->
<%@ include file="/WEB-INF/view/layout/header.jsp"%>

<!-- start of content.jsp(xxx.jsp)   -->
<div class="col-sm-8">
	<h2>회원 가입</h2>
	<h5>Bank App에 오신걸 환영합니다</h5>
	
	<form action="/user/sign-up" method="post" enctype="multipart/form-data"> 
		<div class="form-group">
			<label for="username">username:</label>
			<input type="text" class="form-control" placeholder="Enter username" id="username" name="username" value="야스오1"  >
		</div>
		<div class="form-group">
			<label for="pwd">Password:</label>
			<input type="password" class="form-control" placeholder="Enter password" id="pwd" name="password" value="asd123">
		</div>
		<div class="form-group">
			<label for="fullname">fullname:</label>
			<input type="text" class="form-control" placeholder="Enter fullname" id="fullname" name="fullname" value="바람검객">
		</div>
		 <div class="custom-file">
    	  <input type="file" class="custom-file-input" id="customFile" name="mFile">
      	  <label class="custom-file-label" for="customFile">Choose file</label>
     	</div>
		<div class="d-flex justify-content-end">
			<button type="submit" class="btn btn-primary mt-md-4">회원가입</button>
		</div>
	</form>


</div>
<!-- end of col-sm-8  -->
</div>
</div>
<!-- end of content.jsp(xxx.jsp)   -->

<script>
// Add the following code if you want the name of the file appear on select
$(".custom-file-input").on("change", function() {
  var fileName = $(this).val().split("\\").pop();
  $(this).siblings(".custom-file-label").addClass("selected").html(fileName);
});
</script>

<!-- footer.jsp  -->
<%@ include file="/WEB-INF/view/layout/footer.jsp"%>

 

 

*5. 파일 업로드 메서드 - uploadFile()

private String[] uploadFile(MultipartFile mFile) {
		// 파일 업로드 구현
		
		// 파일 용량 체크
		if(mFile.getSize() > Define.MAX_FILE_SIZE) {
			throw new DataDeliveryException("파일 크기는 20MB보다 클 수 없습니다.", HttpStatus.BAD_REQUEST);
		}
		
		// 서버 컴퓨터에 파일을 넣을 디렉토리가 있는지 검사
		String saveDirectory = Define.UPLOAD_FILE_DERECTORY;
		File directory = new File(saveDirectory);
		if(!directory.exists()) {
			directory.mkdirs();
		}
		
		// 파일 이름 생성 (파일명 중복 예방)
		String uploadFileName = UUID.randomUUID()+"_"+mFile.getOriginalFilename();
		
		// 파일 전체 경로 + 새로 생성한 파일명
		String uploadPath = saveDirectory+uploadFileName;
		File destination = new File(uploadPath);
		
		// 반드시 수행
		try {
			mFile.transferTo(destination);
		} catch (IllegalStateException | IOException e) {
			e.printStackTrace();
			throw new DataDeliveryException("파일 업로드 중에 오류가 발생했습니다.", HttpStatus.INTERNAL_SERVER_ERROR);
		}
		
		return new String[] {mFile.getOriginalFilename(), uploadFileName};
	}

	/**
	 * 서버 운영체제에 파일 업로드 기능
	 * MultipartFile etOriginFilename : 사용자가 작성한 파일 명
	 * uploadFileName : 서버 컴퓨터에 저장될 파일명
	 * @param mFile
	 * @return
	 */
	private String[] uploadFile(MultipartFile mFile) {
		// 파일 업로드 구현
		
		// 파일 용량 체크
		if(mFile.getSize() > Define.MAX_FILE_SIZE) {
			throw new DataDeliveryException("파일 크기는 20MB보다 클 수 없습니다.", HttpStatus.BAD_REQUEST);
		}
		
		// 서버 컴퓨터에 파일을 넣을 디렉토리가 있는지 검사
		String saveDirectory = Define.UPLOAD_FILE_DERECTORY;
		File directory = new File(saveDirectory);
		if(!directory.exists()) {
			directory.mkdirs();
		}
		
		// 파일 이름 생성 (파일명 중복 예방)
		String uploadFileName = UUID.randomUUID()+"_"+mFile.getOriginalFilename();
		
		// 파일 전체 경로 + 새로 생성한 파일명
		String uploadPath = saveDirectory+uploadFileName;
		System.out.println("------------------------------");
		System.out.println(uploadPath);
		System.out.println(uploadFileName);
		System.out.println("------------------------------");
		File destination = new File(uploadPath);
		
		// 반드시 수행
		try {
			mFile.transferTo(destination);
		} catch (IllegalStateException | IOException e) {
			e.printStackTrace();
			throw new DataDeliveryException("파일 업로드 중에 오류가 발생했습니다.", HttpStatus.INTERNAL_SERVER_ERROR);
		}
		
		return new String[] {mFile.getOriginalFilename(), uploadFileName};
	}