ExcelUtil

ExcelUtil.java

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.DateUtil;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

public class ExcelUtil {
	@SuppressWarnings("unchecked")
	// 엑셀 파일로부터 데이터를 추출한 뒤 ArrayList로 반환하는 함수
	public static Object getExcelData(File excelFile, String modelType) {
		Workbook workbook = null;
		Sheet workSheet = null;
		Row row = null;
		String cellData = "";
		Date cellDate = new Date();
		boolean cellBooleanValue = false;
		Cell cell = null;
		Object returnObject = null;
		Object tempObject = null;

		// modelType에 따라서 반환되는 returnObject의 타입을 변경해준다.
		if (modelType.equals("YourModelClassName"))
			returnObject = new ArrayList<YourModelClassName>();

		FileInputStream fileInputStream = null;
		SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

		try {
			// set Workbook
			fileInputStream = new FileInputStream(excelFile);
			// MS office 2007 이상 버전의 엑셀 파일의 경우에는 XSSFWorkbook으로 초기화해줘야 한다.
			if (excelFile.getName().toLowerCase().endsWith("xlsx"))
				workbook = new XSSFWorkbook(fileInputStream);
			// office 2003까지 지원하는 형식의 엑셀파일은 HSSFWorkbook으로 초기화한다.
			else
				workbook = new HSSFWorkbook(fileInputStream);

			workSheet = workbook.getSheetAt(0);
			int rowSize = workSheet.getLastRowNum() + 1;

			//첫 번째 줄은 일반적으로 컬럼의 이름이 들어가기 때문에 생략한다.
			for (int i = 1; i < rowSize; i++) {
				row = workSheet.getRow(i);
				cellData = "";
				cellDate = new Date();
				cellBooleanValue = false;

				if (modelType.equals("YourModelClassName"))
					tempObject = new YourModelClassName();

				for (int j = 0; j < row.getLastCellNum(); j++) {
					cell = row.getCell(j);

					if (cell != null && cell.getCellType() != Cell.CELL_TYPE_BLANK) {
						switch (cell.getCellType()) {
						case Cell.CELL_TYPE_STRING:
							cellData = cell.getStringCellValue();
							break;
						// 들어있는 데이터가 날짜 형식인 경우에도 CELL_TYPE_NUMERIC이 반환된다.
						// 숫자 데이터와 날짜 데이터를 구분하는 코드는 미구현
						case Cell.CELL_TYPE_NUMERIC:
							System.out.println(DateUtil.isCellDateFormatted(cell));
							cellDate = DateUtil.getJavaDate(cell.getNumericCellValue());
							break;

						case Cell.CELL_TYPE_BOOLEAN:
							cellBooleanValue = cell.getBooleanCellValue();
							break;
						}
					}

					if (modelType.equals("YourModelClassName")) {
						// 모델 클레스에 맞게 tempObject 객체에 값을 넣어준다.
						// 순서는 불러온 엑셀 시트의 컬의 순서이다.
						switch (j) {
						case 0:
							((YourModelClassName) tempObject).setName(cellData);
							break;
						case 1:
							((YourModelClassName) tempObject).setDate(cellDate);
							break;
						case 2:
							((YourModelClassName) tempObject).setIstrue(cellBooleanValue);
							break;
						}
					}
				}
				if (modelType.equals("YourModelClassName")) {
					((ArrayList<YourModelClassName>) returnObject).add((YourModelClassName) tempObject);
				}
			}

		} catch (Exception e) {
			e.printStackTrace();
		}

		return returnObject;
	}

	// MultipartFile을 File 객체로 변환해주는 함수
	public static File changeMultipartFileToFile(MultipartFile multipartFile) {
		File convertFile = new File(multipartFile.getOriginalFilename());
		try {
			convertFile.createNewFile();
			FileOutputStream fileOutputStream = new FileOutputStream(convertFile);
			fileOutputStream.write(multipartFile.getBytes());
			fileOutputStream.close();
			return convertFile;
		} catch (Exception e) {
			e.printStackTrace();
			return null;
		}
	}
	
	// 엑셀파일 확장자를 검사하는 함수
	public static boolean checkExcelTypeWithName(String fileName) {
		String[] excelFileType = { "xlsx", "xlsm", "xlsb", "xltx", "xltm", "xls", "xlt", "csv" };

		for (String typeName : excelFileType) {
			if (fileName.toLowerCase().endsWith(typeName)) {
				return true;
			}
		}
		return false;
	}
}