Read monitoring headers and contents in csv format from report file

Author: Dokyeongyun

src/main/java/root/core/repository/constracts/ReportRepository.java

@@ -1,8 +1,14 @@
 package root.core.repository.constracts;
 
-import root.core.domain.MonitoringResult;
+import java.io.File;
+import java.util.List;
 
 public interface ReportRepository {
 
-	void writeReportFile(String filePath, String fileName, String fileExtension, MonitoringResult<?> monitoringResult);
+	<T> void writeReportFile(String filePath, String fileName, String fileExtension, List<T> monitoringResult,
+			Class<T> clazz);
+
+	List<String> getReportHeaders(File reportFile);
+	
+	String getReportContentsInCsv(File reportFile);
 }

src/main/java/root/core/repository/implement/ReportRepositoryImpl.java

@@ -1,15 +1,20 @@
 package root.core.repository.implement;
 
+import java.io.BufferedReader;
 import java.io.BufferedWriter;
 import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.FileReader;
 import java.io.FileWriter;
 import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
 
 import org.apache.commons.lang3.StringUtils;
 
 import lombok.extern.slf4j.Slf4j;
-import root.core.domain.MonitoringResult;
-import root.core.domain.TableSpaceUsage;
 import root.core.repository.constracts.ReportRepository;
 import root.utils.CsvUtils;
 
@@ -31,46 +36,34 @@ public static ReportRepository getInstance() {
 	 * 모니터링 결과를 파일에 기록한다.
 	 */
 	@Override
-	public void writeReportFile(String filePath, String fileName, String fileExtension,
-			MonitoringResult<?> monitoringResult) {
+	public <T> void writeReportFile(String filePath, String fileName, String fileExtension, List<T> monitoringResult,
+			Class<T> clazz) {
 
 		File file = new File(rootDirectory + "/" + filePath + "/" + fileName + fileExtension);
-		File parentDir = file.getParentFile();
-
 		String content = null;
-
 		try {
 
-			boolean isNewFile = false;
 			if (!file.exists()) {
-				parentDir.mkdirs();
-				isNewFile = file.createNewFile();
+				file.getParentFile().mkdirs();
+				file.createNewFile();
 			}
 
-			// 첫 파일작성인 경우 헤더 추가
-			if (isNewFile) {
-				content = StringUtils.joinWith(",", "MONITORING_DATE", "MONITORING_TIME",
-						CsvUtils.createCsvHeader(TableSpaceUsage.class));
-			}
+			content = CsvUtils.createCsvHeader(clazz);
 
-			// 모니터링결과 Row 추가
-			String monitoringDay = monitoringResult.getMonitoringDay();
-			String monitoringTime = monitoringResult.getMonitoringTime();
-			for (Object obj : monitoringResult.getMonitoringResults()) {
-				String rowString = StringUtils.joinWith(",", monitoringDay, monitoringTime,
-						CsvUtils.createCsvRow(obj, obj.getClass()));
-				content = StringUtils.joinWith(System.lineSeparator(), content, rowString);
+			for (Object t : monitoringResult) {
+				String row = CsvUtils.createCsvRow(t, t.getClass());
+				content = StringUtils.joinWith(System.lineSeparator(), content, row);
 			}
 		} catch (Exception e) {
 			e.printStackTrace();
 		}
 
 		if (content == null) {
-			log.error(String.format("파일에 작성할 내용이 없습니다. 파일경로: %s", file.getPath()));
+			log.info(String.format("파일에 작성할 내용이 없습니다. 파일경로: %s", file.getPath()));
 			return;
 		}
 
-		try (BufferedWriter bw = new BufferedWriter(new FileWriter(file, true))) {
+		try (BufferedWriter bw = new BufferedWriter(new FileWriter(file, false))) {
 			bw.append(content);
 			bw.flush();
 			bw.close();
@@ -79,4 +72,46 @@ public void writeReportFile(String filePath, String fileName, String fileExtensi
 		}
 	}
 
+	@Override
+	public List<String> getReportHeaders(File file) {
+		List<String> result = new ArrayList<>();
+
+		try (BufferedReader br = new BufferedReader(new FileReader(file))) {
+
+			String firstLine = br.readLine();
+			Map<Integer, String> headerMap = CsvUtils.parseCsvLine(firstLine);
+			List<Integer> sortedKeyList = headerMap.keySet().stream().sorted().collect(Collectors.toList());
+
+			for (int i : sortedKeyList) {
+				result.add(headerMap.get(i));
+			}
+
+		} catch (FileNotFoundException e) {
+			e.printStackTrace();
+		} catch (IOException e) {
+			e.printStackTrace();
+		}
+
+		return result;
+	}
+
+	@Override
+	public String getReportContentsInCsv(File reportFile) {
+		StringBuilder result = new StringBuilder();
+
+		try (BufferedReader br = new BufferedReader(new FileReader(reportFile))) {
+
+			String line = br.readLine();
+			while ((line = br.readLine()) != null) {
+				result.append(line).append(System.lineSeparator());
+			}
+
+		} catch (FileNotFoundException e) {
+			e.printStackTrace();
+		} catch (IOException e) {
+			e.printStackTrace();
+		}
+
+		return result.toString();
+	}
 }