add CSVHelper for exporting exchange data to CSV file.

Author: lifxue

src/main/java/org/mapleaf/cointda/util/CSVHelper.java

@@ -15,6 +15,7 @@
  */
 package org.mapleaf.cointda.util;
 
+import java.io.File;
 import java.io.FileNotFoundException;
 import java.io.FileReader;
 import java.io.FileWriter;
@@ -40,7 +41,7 @@ public class CSVHelper {
     private static final char DELIMITER = ',';
     private static final Charset CHARSET = Charset.forName("UTF-8");
 
-    public static void writeCsv(String[] header, List<String[]> data, String filePath) {
+    public static void writeCsv(String[] header, List<String[]> data, File filePath) {
         //初始化csvformat
         CSVFormat formator = CSVFormat.DEFAULT
                 .withHeader(header)
@@ -63,8 +64,6 @@ public static void writeCsv(String[] header, List<String[]> data, String filePat
             logger.error(ex);
         }
 
-        logger.info("CSV文件创建成功,文件路径:" + filePath);
-
     }
 
     public static List<String[]> readCsv(String filePath) {

src/main/java/org/mapleaf/cointda/view/RootLayoutController.java

@@ -15,8 +15,11 @@
  */
 package org.mapleaf.cointda.view;
 
+import java.io.File;
 import java.io.IOException;
 import java.net.URL;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.ResourceBundle;
 import javafx.event.ActionEvent;
 import javafx.fxml.FXML;
@@ -29,11 +32,16 @@
 import javafx.scene.control.MenuItem;
 import javafx.scene.layout.AnchorPane;
 import javafx.scene.layout.BorderPane;
+import javafx.stage.FileChooser;
+import javafx.stage.FileChooser.ExtensionFilter;
 import javafx.stage.Stage;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
+import org.mapleaf.cointda.bean.TradeDataBean;
 import org.mapleaf.cointda.crypto.CoinListingCollector;
 import org.mapleaf.cointda.dao.CoinListingDao;
+import org.mapleaf.cointda.dao.TradeDataDao;
+import org.mapleaf.cointda.util.CSVHelper;
 
 /**
  * FXML Controller class
@@ -49,8 +57,6 @@ public class RootLayoutController implements Initializable {
     @FXML
     private ContextMenu settingMenu;
     @FXML
-    private MenuItem typeItem;
-    @FXML
     private MenuItem priceItem;
     @FXML
     private Button settingButton;
@@ -86,18 +92,53 @@ private void handleSetting(ActionEvent event) {
         settingMenu.show(settingButton, Side.RIGHT, 0, 0);
     }
 
+    /**
+     * 导出交易数据到csv文件
+     *
+     * @param event
+     */
+    @FXML
+    private void handleExportData(ActionEvent event) {
+        List<TradeDataBean> list = TradeDataDao.queryAll();
+        List<String[]> data = new ArrayList<>();
+        String[] headers = {"id", "coid_id", "简称", "买卖", "单价", "数量", "总价", "交易时间"};
+        if (list == null) {
+            Alert alert = new Alert(Alert.AlertType.INFORMATION);
+            alert.setTitle("消息");
+            alert.setHeaderText("导出数据失败");
+            alert.setContentText("没有交易数据！");
+            alert.showAndWait();
+            return;
+        }
+        list.stream().map((bean) -> {
+            String[] str = new String[8];
+            str[0] = bean.getId().toString();
+            str[1] = bean.getCoin_id().toString();
+            str[2] = bean.getCoin_symbol();
+            str[3] = bean.getSale_or_buy();
+            str[4] = bean.getPrice().toString();
+            str[5] = bean.getNum().toString();
+            str[6] = bean.getTotal_price().toString();
+            str[7] = bean.getTrade_date();
+            return str;
+        }).forEachOrdered((str) -> {
+            data.add(str);
+        });
+        FileChooser fileChooser = new FileChooser();
+        //文档类型过滤器
+        ExtensionFilter extFilter = new ExtensionFilter("txt files (*.csv)", "*.csv");
+        fileChooser.getExtensionFilters().add(extFilter);
+        File file = fileChooser.showSaveDialog(pane.getScene().getWindow());
+        CSVHelper.writeCsv(headers, data, file);
+    }
+
     @FXML
     private void handleCloseApp(ActionEvent event) {
         //Platform.exit();
         Stage stage = (Stage) pane.getScene().getWindow();
         stage.close();
     }
 
-    @FXML
-    private void handleSettingType(ActionEvent event) {
-        showTypeSettingView();
-    }
-
     @FXML
     private void handleSettingPrice(ActionEvent event) {
         showSettingPriceView();
@@ -135,12 +176,6 @@ public void setPane(BorderPane pane) {
      */
     private boolean update() {
         boolean ok = false;
-//        CoinIDMapCollector coin = new CoinIDMapCollector();
-//        if (CoinMarketCapDao.truncate()) {
-//            if (CoinMarketCapDao.batchInsert(coin.getCoinMarketCapIds()).length > 0) {
-//                ok = true;
-//            }
-//        }
         CoinListingCollector listing = new CoinListingCollector();
         if (CoinListingDao.truncate()) {
             if (CoinListingDao.batchInsert(listing.getCoinMarketListing()).length > 0) {
@@ -198,22 +233,6 @@ public void showPATableView() {
         }
     }
 
-    /**
-     * 显示品种类型设置视图
-     */
-    public void showTypeSettingView() {
-        try {
-            FXMLLoader loader = new FXMLLoader();
-            loader.setLocation(RootLayoutController.class.getResource("TypeSettingView.fxml"));
-            AnchorPane dataEditView = (AnchorPane) loader.load();
-            pane.setCenter(dataEditView);
-            //转移焦点到center
-            dataEditView.requestFocus();
-        } catch (IOException e) {
-            LOGGER.error(e);
-        }
-    }
-
     /**
      * 显示品种价格设置视图
      */

src/main/resources/org/mapleaf/cointda/view/RootLayout.fxml

@@ -7,7 +7,7 @@
 <?import javafx.scene.layout.BorderPane?>
 <?import javafx.scene.layout.VBox?>
 
-<BorderPane prefHeight="700.0" prefWidth="1000.0" xmlns="http://javafx.com/javafx/11.0.2" xmlns:fx="http://javafx.com/fxml/1" fx:controller="org.mapleaf.cointda.view.RootLayoutController">
+<BorderPane prefHeight="700.0" prefWidth="1000.0" xmlns="http://javafx.com/javafx/11.0.1" xmlns:fx="http://javafx.com/fxml/1" fx:controller="org.mapleaf.cointda.view.RootLayoutController">
    <left>
       <VBox alignment="TOP_CENTER" prefHeight="700.0" prefWidth="150.0" spacing="20.0" BorderPane.alignment="CENTER">
          <children>
@@ -26,6 +26,7 @@
                     </items>
                   </ContextMenu>
                </contextMenu></Button>
+            <Button mnemonicParsing="false" onAction="#handleExportData" prefHeight="40.0" prefWidth="100.0" text="导出交易数据" />
             <Button mnemonicParsing="false" onAction="#handleUpdate" prefHeight="40.0" prefWidth="100.0" text="更新基础数据" />
             <Button mnemonicParsing="false" onAction="#handleCloseApp" prefHeight="40.0" prefWidth="100.0" text="退出" />
          </children>