#37 java: covid - excel exporter
This commit is contained in:
47
java/api/src/main/java/kr/covid/scraper/ExcelExporter.java
Normal file
47
java/api/src/main/java/kr/covid/scraper/ExcelExporter.java
Normal file
@@ -0,0 +1,47 @@
|
||||
package kr.covid.scraper;
|
||||
|
||||
import org.apache.poi.ss.usermodel.Cell;
|
||||
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 java.io.FileOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.util.List;
|
||||
|
||||
public class ExcelExporter {
|
||||
|
||||
public static void exportToExcel(String date, List<CovidStatus> covidStatusList, String filename) throws IOException {
|
||||
Workbook workbook = new XSSFWorkbook();
|
||||
Sheet sheet = workbook.createSheet("코로나 현황");
|
||||
|
||||
Row headerRow = sheet.createRow(0);
|
||||
String[] headers = {"시도", "합계", "국내발생", "해외유입", "확진환자", "사망자", "발생률"};
|
||||
|
||||
for (int i = 0; i < headers.length; i++) {
|
||||
Cell cell = headerRow.createCell(i);
|
||||
cell.setCellValue(headers[i]);
|
||||
}
|
||||
|
||||
for (int i = 0; i < covidStatusList.size(); i++) {
|
||||
CovidStatus covidStatus = covidStatusList.get(i);
|
||||
|
||||
Row row = sheet.createRow(i + 1);
|
||||
|
||||
row.createCell(0).setCellValue(covidStatus.getRegion());
|
||||
row.createCell(1).setCellValue(covidStatus.getTotal());
|
||||
row.createCell(2).setCellValue(covidStatus.getDomestic());
|
||||
row.createCell(3).setCellValue(covidStatus.getAbroad());
|
||||
row.createCell(4).setCellValue(covidStatus.getConfirmed());
|
||||
row.createCell(5).setCellValue(covidStatus.getDeaths());
|
||||
row.createCell(6).setCellValue(covidStatus.getRate());
|
||||
}
|
||||
|
||||
try (FileOutputStream fo = new FileOutputStream(filename)) {
|
||||
workbook.write(fo);
|
||||
}
|
||||
|
||||
workbook.close();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user