as-rust/exchanges/tests/xlsx_test.rs

// use std::collections::{BTreeMap, HashSet};
// use chrono::{DateTime, FixedOffset, NaiveDateTime, TimeZone, Utc};
// 
// use tracing::{trace};
// 
// use exchanges::gate_swap_rest::GateSwapRest;
// use exchanges::xlsx_utils::creation_xlsx;
// 
// //ws-订阅公共频道信息
// #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
// async fn test_gate_creation_xlsx() {
//     global::log_utils::init_log_with_trace();
// 
//     //获取不同账号的数据
//     let mut acc_array: Vec<BTreeMap<String, String>> = vec![];
//     let mut acc_all_data: BTreeMap<String, Vec<Vec<String>>> = BTreeMap::new();
//     let mut acc_all_data_clone: BTreeMap<String, Vec<Vec<String>>> = BTreeMap::new();
//     let mut data_array_all: Vec<Vec<String>> = vec![];
//     let mut time_all: Vec<i64> = vec![];
//     loop {
//         let mut gate60: BTreeMap<String, String> = BTreeMap::new();
//         gate60.insert("acc_name".to_string(), String::from("gate60"));
//         gate60.insert("access_key".to_string(), String::from("61e3a1b12c44c8ccfce1f32782b9922f"));
//         gate60.insert("secret_key".to_string(), String::from("f3f533fa685cbae44b3017f73b71e90eaa9aa8d4922a39426744721e4824a527"));
//         acc_array.push(gate60);
// 
//         let mut gate59: BTreeMap<String, String> = BTreeMap::new();
//         gate59.insert("acc_name".to_string(), String::from("gate59"));
//         gate59.insert("access_key".to_string(), String::from("c691d4fcc5a5a98af459f993a3a0c653"));
//         gate59.insert("secret_key".to_string(), String::from("05e7f8640bffeacc146b6f9f08512955d00d89bbdb051c9427f31adf96adeb2f"));
//         acc_array.push(gate59);
// 
//         let mut gate58: BTreeMap<String, String> = BTreeMap::new();
//         gate58.insert("acc_name".to_string(), String::from("gate58"));
//         gate58.insert("access_key".to_string(), String::from("dfdc30687ac71daefa2fb39c706b8afa"));
//         gate58.insert("secret_key".to_string(), String::from("31e8999f85d38f5dd174f8919d9a1611c24d4e35b4d6319d6b160005871bf8b6"));
//         acc_array.push(gate58);
// 
//         let mut gate57: BTreeMap<String, String> = BTreeMap::new();
//         gate57.insert("acc_name".to_string(), String::from("gate57"));
//         gate57.insert("access_key".to_string(), String::from("ba11ea511f343763db7c92c685f20c12"));
//         gate57.insert("secret_key".to_string(), String::from("272d1d38ac4f0af3e6ed96e6a9a2c59dd905c3d7ad730507008604bba14edb3d"));
//         acc_array.push(gate57);
// 
//         let mut gate56: BTreeMap<String, String> = BTreeMap::new();
//         gate56.insert("acc_name".to_string(), String::from("gate56"));
//         gate56.insert("access_key".to_string(), String::from("72f0c351a83b04808baad625f6085599"));
//         gate56.insert("secret_key".to_string(), String::from("8a8d482bcc31f184cce350531cb40ca70564f9c466698d025e16d8943257dc0f"));
//         acc_array.push(gate56);
// 
//         let mut gate55: BTreeMap<String, String> = BTreeMap::new();
//         gate55.insert("acc_name".to_string(), String::from("gate55"));
//         gate55.insert("access_key".to_string(), String::from("036408d5cbd8ddf1c6e0baab61ee641a"));
//         gate55.insert("secret_key".to_string(), String::from("940414a37711e59848011e99e223fa11b0e69f8badd35347301e3f8e41957a60"));
//         acc_array.push(gate55);
// 
//         let mut gate54: BTreeMap<String, String> = BTreeMap::new();
//         gate54.insert("acc_name".to_string(), String::from("gate54"));
//         gate54.insert("access_key".to_string(), String::from("fbe564e8bd4efaa0c3c023ca8c057b36"));
//         gate54.insert("secret_key".to_string(), String::from("0be3c0223fd021fdacacc03f183f467e988ceee6eb1b0e09ca96bb1eebd45f39"));
//         acc_array.push(gate54);
// 
//         let mut gate53: BTreeMap<String, String> = BTreeMap::new();
//         gate53.insert("acc_name".to_string(), String::from("gate53"));
//         gate53.insert("access_key".to_string(), String::from("16d91ba0a3d79a2925d16ea01a615fa1"));
//         gate53.insert("secret_key".to_string(), String::from("607b07cf240466656c00beb0c6fff252839467583fd3f8b14782eb007b3d99ce"));
//         acc_array.push(gate53);
// 
//         let mut gate52: BTreeMap<String, String> = BTreeMap::new();
//         gate52.insert("acc_name".to_string(), String::from("gate52"));
//         gate52.insert("access_key".to_string(), String::from("31054006c457a80a027e961cf3e5e3a4"));
//         gate52.insert("secret_key".to_string(), String::from("a43f8f5672b49bfcc304d0731100610de1c55e7d2b6c00199b267993f0b189d1"));
//         acc_array.push(gate52);
// 
//         let mut gate51: BTreeMap<String, String> = BTreeMap::new();
//         gate51.insert("acc_name".to_string(), String::from("gate51"));
//         gate51.insert("access_key".to_string(), String::from("edcedefe7830dd5722c2c37704ae700f"));
//         gate51.insert("secret_key".to_string(), String::from("41db86a8463ac7c66023a8505e5fddd3448e551a11a52141bf57ca8478e2149b"));
//         acc_array.push(gate51);
// 
// 
//         for acc in acc_array {
//             let mut gate_exc = GateSwapRest::new(false, acc.clone());
//             let data = gate_exc.account_book("usdt".to_string()).await;
//             // trace!("data???????：{:?}",data.clone());
//             if data.code.as_str() == "200" {
//                 let acc_name = acc.get("acc_name").clone();
// 
//                 //账号
//                 let json_value: serde_json::Value = serde_json::from_str(&data.data).unwrap();
//                 if let serde_json::Value::Array(array) = json_value {
//                     let mut name_data_all: Vec<Vec<String>> = vec![];
//                     let mut name_data_all_clone: Vec<Vec<String>> = vec![];
//                     for item in array {
//                         let time = item["time"].as_i64().unwrap();//秒级
//                         // 使用秒构建一个DateTime<Utc>对象
//                         // let datetime: DateTime<Utc> = Utc.timestamp(time, 0);
//                         //2023-12-10 00:00:00
//                         if 1702051200 > time {
//                             continue;
//                         }
//                         time_all.push(time);
//                         let time_str = NaiveDateTime::from_timestamp_millis((time + 8 * 3600) * 1000).unwrap().format("%Y-%m-%d %H:%M:%S%.3f").to_string();
// 
//                         trace!("数据时间解析:{:?}---{:?}",time,time_str);
//                         let change = item["change"].as_str().unwrap();
// 
//                         let balance = item["balance"].as_str().unwrap();
//                         let type_str = match item["type"].as_str().unwrap() {
//                             "dnw" => { "转入转出" }
//                             "pnl" => { "减仓盈亏" }
//                             "fee" => { "交易手续费" }
//                             "refr" => { "推荐人返佣" }
//                             "fund" => { "资金费用" }
//                             "point_dnw" => { "点卡转入转出" }
//                             "point_fee" => { "点卡交易手续费" }
//                             "point_refr" => { "点卡推荐人返佣" }
//                             _ => {
//                                 "未知-变更类型"
//                             }
//                         };
//                         let text = item["text"].as_str().unwrap();
//                         let contract = item["contract"].as_str().unwrap();
//                         let trade_id = item["trade_id"].as_str().unwrap();
// 
//                         let mut name_data_array: Vec<String> = vec![];
//                         name_data_array.push(time_str.to_string());
//                         name_data_array.push(trade_id.to_string());
//                         name_data_array.push(change.to_string());
//                         name_data_array.push(balance.to_string());
//                         name_data_array.push(type_str.to_string());
//                         name_data_array.push(contract.to_string());
//                         name_data_array.push(text.to_string());
// 
//                         let mut name_data_array_clone: Vec<String> = vec![];
//                         name_data_array_clone.push(time.to_string());
//                         name_data_array_clone.push(trade_id.to_string());
//                         name_data_array_clone.push(change.to_string());
//                         name_data_array_clone.push(balance.to_string());
//                         name_data_array_clone.push(type_str.to_string());
//                         name_data_array_clone.push(contract.to_string());
//                         name_data_array_clone.push(text.to_string());
// 
//                         name_data_all.push(name_data_array.clone());
//                         name_data_all_clone.push(name_data_array_clone.clone());
//                         let mut cp =  name_data_array.clone();
//                         cp.push(acc_name.clone().unwrap().to_string());
//                         data_array_all.push(cp);
//                     }
//                     acc_all_data.insert(acc_name.clone().unwrap().to_string(), name_data_all.clone());
//                     acc_all_data_clone.insert(acc_name.clone().unwrap().to_string(), name_data_all_clone.clone());
//                 } else {
//                     trace!("不是数组 检查数据");
//                 }
//             }
//             // break;
//         }
//         break;//这里是为了 代码收纳，用了loop来放置代码
//     }
//     trace!("数据如下：{:?}",acc_all_data);
// 
//     //汇总
//     //1. 生成时间片
//     let mut unique = HashSet::new();
//     time_all.retain(|e| unique.insert(*e));
//     trace!("时间片：{:?}",time_all);
// 
//     //2. 根据时间片 去求每个时间片的  总余额，
//     let mut sum_data_array_all: Vec<Vec<String>> = vec![];
//     for time in time_all.clone() {
//         let mut sum_data_array: Vec<String> = vec![];
//         let mut sum_balance: f64 = 0 as f64;
//         for (key, value) in acc_all_data_clone.clone() {
//             let acc_key = key;
//             trace!("读取value：{:?}",value);
//             let filter_arrya: Vec<Vec<String>> = value.clone().into_iter().filter(|s| {
//                 trace!("读取time：{:?}",s);
//                 let time_v = s[0].clone();
//                 time_v < (time - 1).to_string()
//             }).collect();
// 
//             let balance: f64 = if filter_arrya.len() > 0 {
//                 let row = filter_arrya[filter_arrya.len() - 1].clone();
//                 trace!("读取balance：{:?}",row);
//                 let balance_clone = row[3].clone();
//                 balance_clone.parse::<f64>().unwrap()
//             } else {
//                 0 as f64
//             };
//             sum_balance = balance + sum_balance;
//         }
// 
//         // 使用秒构建一个DateTime<Utc>对象
//         let time_str = NaiveDateTime::from_timestamp_millis((time + 8 * 3600) * 1000).unwrap().format("%Y-%m-%d %H:%M:%S%.3f").to_string();
// 
//         sum_data_array.push(time_str.to_string());
//         sum_data_array.push("".to_string());
//         sum_data_array.push("".to_string());
//         sum_data_array.push(sum_balance.to_string());
//         sum_data_array.push("".to_string());
//         sum_data_array.push("".to_string());
//         sum_data_array.push("".to_string());
// 
//         sum_data_array_all.push(sum_data_array);
//     }
//     if sum_data_array_all.len() > 0 {
//         acc_all_data.insert("total".to_string(), sum_data_array_all);
//     }
//     if data_array_all.len() > 0{
//         acc_all_data.insert("gather".to_string(), data_array_all);
//     }
// 
//     //数据组装.
//     let noe_row_name = vec!["时间", "成交Id", "变更金额", "变更后余额", "变更类型", "合约标识", "注释"];
//     //创建表格
//     //提示。涉及到 需要转换的数据，请提前自行转换，工具只负责写入生成
//     // after - 之后的，时间戳> after 才是有效数据
//     match creation_xlsx(&noe_row_name, &acc_all_data,
//                         "okx".to_string(),
//     ) {
//         Ok(d) => {
//             trace!("完成");
//         }
//         Err(z) => {
//             eprint!("{:?}", z);
//             trace!("失败");
//         }
//     }
// }