[Java] 엑셀 파일 읽고 쓰기 (2탄)

  엑셀 파일 읽고 쓰기

이번에 프로젝트를 하면서 쓰게된 엑셀 파일을 읽고 쓰는 방법에 대하 포스팅해보려고 합니다. 이전에는 주로 띄어쓰기나 콤마( , )로 구분된 텍스트 파일을 읽어서 데이터를 처리하거나 반대로 쓰는 것을 했었습니다. 이번에 엑셀 파일로 저장된 데이터를 사용하게 되었고 이전에 방식으로는 지겨우니 엑셀로 바로 처리하는 것을 한번 배워보도록 하겠습니다.

이전에 이미 읽는 방법에 대해선 포스팅을 남겼습니다. 이번 포스팅은 쓰는 방법을 알아보도록 하겠습니다.

  코드 

일단 저는 데이터를 읽고 그 데이터를 사용할 수 있도록 가공하고 새로운 엑셀 파일에 저장까지 할 계획입니다. 원래는 가공하고 바로 DB로 저장시켜도 되지만 이때 아니면 해볼 기회가 없을 것 같아서 전부 한번 해보겠습니다.

writeExcel 메소드

2007버전 이후는 XSSFWorkbook을 사용하시면 되며 밑에 현재 sheet와 같은 것은 앞에 H->X로만 바꿔주면 사용이 가능하실 겁니다. 이번에도 그렇게 어려운 부분은 없을 것입니다. 

    public void writeexcel(ArrayList<theaterVo> list) {
 
        String path = "transferTheaterList.xls"; // 저장할 파일 경로
        try {
            File file = new File(path);
            FileOutputStream fileout = new FileOutputStream(file); 
 
            HSSFWorkbook hworkbook = new HSSFWorkbook();
            HSSFSheet sheet = hworkbook.createSheet("theaterList"); // sheet 생성
            
            HSSFRow curRow;
            
            int row = list.size(); // list 크기
            for (int i = 0; i < row; i++) {
                curRow = sheet.createRow(i); // row 생성
                curRow.createCell(0).setCellValue(list.get(i).getTheaterno()); // row에 각 cell에 저장
                curRow.createCell(1).setCellValue(list.get(i).getBrandno());
                curRow.createCell(2).setCellValue(list.get(i).getTheatername());
                curRow.createCell(3).setCellValue(list.get(i).getTheateraddress());
                curRow.createCell(4).setCellValue(list.get(i).getTheaterxgps());
                curRow.createCell(5).setCellValue(list.get(i).getTheaterygps());
            }
            hworkbook.write(fileout); // 파일 쓰기
            fileout.close();
        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

getTheaterxgps() 와 getTeaterygps()는 제가 필요해서 만든 부분이기 때문에 빼도 상관없습니다.

이제 실행을 시켜보도록 하겠습니다.

아래와 같이 파일이 생성되었습니다.

파일의 경로 위치는 읽는 것 쓰는 것 모두 사용자가 프로젝트를 만든 파일 안에 넣어서 사용했습니다.

파일 경로는 알아서 잘 설정하시면 됩니다.

그리고 그 파일을 열어 보면 아래와 같이 정상적으로 필요한 정보만 들어간 것을 알 수 있습니다.

20개만 넣어봤습니다. 그리고 제일 뒤에 좌표는 제가 임의로 넣어놓은 입니다.

노란색 박스는 Sheet 이름이입니다 (writeExcel 메소드 9 줄 코드)

아래는 모든 코드 입니다.

만약 문제가 있거나 더 좋은 코드가 있다면 댓글 달아주시면 참고 하겠습니다.

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLEncoder;
import java.util.ArrayList;
 
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
 
public class TransferAddress {
 
    public ArrayList<theaterVo> readexcel() {
        String path = "./TheaterList.xls"; // 읽을 파일 경로
        ArrayList<theaterVo> theaterlist = new ArrayList<theaterVo>(); // 데이터를 담을 list
 
        try {
            File file = new File(path);
            FileInputStream inputStream = new FileInputStream(file);
            HSSFWorkbook hworkbook = new HSSFWorkbook(inputStream); // 2007 이전 버전(xls파일)
            // XSSFWorkbook xworkbook = new XSSFWorkbook(inputStream); // 2007 이후 버전(xlsx파일)
 
            HSSFSheet curSheet; // 현재 sheet
            HSSFCell curCell; // 현재 cell
            HSSFRow curRow; // 현재 row
 
            int sheetNumber = hworkbook.getNumberOfSheets(); // 엑셀 Sheet 총 갯수
            // System.out.println("sheetNumber : "+sheetNumber); Sheet 갯수 확인
            while (sheetNumber != 0) {
                sheetNumber--;
 
                curSheet = hworkbook.getSheetAt(sheetNumber);
                int row = curSheet.getPhysicalNumberOfRows();
                // System.out.println(row); 현재 sheet의 row 갯수 확인
                for (int i = 1; i < 20; i++) {
                    theaterVo vo = new theaterVo();
                    curRow = curSheet.getRow(i);
 
                    // vo의 setter를 이용해 담고 있습니다.
                    // 여기서는 그대로 담는 것이 아니라 자료형에 맞춰서 형변환을 해주셔야 합니다.
                    vo.setTheaterno(Integer.valueOf(String.valueOf(curRow.getCell(2))));
                    vo.setTheatername(String.valueOf(curRow.getCell(3)));
                    vo.setTheateraddress(String.valueOf(curRow.getCell(18)));
                    // 아래의 brandno는 필요한 부분이라 따로 제가 만든 메소드 이기 떄문에 아래 2줄은 빼셔도 됩니다.
                    int brandno = TransferAddress.brandno(String.valueOf(curRow.getCell(14)));
                    vo.setBrandno(brandno);
                    //String address = TransferAddress.transfer(String.valueOf(curRow.getCell(18)));
                    vo.setTheaterxgps("100"); // 임의로 넣은 값
                     vo.setTheaterygps("100"); // 임의로 넣은 값
 
                    theaterlist.add(vo);
                }
            }
 
        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
 
        return theaterlist;
    }
 
    public void writeexcel(ArrayList<theaterVo> list) {
 
        String path = "transferTheaterList.xls"; // 저장할 파일 경로
        try {
            File file = new File(path);
            FileOutputStream fileout = new FileOutputStream(file); 
 
            HSSFWorkbook hworkbook = new HSSFWorkbook();
            HSSFSheet sheet = hworkbook.createSheet("theaterList"); // sheet 생성
            
            HSSFRow curRow;
            
            int row = list.size(); // list 크기
            for (int i = 0; i < row; i++) {
                curRow = sheet.createRow(i); // row 생성
                curRow.createCell(0).setCellValue(list.get(i).getTheaterno()); // row에 각 cell에 저장
                curRow.createCell(1).setCellValue(list.get(i).getBrandno());
                curRow.createCell(2).setCellValue(list.get(i).getTheatername());
                curRow.createCell(3).setCellValue(list.get(i).getTheateraddress());
                curRow.createCell(4).setCellValue(list.get(i).getTheaterxgps());
                curRow.createCell(5).setCellValue(list.get(i).getTheaterygps());
            }
            hworkbook.write(fileout); // 파일 쓰기
            fileout.close();
        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }
 
    // 1. 메가박스
    // 2. CGV
    // 3. 롯데시네마
    // 4. 이외 시네마
    public static int brandno(String brandname) {
        int result = 0;
 
        if (brandname.equals("")) {
            return result;
        } else if (brandname.contains("메가박스")) {
            result = 1;
        } else if (brandname.contains("CJ")) {
            result = 2;
        } else if (brandname.contains("롯데시네마")) {
            result = 3;
        } else {
            result = 4;
        }
        return result;
    }
 
    public static void main(String[] args) {
        TransferAddress transfer = new TransferAddress();
        ArrayList<theaterVo> list = new ArrayList<theaterVo>();
        list = transfer.readexcel();
        
        transfer.writeexcel(list);
        
        int size = list.size();
        for (int i = 0; i < size; i++) {
            System.out.println(list.get(i).toString());
        }
    }
}