[JAVA-958] Standardize packages in spring-batch module

2020-03-20 08:06:25 +01:00
parent 31d88cba38
commit e368815ec8
47 changed files with 106 additions and 102 deletions
--- a/spring-batch/src/main/java/com/baeldung/taskletsvschunks/utils/FileUtils.java
+++ b/spring-batch/src/main/java/com/baeldung/taskletsvschunks/utils/FileUtils.java
@@ -0,0 +1,95 @@
+package com.baeldung.taskletsvschunks.utils;
+
+import com.opencsv.CSVReader;
+import com.opencsv.CSVWriter;
+import com.baeldung.taskletsvschunks.model.Line;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.File;
+import java.io.FileReader;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.time.LocalDate;
+import java.time.format.DateTimeFormatter;
+
+public class FileUtils {
+
+    private final Logger logger = LoggerFactory.getLogger(FileUtils.class);
+
+    private String fileName;
+    private CSVReader CSVReader;
+    private CSVWriter CSVWriter;
+    private FileReader fileReader;
+    private FileWriter fileWriter;
+    private File file;
+
+    public FileUtils(String fileName) {
+        this.fileName = fileName;
+    }
+
+    public Line readLine() {
+        try {
+            if (CSVReader == null) initReader();
+            String[] line = CSVReader.readNext();
+            if (line == null) return null;
+            return new Line(line[0], LocalDate.parse(line[1], DateTimeFormatter.ofPattern("MM/dd/yyyy")));
+        } catch (Exception e) {
+            logger.error("Error while reading line in file: " + this.fileName);
+            return null;
+        }
+    }
+
+    public void writeLine(Line line) {
+        try {
+            if (CSVWriter == null) initWriter();
+            String[] lineStr = new String[2];
+            lineStr[0] = line.getName();
+            lineStr[1] = line
+              .getAge()
+              .toString();
+            CSVWriter.writeNext(lineStr);
+        } catch (Exception e) {
+            logger.error("Error while writing line in file: " + this.fileName);
+        }
+    }
+
+    private void initReader() throws Exception {
+        ClassLoader classLoader = this
+          .getClass()
+          .getClassLoader();
+        if (file == null) file = new File(classLoader
+          .getResource(fileName)
+          .getFile());
+        if (fileReader == null) fileReader = new FileReader(file);
+        if (CSVReader == null) CSVReader = new CSVReader(fileReader);
+    }
+
+    private void initWriter() throws Exception {
+        if (file == null) {
+            file = new File(fileName);
+            file.createNewFile();
+        }
+        if (fileWriter == null) fileWriter = new FileWriter(file, true);
+        if (CSVWriter == null) CSVWriter = new CSVWriter(fileWriter);
+    }
+
+    public void closeWriter() {
+        try {
+            CSVWriter.close();
+            fileWriter.close();
+        } catch (IOException e) {
+            logger.error("Error while closing writer.");
+        }
+    }
+
+    public void closeReader() {
+        try {
+            CSVReader.close();
+            fileReader.close();
+        } catch (IOException e) {
+            logger.error("Error while closing reader.");
+        }
+    }
+
+}