From 1d08023cb54ac62ae7427ef92116a024857f9d98 Mon Sep 17 00:00:00 2001
From: Xi Xu <i@xi-xu.me>
Date: Mon, 1 Jul 2024 22:18:40 +0800
Subject: [PATCH] a

---
 src/server/tools/ExportToCSV.java | 45 +++++++++++++++++++++++++++
 src/server/tools/ExportToXLS.java | 51 +++++++++++++++++++++++++++++++
 2 files changed, 96 insertions(+)
 create mode 100644 src/server/tools/ExportToCSV.java
 create mode 100644 src/server/tools/ExportToXLS.java

diff --git a/src/server/tools/ExportToCSV.java b/src/server/tools/ExportToCSV.java
new file mode 100644
index 0000000..8d9eca2
--- /dev/null
+++ b/src/server/tools/ExportToCSV.java
@@ -0,0 +1,45 @@
+package server.tools;
+
+import java.io.FileWriter;
+import java.io.IOException;
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+
+public class ExportToCSV {
+    public static void to() {
+        String outputFile = "图书信息.csv";
+
+        try (Connection connection = DBConnection.getConnection();
+                Statement statement = connection.createStatement();
+                ResultSet resultSet = statement.executeQuery("SELECT * FROM book");
+                FileWriter fileWriter = new FileWriter(outputFile)) {
+
+            // Write column names to the CSV file
+            for (int i = 1; i <= resultSet.getMetaData().getColumnCount(); i++) {
+                fileWriter.append(resultSet.getMetaData().getColumnName(i));
+                if (i < resultSet.getMetaData().getColumnCount()) {
+                    fileWriter.append(",");
+                }
+            }
+            fileWriter.append("\n");
+
+            // Write data rows to the CSV file
+            while (resultSet.next()) {
+                for (int i = 1; i <= resultSet.getMetaData().getColumnCount(); i++) {
+                    fileWriter.append(resultSet.getString(i));
+                    if (i < resultSet.getMetaData().getColumnCount()) {
+                        fileWriter.append(",");
+                    }
+                }
+                fileWriter.append("\n");
+            }
+
+            System.out.println("Data exported successfully!");
+
+        } catch (SQLException | IOException e) {
+            e.printStackTrace();
+        }
+    }
+}
diff --git a/src/server/tools/ExportToXLS.java b/src/server/tools/ExportToXLS.java
new file mode 100644
index 0000000..151b38a
--- /dev/null
+++ b/src/server/tools/ExportToXLS.java
@@ -0,0 +1,51 @@
+package server.tools;
+
+import java.io.File;
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.Statement;
+
+import jxl.Workbook;
+import jxl.write.Label;
+import jxl.write.WritableSheet;
+import jxl.write.WritableWorkbook;
+
+public class ExportToXLS {
+    public static void to() {
+        try (Connection connection = DBConnection.getConnection();
+                Statement statement = connection.createStatement();
+                ResultSet rs = statement.executeQuery("SELECT * FROM book");) {
+
+            WritableWorkbook book = Workbook.createWorkbook(new File("图书信息.xls"));
+            WritableSheet sheet = book.createSheet("sheet1", 0);
+
+            Label label = new Label(0, 0, "title");
+            sheet.addCell(label);
+            label = new Label(1, 0, "author");
+            sheet.addCell(label);
+            label = new Label(2, 0, "publisher");
+            sheet.addCell(label);
+            label = new Label(3, 0, "oldprice");
+            sheet.addCell(label);
+            label = new Label(4, 0, "newprice");
+            sheet.addCell(label);
+            label = new Label(5, 0, "href");
+            sheet.addCell(label);
+            int i = 1;
+            while (rs.next()) {
+                for (int j = 0; j <= 5; j++) {
+                    label = new Label(j, i, rs.getString(j + 1));
+                    sheet.addCell(label);
+                }
+                i++;
+            }
+
+            book.write();
+            book.close();
+            System.out.println("Data exported successfully!");
+
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+}