[Java]CSV 파일 생성 클래스 예시

| writeListToCSVFile 메서드를 호출하여 파일을 생성

- 해당 클래스는 제목행(header row)와 내용행(data row)를 각각 분리하여 생성한다.

- fileName : 경로를 포함한 파일이름(ex. /portal/upload/DATA_202305240323420290)

- headerlist : 제목행

- dataList : 내용행

- dataList 자체는 컬럼과 데이터로 짝지어 있지만 실제 파일 생성은 데이터만을 사용한다. 즉 제목행은 빨강,  파랑, 노랑이고 데이터 1, 데이터2, 데이터3이 쌓인다.

List<String> headers = Arrays.asList("빨강", "파랑", "노랑");
List<Map<String, String>> data = new ArrayList<>();
Map<String, String> row1 = new HashMap<>();
row1.put("컬럼1", "데이터1");
row1.put("컬럼2", "데이터2");
row1.put("컬럼3", "데이터3");
data.add(row1);

DataCSVUtil.writeListToCSVFile("output.csv", headers, data);​

package chungnam.jobutil;

import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.List;
import java.util.Map;

public class DataCSVUtil {

    public static void writeListToCSVFile(String fileName, List<String> headerList, List<Map<String, String>> dataList) {
        Writer writer = null;

        try {
            writer = new OutputStreamWriter(new FileOutputStream(fileName), "UTF-8");

            // Write BOM for UTF-8
            writer.write('\uFEFF');

            // Header row
            for (int i = 0; i < headerList.size(); i++) {
                writer.append(escapeSpecialCharacters(headerList.get(i)));
                if (i < headerList.size() - 1) {
                    writer.append(",");
                }
            }
            writer.append("\n");

            // Write the data rows
            for (Map<String, String> data : dataList) {
                int cellIndex = 0;
                for (String key : data.keySet()) {
                    String value = (String) data.values().toArray()[cellIndex++];
                    writer.append(escapeSpecialCharacters(value != null ? value : ""));
                    if (cellIndex < data.size()) {
                        writer.append(",");
                    }
                }
                writer.append("\n");
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (writer != null) {
                    writer.flush();
                    writer.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    private static String escapeSpecialCharacters(String data) {
        if (data == null) {
            return "";
        }
        String escapedData = data.replace("\"", "\"\"");
        if (data.contains(",") || data.contains("\n") || data.contains("\"")) {
            escapedData = "\"" + escapedData + "\"";
        }
        return escapedData;
    }
}