v0.5.2:support futures main

Author: Xiaobaishushu25

UpdateRecord.md

@@ -1,10 +1,14 @@
+### version: 0.5.2 (2025年2月7日01:20:52)
+1. 优化了历史k线保存的逻辑。
+2. 优化了应用的启动过程。
+
 ### version: 0.5.1 (2025年2月5日15:47:11)
 1. 完善了日志模块，现在panic也会被记录到日志中。
 2. 现在期货数据支持主力合约。
 
 ### version: 0.5.0 (2025年2月4日23:46:44)
 1. 修复了一个bug，今后在"全部"分组中使用“从当前分组移除”功能将会直接删除该股票以及所有相关数据。
-2. 添加新功能，现在支持期货。
+2. 添加新功能，现在支持期货。（https://aktools.akfamily.xyz/aktools/）
 
 ### version: 0.4.16 (2025年1月16日01:51:02)
 1. 修复了一个bug，该bug导致当前最新日k线显示不正确。

src-tauri/Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "trade_tip"
-version = "0.5.1"
+version = "0.5.2"
 description = "金融助手"
 authors = ["xbss"]
 edition = "2021"
@@ -39,6 +39,8 @@ encoding_rs = "0.8.34"
 tauri-plugin-dialog = "2"
 ashares = "0.1.0"
 itertools = "0.14.0"#支持多个迭代器的扁平化组合,计算均线时要用
+
+async-recursion = "1.1.1"
 [features]
 # This feature is used for production builds or when a dev server is not specified, DO NOT REMOVE!!
 custom-protocol = ["tauri/custom-protocol"]

src-tauri/Cargo.toml

@@ -13,7 +13,7 @@ mod utils;
 
 use crate::app_errors::AppResult;
 use crate::cache::intraday_chart_cache::IntradayChartCache;
-use crate::config::config::Config;
+use crate::config::config::{Config, DataConfig};
 use crate::entities::init_db_coon;
 use crate::service::command::tauri_command::{
     add_stock_info, create_group, delete_all_transaction_records, delete_graphic_by_group_id,
@@ -35,12 +35,14 @@ use std::collections::HashMap;
 use std::{env, panic};
 use std::sync::atomic::{AtomicBool};
 use std::sync::{Arc, LazyLock, Mutex};
+use std::time::Duration;
 use log4rs::append::console::ConsoleAppender;
 use log4rs::append::file::FileAppender;
 use log4rs::encode::pattern::PatternEncoder;
 use log4rs::filter::threshold::ThresholdFilter;
-use tauri::{AppHandle, Manager};
+use tauri::{Manager};
 use tokio::sync::mpsc;
+use tokio::sync::mpsc::Receiver;
 use tokio::task::JoinHandle;
 
 ///是否需要实时更新
@@ -118,23 +120,14 @@ pub async fn get_close_prices(
 #[tokio::main]
 async fn main() {
     init_app().await;
-    //todo 下面这俩更新如果出错了在前端页面是没有提醒的，如果想要emit一个错误事件到前端的话，有可能存在前端页面还没渲染完就emit了，就收不到了；
-    // 如果弄一个本地变量保存错误信息，前端页面渲染完成后主动来invoke读取这个变量判断是否出错应该可以，但是目前懒得搞。
-    tokio::spawn(async {
-        info!("update stock info start");
-        if let Err(e) = service::curd::update_all_day_k().await{
-            error!("update stock info failed:{}",e)
-        }
-        // update().await;
-    });
-    tokio::spawn(async {
-        info!("update position start");
-        if let Err(e) = service::curd::update_all_position().await{
-            error!("update position failed:{}",e)
-        }
-    });
+
+    let config = Config::load().await;
+    let data_config = config.data_config.clone();
+    let (send,recv) = mpsc::channel::<()>(1);
+    init_data_server_daily_position(recv,data_config).await;
+
     let state = MyState::new().await;
-    info!("ui start");
+    info!("tauri start...");
     tauri::Builder::default()
         .plugin(tauri_plugin_single_instance::init(|app, args, cwd| {
             error!("the app is already running, args: {:?}, cwd: {}", args, cwd);
@@ -147,40 +140,29 @@ async fn main() {
         .manage(state)
         .manage(IntradayChartCache::new())
         // .manage(Mutex::new(Config::load().await))
-        .manage(Mutex::new(Config::load().await))
+        .manage(Mutex::new(config))
+        .manage(send)
         .plugin(tauri_plugin_shell::init())
         .plugin(tauri_plugin_fs::init())
         .plugin(tauri_plugin_window_state::Builder::default().build())
-        .setup(|app| {
-            let app_handle: &AppHandle = app.app_handle();
-            let config = app_handle.state::<Mutex<Config>>();
-            let data_config = config.lock().unwrap().data_config.clone();
-            let (send,recv) = mpsc::channel::<()>(1);
-            app.manage(send);
-            tokio::spawn(async move{
-                if let Err(e) = start_data_server(&data_config,recv).await{
-                    error!("start data server error:{}",e);
-                }
-            });
-            Ok(())
-        })
-        // .setup(|app|{
-        //     info!("{:?}",app.app_handle().path().app_config_dir().unwrap()); //"C:\\Users\\Xbss\\AppData\\Roaming\\com.xbss.trade-tip"
-        //     let main_window = app.get_webview_window("main").unwrap();
-        //     // main_window.as_ref().window().restore_state(StateFlags::all())
-        //     let tool_window = app.get_webview_window("tool").unwrap();
+        // .setup(|app| {
+        //     let app_handle: &AppHandle = app.app_handle();
+        //     let config = app_handle.state::<Mutex<Config>>();
+        //     let data_config = config.lock().unwrap().data_config.clone();
+        //     let (send,recv) = mpsc::channel::<()>(1);
+        //     app.manage(send);
         //     tokio::spawn(async move{
-        //         // let mut result = main_window.restore_state(StateFlags::all());
-        //         // info!("restore main window state result:{:?}",result);
-        //         // if result.is_err(){
-        //         //     error!("restore main window state error:{}",result.err().unwrap().to_string());
-        //         // };
-        //         main_window.show().unwrap();
-        //         //  result = tool_window.restore_state(StateFlags::all());
-        //         // if result.is_err(){
-        //         //     error!("restore tool window state error:{}",result.err().unwrap().to_string());
-        //         // };
+        //         tokio::spawn(async move{
+        //             info!("update daily info start...");
+        //             if let Err(e) = service::curd::update_all_day_k(data_config.use_ak_share).await{
+        //                 error!("update daily info failed:{}",e)
+        //             };
+        //         });
+        //         if let Err(e) = start_data_server(&data_config,recv).await{
+        //             error!("start data server error:{}",e);
+        //         };
         //     });
+        //     info!("setup ok.");
         //     Ok(())
         // })
         .invoke_handler(tauri::generate_handler![
@@ -285,24 +267,44 @@ async fn init_logger() {
         }
     }));
 }
-// async fn update() {
-//     match service::curd::update_all_day_k().await {
-//         Ok(_) => {
-//             info!("更新日线数据成功");
-//             NOTICE
-//                 .lock()
-//                 .unwrap()
-//                 .replace("更新日线数据成功".to_string());
-//         }
-//         Err(e) => {
-//             error!("更新日线数据失败:{}", e);
-//             NOTICE
-//                 .lock()
-//                 .unwrap()
-//                 .replace(format!("更新日线数据失败:{}", e.to_string()));
-//         }
-//     };
-// }
+async fn init_data_server_daily_position(recv:Receiver<()>,data_config:DataConfig){
+    let can_handle_futures = data_config.use_ak_share;
+    tokio::spawn(async move{
+        if let Err(e) = start_data_server(&data_config,recv).await{
+            error!("start data server error:{}",e);
+        };
+    });
+    //todo 下面这俩更新如果出错了在前端页面是没有提醒的，如果想要emit一个错误事件到前端的话，有可能存在前端页面还没渲染完就emit了，就收不到了；
+    // 如果弄一个本地变量保存错误信息，前端页面渲染完成后主动来invoke读取这个变量判断是否出错应该可以，但是目前懒得搞。
+    tokio::spawn(async move{
+        info!("update daily info start...");
+        if let Err(e) = service::curd::update_all_day_k(can_handle_futures,false).await{
+            error!("update daily info failed:{:?}.",e)
+        }
+        info!("update daily info end.");
+        // match service::curd::update_all_day_k(can_handle_futures).await {
+        //     Ok(failure) => {
+        //         if failure {
+        //             error!("first update daily info failed, 20s later will retry again.");
+        //             sleep(Duration::from_secs(20)).await;
+        //             match service::curd::update_all_day_k(can_handle_futures).await{
+        //                 Ok(failure) =>{
+        //
+        //                 }
+        //                 Err(e) => {error!("second update stock info failed:{:?}",e)}
+        //             }
+        //         }
+        //     }
+        //     Err(e) => {error!("update stock info failed:{:?}",e)}
+        // }
+    });
+    tokio::spawn(async {
+        info!("update position start");
+        if let Err(e) = service::curd::update_all_position().await{
+            error!("update position failed:{}",e)
+        }
+    });
+}
 // ///判断是否开市,先发起一个请求，然后sleep 1秒，再发起一个请求，如果两个请求的返回值不一样，则证明开市了，否则证明闭市了。
 // async fn judge_market_open() {
 //     let response1 = REQUEST.get().unwrap().get("https://qt.gtimg.cn/q=sz159992").await.unwrap();

src-tauri/src/main.rs

@@ -10,7 +10,6 @@ use crate::service::curd::stock_data_curd::StockDataCurd;
 use crate::service::curd::stock_group_curd::StockGroupCurd;
 use crate::service::curd::stock_info_curd::StockInfoCurd;
 use crate::service::curd::transaction_record_curd::TransactionRecordCurd;
-use crate::service::curd::update_all_day_k;
 use crate::service::http::{init_http, REQUEST};
 use crate::{get_close_prices, MyState, IS_MARKET_OPEN, UPDATEING};
 use log::{error, info};
@@ -27,7 +26,7 @@ use crate::service::curd::position_curd::PositionCurd;
 #[tauri::command]
 pub async fn update_live_state<'r>(
     state: State<'r, MyState>,
-    app_handle: tauri::AppHandle,
+    app_handle: AppHandle,
     config: State<'r, Mutex<Config>>,
     group_name: String,
     live_state: bool,
@@ -39,17 +38,17 @@ pub async fn update_live_state<'r>(
     }
     Ok(())
 }
-#[tauri::command]
-pub async fn update_all_stock_day_k() -> Result<String, String> {
-    // state.update_live_state(live_state);
-    match update_all_day_k().await {
-        Ok(_) => Ok("更新成功".to_string()),
-        Err(e) => {
-            handle_error("更新日线数据失败", e.to_string())
-            // Err(format!("更新日线数据失败:{}",e.to_string()))
-        }
-    }
-}
+// #[tauri::command]
+// pub async fn update_all_stock_day_k() -> Result<String, String> {
+//     // state.update_live_state(live_state);
+//     match update_all_day_k().await {
+//         Ok(_) => Ok("更新成功".to_string()),
+//         Err(e) => {
+//             handle_error("更新日线数据失败", e.to_string())
+//             // Err(format!("更新日线数据失败:{}",e.to_string()))
+//         }
+//     }
+// }
 #[tauri::command]
 pub async fn get_response(url: String) -> Result<String, String> {
     match REQUEST.get().unwrap().get_response(&url).await {
@@ -112,9 +111,11 @@ pub async fn query_all_groups() -> Result<Vec<StockGroup>, String> {
 
 #[tauri::command]
 pub async fn query_stocks_by_group_name(name: String) -> Result<Vec<StockInfoG>, String> {
+    // error!("根据分组名查询分组中的股票:{}",name);
     if name == "持有" {
         return match StockInfoCurd::query_all_hold_info().await {
             Ok(more_infos) => {
+                // error!("查询持有分组成功:{:?}",more_infos);
                 Ok(more_infos)
             }
             Err(e) => {
@@ -501,7 +502,10 @@ pub async fn query_live_stocks_data_by_group_name<'r>(
     group_name: String,
     app_handle: AppHandle,
 ) -> Result<(), String> {
-    info!("查询{}分组内的实时数据", group_name);
+    if group_name.is_empty(){//应用刚启动会有一个空的分组名进来，不知道哪里传来
+        return Ok(());
+    }
+    error!("查询{}分组内的实时数据", group_name);
     // if !IS_MARKET_OPEN.load(Ordering::Relaxed) {
     //     info!("市场未开市,不进行查询操作！");
     //     return Ok(());
@@ -510,10 +514,9 @@ pub async fn query_live_stocks_data_by_group_name<'r>(
     let result = if group_name == "持有" {
         StockInfoCurd::query_all_hold_only_code().await
     } else {
-        GroupStockRelationCurd::query_only_code_by_group_name(group_name).await
+        GroupStockRelationCurd::query_only_code_by_group_name(group_name.clone()).await
     };
     let update_freq = config.lock().unwrap().data_config.update_freq;
-    info!("更新频率为{}秒", update_freq);
     match result {
         Ok(codes) => {
             if codes.is_empty() {
@@ -542,7 +545,6 @@ pub async fn query_live_stocks_data_by_group_name<'r>(
                                 .await
                                 {
                                     Ok(_) => {
-                                        // info!("查询成功:{:?}",stock_data_list);
                                         app_handle
                                             .emit("live_stock_data", stock_data_list)
                                             .unwrap();