feat: reworked classes StudentCsvWriter and StudentCsvReader

2026-08-30 20:34:10 +07:00
parent 6401d638eb
commit ba59fa12da
2 changed files with 152 additions and 0 deletions
@@ -0,0 +1,79 @@
+package ru.project.datasource;
+
+import ru.project.student.Student;
+import ru.project.student.StudentBuilder;
+
+import org.apache.commons.csv.CSVFormat;
+import org.apache.commons.csv.CSVParser;
+import org.apache.commons.csv.CSVRecord;
+
+import java.io.IOException;
+import java.io.UncheckedIOException;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.Arrays;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.NoSuchElementException;
+
+public class StudentCsvReader implements Iterator<Student> {
+    private static final String GROUP_NUMBER = "groupNumber";
+    private static final String AVERAGE_GRADE = "averageGrade";
+    private static final String RECORD_BOOK_NUMBER = "recordBookNumber";
+
+    private final CSVParser parser;
+    private final Iterator<CSVRecord> records;
+
+    public StudentCsvReader(String absoluteFilePath) {
+        Path path = Paths.get(absoluteFilePath);
+        if (!path.isAbsolute()) {
+            throw new IllegalArgumentException("File path must be absolute");
+        }
+
+        try {
+            CSVFormat format = CSVFormat.DEFAULT.builder()
+                    .setHeader()
+                    .setSkipHeaderRecord(true)
+                    .build();
+            this.parser = CSVParser.parse(path, StandardCharsets.UTF_8, format);
+            validateHeader(parser.getHeaderMap());
+            this.records = parser.iterator();
+        } catch (IOException e) {
+            throw new UncheckedIOException(e);
+        }
+    }
+
+    @Override
+    public boolean hasNext() {
+        return records.hasNext();
+    }
+
+    @Override
+    public Student next() {
+        if (!records.hasNext()) {
+            throw new NoSuchElementException();
+        }
+
+        CSVRecord record = records.next();
+        if (record.size() != 3) {
+            throw new IllegalArgumentException(
+                    "Expected 3 columns, but got " + record.size());
+        }
+
+        double averageGrade = Double.parseDouble(record.get(AVERAGE_GRADE));
+        return new StudentBuilder()
+                .setGroupNumber(record.get(GROUP_NUMBER))
+                .setAverageGrade(averageGrade)
+                .setRecordBookNumber(record.get(RECORD_BOOK_NUMBER))
+                .build();
+    }
+
+    private static void validateHeader(Map<String, Integer> headerMap) {
+        String[] expected = {GROUP_NUMBER, AVERAGE_GRADE, RECORD_BOOK_NUMBER};
+        if (headerMap.size() != expected.length
+                || !headerMap.keySet().containsAll(Arrays.asList(expected))) {
+            throw new IllegalArgumentException("Invalid CSV header");
+        }
+    }
+}
@@ -0,0 +1,73 @@
+package ru.project.datasource;
+
+import org.apache.commons.csv.CSVFormat;
+import org.apache.commons.csv.CSVPrinter;
+import java.io.BufferedWriter;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardOpenOption;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+
+import ru.project.student.Student;
+import ru.project.collection.MyList;
+import ru.project.collection.MyLinkedList;
+
+
+public class StudentCsvWriter{
+    private static final String[] HEADER = {
+            "groupNumber", "averageGrade", "recordBookNumber"
+    };
+    private static final DateTimeFormatter FILE_NAME_FORMAT =
+            DateTimeFormatter.ofPattern("yyyy-MM-dd-HH-mm");
+
+    private final Path requestedPath;
+    private Path actualPath;
+
+    public StudentCsvWriter(String absoluteFilePath) {
+        Path path = Paths.get(absoluteFilePath);
+        if (!path.isAbsolute()) {
+            throw new IllegalArgumentException("File path must be absolute");
+        }
+        this.requestedPath = path;
+    }
+
+    public void write(Student student) throws IOException {
+        Path target = resolveTargetPath();
+        boolean fileExists = Files.exists(target);
+
+        StandardOpenOption[] options = fileExists
+                ? new StandardOpenOption[]{StandardOpenOption.CREATE, StandardOpenOption.WRITE, StandardOpenOption.APPEND}
+                : new StandardOpenOption[]{StandardOpenOption.CREATE_NEW, StandardOpenOption.WRITE};
+
+        try (BufferedWriter writer = Files.newBufferedWriter(target, StandardCharsets.UTF_8, options);
+             CSVPrinter printer = new CSVPrinter(writer, CSVFormat.DEFAULT)) {
+            if (!fileExists) {
+                printer.printRecord((Object[]) HEADER);
+            }
+            printer.printRecord(
+                    student.getGroupNumber(),
+                    Double.toString(student.getAverageGrade()),
+                    student.getRecordBookNumber()
+            );
+        }
+    }
+
+    private Path resolveTargetPath() {
+        if (actualPath != null) {
+            return actualPath;
+        }
+        if (Files.exists(requestedPath)) {
+            actualPath = requestedPath;
+            return actualPath;
+        }
+
+        Path parent = requestedPath.getParent();
+        String fileName = "students-" + LocalDateTime.now().format(FILE_NAME_FORMAT) + ".csv";
+        actualPath = parent == null ? Paths.get(fileName).toAbsolutePath() : parent.resolve(fileName);
+        return actualPath;
+    }
+}