iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated to use a Stream

This commit is contained in:
amdegregorio 2019-01-10 10:50:03 -05:00
parent 91f8c421c9
commit 81e2750ee0
2 changed files with 9 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
core-java-io/src/main/java/com/baeldung/csv/WriteCsvFileExample.java
View File

@ -1,11 +1,8 @@
package com.baeldung.csv;
import java.io.BufferedWriter;
import java.io.IOException;
public class WriteCsvFileExample {
public void writeLine(BufferedWriter writer, String[] data) throws IOException {
public String convertToCSV(String[] data) {
StringBuilder csvLine = new StringBuilder();
for (int i = 0; i < data.length; i++) {
@ -15,7 +12,7 @@ public class WriteCsvFileExample {
csvLine.append(escapeSpecialCharacters(data[i]));
}
writer.write(csvLine.toString());
return csvLine.toString();
}
public String escapeSpecialCharacters(String data) {

20
core-java-io/src/test/java/com/baeldung/csv/WriteCsvFileExampleUnitTest.java
View File

@ -3,12 +3,10 @@ package com.baeldung.csv;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.FileNotFoundException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import org.junit.Before;
@ -73,15 +71,11 @@ public class WriteCsvFileExampleUnitTest {
dataLines.add(new String[] { "Jane", "Doe, Jr.", "19", "She said \"I'm being quoted\"" });
File csvOutputFile = new File(CSV_FILE_NAME);
try (BufferedWriter writer = new BufferedWriter(new FileWriter(csvOutputFile))) {
for (Iterator<String[]> dataIterator = dataLines.iterator(); dataIterator.hasNext();) {
csvExample.writeLine(writer, dataIterator.next());
if (dataIterator.hasNext()) {
writer.newLine();
}
}
writer.flush();
} catch (IOException e) {
try (PrintWriter pw = new PrintWriter(csvOutputFile)) {
dataLines.stream()
.map(csvExample::convertToCSV)
.forEach(pw::println);
} catch (FileNotFoundException e) {
LOG.error("IOException " + e.getMessage());
}