From 6cb2134a53422f471f4f9b77c34d67e1fb4d31db Mon Sep 17 00:00:00 2001
From: 张洋洋 <10611411+yang-yang-z@user.noreply.gitee.com>
Date: 星期四, 09 一月 2025 10:16:56 +0800
Subject: [PATCH] [add]h5读取改造

---
 src/main/java/com/se/simu/utils/CsvToSQLiteUtils.java |   22 +++++++++++++++++-----
 1 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/se/simu/utils/CsvToSQLiteUtils.java b/src/main/java/com/se/simu/utils/CsvToSQLiteUtils.java
index 165bb0d..d4df8b0 100644
--- a/src/main/java/com/se/simu/utils/CsvToSQLiteUtils.java
+++ b/src/main/java/com/se/simu/utils/CsvToSQLiteUtils.java
@@ -1,5 +1,8 @@
 package com.se.simu.utils;
 
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
 import org.apache.commons.csv.CSVFormat;
 import org.apache.commons.csv.CSVRecord;
 
@@ -156,8 +159,9 @@
      * @param stationRainFile Station Rain 鏂囦欢
      * @param tableName       琛ㄥ悕绉�
      */
-    public static void readCsvSaveLocal(String stationRainFile, String tableName) {
+    public static JSONArray readCsvSaveLocal(String stationRainFile, String tableName) {
         // 1. 鍒涘缓SQLite鏁版嵁搴撹繛鎺�
+        JSONArray array=null;
         try (Connection conn = DriverManager.getConnection("jdbc:sqlite:rainfall.db")) {
             if (conn != null) {
                 // 2. 鍒涘缓琛紙濡傛灉涓嶅瓨鍦級
@@ -170,7 +174,7 @@
                 // 4. 寮�濮嬩簨鍔�
                 conn.setAutoCommit(false);
                 // 5. 璇诲彇CSV鏂囦欢骞舵彃鍏ユ暟鎹�
-                readCsvAndInsertDatas(stationRainFile, conn, tableName);
+                array = readCsvAndInsertDatas(stationRainFile, conn, tableName);
                 // 6. 鎻愪氦浜嬪姟
                 conn.commit();
                 System.out.println("鏁版嵁鎴愬姛鎻掑叆鍒癝QLite鏁版嵁搴擄紒");
@@ -178,11 +182,13 @@
         } catch (SQLException e) {
             System.err.println("SQLite杩炴帴澶辫触: " + e.getMessage());
         }
+        return array;
     }
 
 
-    private static void readCsvAndInsertDatas(String csvFilePath, Connection conn, String tableName) {
+    private static JSONArray readCsvAndInsertDatas(String csvFilePath, Connection conn, String tableName) {
         // 浣跨敤 Apache Commons CSV 璇诲彇CSV鏂囦欢
+        JSONArray array = new JSONArray();
         try (Reader reader = new InputStreamReader(Files.newInputStream(Paths.get(csvFilePath)), "GBK")) {
             Iterable<CSVRecord> records = CSVFormat.DEFAULT
                     .withHeader("闆ㄩ噺绔�", "闄嶉洦閲�", "缁忓害", "绾害", "datatime")
@@ -190,7 +196,6 @@
                     .parse(reader);
 
             String insertDataSql = "INSERT INTO " + tableName + " (station_name, rainfall, longitude, latitude, datetime) VALUES (?, ?, ?, ?, ?);";
-
             try (PreparedStatement pstmt = conn.prepareStatement(insertDataSql)) {
                 // 鎵归噺澶у皬
                 int batchSize = 1000;
@@ -210,7 +215,13 @@
                     pstmt.setDouble(3, longitude);
                     pstmt.setDouble(4, latitude);
                     pstmt.setString(5, datetime);
-
+                    JSONObject jsonObject = new JSONObject();
+                    jsonObject.put("stationName", stationName);
+                    jsonObject.put("rainfall", rainfall);
+                    jsonObject.put("longitude", longitude);
+                    jsonObject.put("latitude", latitude);
+                    jsonObject.put("datetime", datetime);
+                    array.add(jsonObject);
                     // 娣诲姞鍒版壒澶勭悊
                     pstmt.addBatch();
                     count++;
@@ -228,6 +239,7 @@
         } catch (IOException e) {
             System.err.println("璇诲彇CSV鎴栨彃鍏ユ暟鎹椂鍑洪敊: " + e.getMessage());
         }
+        return array;
     }
 
     /**

--
Gitblit v1.9.3