과제 - POI 라이브러리

Apache POI (Poor Obfuscation Implementation)​​ Java 프로그래밍 언어를 사용하여 Microsoft Office형식의 문서를 읽고 쓰는데 사용되는 라이브러리

1. 라이브러리 파일을 다운받아 수동으로 프로젝트에 라이브러리로 연결​
2. maven 혹은 gradle에서 의존성(dependency)을 추가하여 사용하는 방법

반드시 필요한 jar 파일은 ​

poi.jar, ​
poi-exelant.jar, ​
poi-ooxml.jar, ​
poi-ooxml-ful.jar, ​

ooxml-lib 폴더의 ​
commons-compress.jar, ​
xmlbeans.jar, ​
​
lib 폴더의 ​
commons-collections.jar, ​
commons-math.jar, ​
commons-io.jar ,

log4j-api.jar

 

log4j-core.jar

POI라이브러리 에는 HSSF, XSSF, SXSSF 3가지 클래스가 존재​
HSSF : 엑셀 97 ~ 2003버전 xls​
XSSF : 엑셀 2007버전이상  xlsx​
SXSSF : 가장 최근에 나온 성능개선버전 

엑셀 쓰기 / 읽기

package poiTest;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;

import org.apache.poi.ss.usermodel.FormulaEvaluator;
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.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

public class Xlsx {
    
    public static void main(String[] args) throws IOException {
        Xlsx x = new Xlsx();
        x.write("d:/d_other", "학생.xlsx");
        
        x.reader("d:/d_other", "학생.xlsx");
    }

    // 엑셀 파일을 생성하고 데이터를 쓰는 메서드
    // createSheet(), createRow(index), createCell(index), setCellValue(value값)
    // 파일저장 workBook객체.write(FileOutputStream객체);
    public void write(String path, String fileName) throws IOException{
        Workbook workBook = new XSSFWorkbook(); // 엑셀 워크북 생성

        // 시트 생성
        // createSheet()
        Sheet sheet0 = workBook.createSheet();
        Sheet sheet1 = workBook.createSheet();
        Sheet sheet2 = workBook.createSheet();
        Sheet sheet3 = workBook.createSheet();

        // 행 생성
        // createRow(index);
        Row row0 = sheet0.createRow(0);
        Row row1 = sheet0.createRow(1);
        Row row2 = sheet0.createRow(2);
        Row row3 = sheet0.createRow(3);
        Row row4 = sheet0.createRow(4);

        
        // 열 생성
        // createCell(index)
        
        // 값 입력
        // setCellValue(value값)
        row0.createCell(0).setCellValue("번호");
        row0.createCell(1).setCellValue("이름");
        row0.createCell(2).setCellValue("나이");
        row0.createCell(3).setCellValue("주소");

        // 데이터 행에 값 쓰기
        row1.createCell(0).setCellValue("1");
        row1.createCell(1).setCellValue("홍길동");
        row1.createCell(2).setCellValue("19");
        row1.createCell(3).setCellValue("대전");

        row2.createCell(0).setCellValue("2");
        row2.createCell(1).setCellValue("이순신");
        row2.createCell(2).setCellValue("24");
        row2.createCell(3).setCellValue("서울");

        row3.createCell(0).setCellValue("3");
        row3.createCell(1).setCellValue("김영희");
        row3.createCell(2).setCellValue("20");
        row3.createCell(3).setCellValue("청주");
        
        row4.createCell(0).setCellValue("4");
        row4.createCell(1).setCellValue("김철수");
        row4.createCell(2).setCellValue("20");
        row4.createCell(3).setCellValue("대전");

        // 파일 출력 스트림 생성 및 엑셀 파일로 저장
        FileOutputStream fout = new FileOutputStream(path+"/"+fileName);
        
        // 파일저장 workBook객체.write(FileOutputStream객체);
        workBook.write(fout);
        fout.close();
    }
    
    // 엑셀 파일을 읽어서 데이터를 출력하는 메서드
    // getNumberOfSheets(), getSheetAt(index)
    // getPhysicalNumberOfRows(), getPhysicalNumberOfCells()
    // getRow(index), getCell(index)
    // getCellType()
    // get[타입]CellValue()
    public void reader(String path, String name) {
        try {
            File file = new File(path+"/"+name);
            FileInputStream fis = new FileInputStream(file);
            XSSFWorkbook workbook = new XSSFWorkbook(fis); // 엑셀 워크북 생성
            
            // 시트 갯수 가져오기
            // getNumberOfSheets()
            int sheetNum = workbook.getNumberOfSheets();
            
            // 각 시트별로 데이터 출력
            for(int s = 0; s < sheetNum; s++) {
                // 시트 가져오기
            	// getSheetAt(index)
                XSSFSheet sheet = workbook.getSheetAt(s);
                
                // 존재하는 행 갯수 가져오기 (반복문을 위해 존재하는 row 갯수를 가져옴)
                // getPhysicalNumberOfRows()
                int rows = sheet.getPhysicalNumberOfRows();
                
                // 각 행별로 데이터 출력
                for(int r = 0 ; r < rows ; r++) {
                    // 행 가져오기
                	// getRow(index)
                    XSSFRow row = sheet.getRow(r);
                    
                    // 행의 셀 갯수 가져오기 (반복문을 위해 존재하는 cell 갯수를 가져옴)
                    // getPhysicalNumberOfCells()
                    int cells = row.getPhysicalNumberOfCells(); 
                    
                    System.out.print("|    "+r+"    |");
                    
                    for(int c = 0 ; c < cells; c++) {
                    	// 각 셀별로 데이터 출력
                    	// getCell(index)
                        XSSFCell cell = row.getCell(c);
                        
                        String value = "";
                        if(cell!=null) {
                            // 셀 타입 확인하기
                        	// getCellType()
                        	// STRING, NUMERIC, FORMULA, BLANK, ERROR
                        	
                            switch(cell.getCellType()) {
                            case STRING: // 텍스트
                            	// getStringCellValue()
                                value = cell.getStringCellValue();
                                break;
                            case NUMERIC: // 숫자
                            	// getNumericCellValue()
                                value = cell.getNumericCellValue()+"";
                                break;
                            case FORMULA: // = 붙은 계산식 처리
								// getCachedFormulaResultType() 결과타입 불러올때 
                            	//value = cell.getCellFormula() 를 사용하면 =123+100 일때 123+100 을 표시한다. 따라서 계산을 하려면 아래코드를 쓴다. 
                            	// 결과값이 필요할 때 FormulaEvaluator객체를 사용한다
								FormulaEvaluator formulaEval = workbook.getCreationHelper().createFormulaEvaluator();
								value = formulaEval.evaluate(cell).formatAsString();
								break;
                            case BLANK: // 빈칸
                            	// getBooleanCellValue()
                                value = cell.getBooleanCellValue()+"";
                                break;
                            case ERROR: // 에러처리
                            	// getErrorCellValue()
                                value = cell.getErrorCellValue()+"";
                                break;
                            }
                        }
                        // 셀의 값 출력
                        System.out.print("\t"+value+"\t|");
                    }
                    System.out.println(); // 한 행 출력 후 줄 바꿈
                }
            }
        } catch(FileNotFoundException e) {
            e.printStackTrace();
        } catch(IOException e){
            e.printStackTrace();
        }
    }
}

 

활용한 예제 - 전화번호부 관리

package poiTest;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

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.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;


/*
	load()와 saveData()메서드에서 POI를 활용한 데이터 저장/출력을 함
 */
public class PhoneBookTest {
	// 띄어쓰기를 쓰면 scanner에서 오류 나중에 알려줌
	Scanner scan;
	private HashMap<String, Phone> phoneMap;
	// 데이터 변경 여부
	private boolean changeChk;

	// 생성자
	public PhoneBookTest() {
		phoneMap = (HashMap<String, Phone>) load();
		scan = new Scanner(System.in);
		changeChk = false;
	}

	public static void main(String[] args) {
		PhoneBookTest p = new PhoneBookTest();
		p.startProgram();
	}

	public void startProgram() {
		// 시작 시 데이터가 있다면 데이터를 phoneMap에 저장함
		load();

		while (true) {
			int choice = menu();
			System.out.println();
			switch (choice) {
			case 1:
				insert();
				break;
			case 2:
				update();
				break;
			case 3:
				delete();
				break;
			case 4:
				search();
				break;
			case 5:
				print();
				break;
			case 6:
				saveData();
				break;
			case 0:
				saveData();
				System.out.println("프로그램을 종료합니다...");
				return;
			default:
				System.out.println("잘못된 입력입니다.");
			}
		}
	}

	private int menu() {
		System.out.println();
		System.out.println("\t메  뉴");
		System.out.println();
		System.out.println("1. 전화번호 등록");
		System.out.println("2. 전화번호 수정");
		System.out.println("3. 전화번호 삭제");
		System.out.println("4. 전화번호 검색");
		System.out.println("5. 전화번호 전체 출력");
		System.out.println("6. 전화번호 저장");
		System.out.println("0. 프로그램 종료");
		System.out.print("메뉴선택 : ");
		return scan.nextInt();
	}

	// 초기에 엑셀 데이터가 있다면 phoneMap에 저장
	// 이름 전화번호 주소
	private Map<String, Phone> load() {
		File file = new File("D:/d_other/phonedata.xlsx");
		// 데이터가 없다면 종료
		if (!file.exists()) {
			return new HashMap<String, Phone>();
		}
		HashMap<String, Phone> pMap = new HashMap<String, Phone>();

		FileInputStream fin =null;
		try {
			fin = new FileInputStream(file);
			
			XSSFWorkbook workbook = new XSSFWorkbook(fin);
			// 시트 가져오기 ( 첫번째 시트에 데이터 저장 )
			XSSFSheet sheet = workbook.getSheetAt(0);
			// 존재하는 행 갯수 가져오기
			int rows = sheet.getPhysicalNumberOfRows();
			
			// 각 행별로 데이터 출력
			// 첫번째 row는 인덱스임으로 포함시키지않음
			for(int r = 1 ; r < rows ; r++) {
				// row번호가 변할 때마다 데이터 초기화
				String name = "";
				Phone phone = new Phone();
				
				// 행 가져오기
				XSSFRow row = sheet.getRow(r);
				
				int cells = row.getPhysicalNumberOfCells(); // 행의 셀 갯수 가져오기
				
				for(int c = 0 ; c < cells; c++) {
					XSSFCell cell = row.getCell(c); // 셀 값 가져오기
					
					String value = "";
					if(cell!=null) {
						// 모든 타입이 String이기때문에 String으로 가져옴
						value = cell.getStringCellValue();
					}
					// 셀의 값 저장
					switch (c) {
					// c = 0 이름
					case 0:
						name = value;
						phone.setName(value);
						break;
					// c = 1 전화번호
					case 1:
						phone.setTelNo(value);
						break;
					// c = 2 주소
					case 2:
						phone.setAddress(value);
						break;
					default:
						break;
					}
				}
				// 해당 phone 객체에 데이터를 모두 저장한 뒤에 하나의 데이터를 map에 저장
				// 맵에 값을 저장
				pMap.put(name, phone);
			}
			
		} catch (Exception e) {
			// TODO: handle exception
		}
		
			 
		return pMap;
	}

	// 데이터 저장
	// 엑셀로 데이터 저장함
	private void saveData() {
		// true값은 변경
		if (!changeChk) {
			System.out.println("변경할 것이 없습니다");
			return;
		}
		// 해당 엑셀을 가져옴
		File file = new File("D:/d_other/phonedata.xlsx");
		FileOutputStream fout = null;
		XSSFWorkbook workbook = new XSSFWorkbook();
		Sheet sheet = workbook.createSheet();
		
		try {
			fout = new FileOutputStream(file);
			
			// 첫번째 row는 인덱스
			Row row0 = sheet.createRow(0);
			row0.createCell(0).setCellValue("이름");
			row0.createCell(1).setCellValue("전화번호");
			row0.createCell(2).setCellValue("주소");
			
			// 1부터 데이터를 찾음
			int rowNum = 1;
			// 데이터 입력
			for(String name : phoneMap.keySet()) {
				// 이름을 꺼내서 저장
				Phone phone = phoneMap.get(name);
				// rowNum 값을 저장한 뒤에 나중에 +1을 해줌
				Row row = sheet.createRow(rowNum++);
				
				String telNo = phone.getTelNo();
				String addr = phone.getAddress();
				
				// 해당 cell번호에 해당하는 값에 데이터 입력
				row.createCell(0).setCellValue(name);
				row.createCell(1).setCellValue(telNo);
				row.createCell(2).setCellValue(addr);
			}
			workbook.write(fout); // workbook을 FileOutputStream을 사용하여 파일에 씁니다.
			System.out.println("저장되었습니다");
			
		} catch (FileNotFoundException e1) {
			// TODO Auto-generated catch block
			e1.printStackTrace();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} finally {
			if(fout!=null) try {
				fout.close(); // 스트림을 닫습니다.
			} catch (Exception e2) {
				// TODO: handle exception
			}
		}
	}

	
	
	
	// 변화 XX
	private void insert() {
		System.out.println("새롭게 등록할 전화번호 정보를 입력하세요.");
		System.out.print("이름 >> ");
		String name = scan.next();
		if (phoneMap.containsKey(name)) {
			System.out.println("'" + name + "'은 이미 등록된 사람입니다.");
			return;
		}
		System.out.print("전화번호 >> ");
		String telNo = scan.next();
		scan.nextLine(); // 입력 버퍼 엔터 비움
		System.out.print("주소 >> ");
		String address = scan.nextLine();
		Phone p = new Phone();
		p.setName(name);
		p.setTelNo(telNo);
		p.setAddress(address);
		phoneMap.put(name, p);
		System.out.println("'" + name + "' 전화번호 등록 완료!!");
		changeChk = true;
	}

	private void update() {
		System.out.println("변경할 이름을 입력해주세요");
		String name = scan.next();
		if (!phoneMap.containsKey(name)) {
			System.out.println("'" + name + "'은 등록되지 않은 사람입니다.");
			return;
		}
		System.out.print("전화번호 >> ");
		String telNo = scan.next();
		scan.nextLine(); // 입력 버퍼 엔터 비움
		System.out.print("주소 >> ");
		String address = scan.nextLine();
		Phone p = new Phone();
		p.setName(name);
		p.setTelNo(telNo);
		p.setAddress(address);
		phoneMap.put(name, p);
		System.out.println("'" + name + "' 전화번호 변경 완료!!");
		changeChk = true;
	}

	private void delete() {
		System.out.println("삭제할 이름을 입력해주세요");
		String name = scan.next();
		if (!phoneMap.containsKey(name)) {
			System.out.println("'" + name + "'은 등록되지 않은 사람입니다.");
			return;
		}
		phoneMap.remove(name);
		System.out.println("'" + name + "' 삭제 완료!!");
		changeChk = true;
	}

	private void search() {
		System.out.println("검색할 이름을 입력해주세요");
		String name = scan.next();
		if (!phoneMap.containsKey(name)) {
			System.out.println("조회 결과가 없습니다.");
			return;
		}
		System.out.println("----------------------------------------");
		System.out.println("이름\t전화번호\t\t주소");
		System.out.println("----------------------------------------");
		System.out.println(name + "\t" + phoneMap.get(name).getTelNo() + "\t" + phoneMap.get(name).getAddress());
		System.out.println("----------------------------------------");
	}

	private void print() {
		System.out.println();
		System.out.println("----------------------------------------");
		System.out.println("번호\t이름\t전화번호\t\t주소");
		System.out.println("----------------------------------------");
		int count = 0;
		if (phoneMap.size() == 0) {
			System.out.println("등록된 전화번호 정보가 하나도 없습니다...");
		} else {
//			for(String name : phoneMap.keySet()) {
//				Phone phone = phoneMap.get(name);
//				String telNo = phone.getTelNo();
//				String address = phone.getAddress();
//				System.out.println(++count + "\t" + name + "\t" + telNo + "\t" + address);
//			}

			for (Map.Entry<String, Phone> entry : phoneMap.entrySet()) {
				String name = entry.getKey();
				Phone phone = entry.getValue();
				String telNo = phone.getTelNo();
				String address = phone.getAddress();
				System.out.println(++count + "\t" + name + "\t" + telNo + "\t" + address);
			}
		}
		System.out.println("----------------------------------------");
		System.out.println("출력 끝...");
		System.out.println();
	}

}

class Phone implements Serializable {
	private String name;
	private String address;
	private String telNo;

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	public String getAddress() {
		return address;
	}

	public void setAddress(String address) {
		this.address = address;
	}

	public String getTelNo() {
		return telNo;
	}

	public void setTelNo(String telNo) {
		this.telNo = telNo;
	}

}