diff --git a/src/main/java/millions/controller/CSVStockFileWriter.java b/src/main/java/millions/controller/CSVStockFileWriter.java new file mode 100644 index 0000000..ceb4677 --- /dev/null +++ b/src/main/java/millions/controller/CSVStockFileWriter.java @@ -0,0 +1,35 @@ +package millions.controller; + +import millions.model.Stock; + +import java.io.FileWriter; +import java.util.ArrayList; +import java.util.List; + +public class CSVStockFileWriter implements StockFileWriter { + List stocks; + String finalString; + + public CSVStockFileWriter(List stocks) { + this.stocks = stocks; + } + + @Override + public void formatString() { + StringBuilder builder = new StringBuilder(); + stocks.forEach(stock -> { + builder.append(stock.getSymbol()); + builder.append(","); + builder.append(stock.getCompany()); + builder.append(","); + builder.append(stock.getLatestPriceChange().toString()); + }); + this.finalString = builder.toString(); + } + + @Override + public boolean write(){ + // TODO: handle file creation/file selection when writing to file + return false; + } +} diff --git a/src/main/java/millions/controller/StockFileWriter.java b/src/main/java/millions/controller/StockFileWriter.java new file mode 100644 index 0000000..2e2046d --- /dev/null +++ b/src/main/java/millions/controller/StockFileWriter.java @@ -0,0 +1,8 @@ +package millions.controller; + +import java.util.List; + +public interface StockFileWriter { + public void formatString(); + public boolean write(); +} diff --git a/src/main/java/millions/controller/StockInformationWriter.java b/src/main/java/millions/controller/StockInformationWriter.java deleted file mode 100644 index 692bc75..0000000 --- a/src/main/java/millions/controller/StockInformationWriter.java +++ /dev/null @@ -1,34 +0,0 @@ -package millions.controller; - -import millions.model.Stock; - -import java.io.*; -import java.util.List; - -public class StockInformationWriter { - private final List stocks; - private final File destinationFile; - - public StockInformationWriter(List stocks, File destinationFile) { - this.stocks = stocks; - this.destinationFile = destinationFile; - } - - public void write() { - StringBuilder builder = new StringBuilder(); - for (Stock stock : stocks) { - builder.append(stock.getSymbol()); - builder.append(","); - builder.append(stock.getCompany()); - builder.append(","); - // Unsure if price history or just latest price should be saved - builder.append(stock.getSalesPrice().toPlainString()); - builder.append("\n"); - } - try (FileWriter writer = new FileWriter(destinationFile); BufferedWriter bufferedWriter = new BufferedWriter(writer)) { - bufferedWriter.write(builder.toString()); - } catch (IOException e) { - e.printStackTrace(); - } - } -}