as-rust/exchanges/tests/okx_swap_test.rs

// use std::collections::BTreeMap;
// use std::sync::Arc;
// use std::sync::atomic::AtomicBool;
// 
// use futures_util::StreamExt;
// use tokio::sync::Mutex;
// use tracing::trace;
// 
// use exchanges::okx_swap_rest::OkxSwapRest;
// use exchanges::okx_swap_ws::{OkxSwapLogin, OkxSwapSubscribeType, OkxSwapWs, OkxSwapWsType};
// 
// const ACCESS_KEY: &str = "";
// const SECRET_KEY: &str = "";
// const PASS_KEY: &str = "";
// 
// 
// //ws-订阅公共频道信息
// #[tokio::test(flavor = "multi_thread", worker_threads = 5)]
// async fn ws_custom_subscribe_pu() {
//     global::log_utils::init_log_with_trace();
// 
// 
//     let (write_tx, write_rx) = futures_channel::mpsc::unbounded();
//     let (read_tx, mut read_rx) = futures_channel::mpsc::unbounded();
// 
// 
//     let mut ws = get_ws(None, OkxSwapWsType::Public).await;
//     ws.set_symbols(vec!["BTC_USDT".to_string()]);
//     ws.set_subscribe(vec![
//         // OkxSwapSubscribeType::PuBooks5,
//         // OkxSwapSubscribeType::Putrades,
//         OkxSwapSubscribeType::PuBooks50L2tbt,
//         // OkxSwapSubscribeType::PuIndexTickers,
//     ]);
// 
// 
//     let write_tx_am = Arc::new(Mutex::new(write_tx));
//     let is_shutdown_arc = Arc::new(AtomicBool::new(true));
// 
//     //读取
//     let _is_shutdown_arc_clone = Arc::clone(&is_shutdown_arc);
//     let _tr = tokio::spawn(async move {
//         trace!("线程-数据读取-开启");
//         loop {
//             if let Some(data) = read_rx.next().await {
//                 trace!("读取数据data:{:?}",data)
//             }
//         }
//         // trace!("线程-数据读取-结束");
//     });
// 
//     //写数据
//     // let bool_v2_clone = Arc::clone(&is_shutdown_arc);
//     // let write_tx_clone = Arc::clone(&write_tx_am);
//     // let su = ws.get_subscription();
//     // let tw = tokio::spawn(async move {
//     //     trace!("线程-数据写入-开始");
//     //     loop {
//     //         tokio::time::sleep(Duration::from_millis(20 * 1000)).await;
//     //         // let close_frame = CloseFrame {
//     //         //     code: CloseCode::Normal,
//     //         //     reason: Cow::Borrowed("Bye bye"),
//     //         // };
//     //         // let message = Message::Close(Some(close_frame));
//     //
//     //
//     //         let message = Message::Text(su.clone());
//     //         AbstractWsMode::send_subscribe(write_tx_clone.clone(), message.clone()).await;
//     //         trace!("发送指令成功");
//     //     }
//     //     trace!("线程-数据写入-结束");
//     // });
// 
//     let t1 = tokio::spawn(async move {
//         //链接
//         let bool_v3_clone = Arc::clone(&is_shutdown_arc);
//         ws.ws_connect_async(bool_v3_clone, &write_tx_am, write_rx, read_tx).await.expect("链接失败(内部一个心跳线程应该已经关闭了)");
//         trace!("test 唯一线程结束--");
//     });
//     tokio::try_join!(t1).unwrap();
//     trace!("当此结束");
//     trace!("重启！");
//     trace!("参考交易所关闭");
//     return;
// }
// 
// //ws-订阅私有频道信息
// #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
// async fn ws_custom_subscribe_bu() {
//     global::log_utils::init_log_with_trace();
// 
//     let (write_tx, write_rx) = futures_channel::mpsc::unbounded();
//     let (read_tx, mut read_rx) = futures_channel::mpsc::unbounded();
// 
//     let mut ws = get_ws(None, OkxSwapWsType::Business).await;
//     ws.set_symbols(vec!["BTC-USD".to_string()]);
//     ws.set_subscribe(vec![
//         OkxSwapSubscribeType::BuIndexCandle30m,
//     ]);
// 
//     let write_tx_am = Arc::new(Mutex::new(write_tx));
//     let is_shutdown_arc = Arc::new(AtomicBool::new(true));
// 
//     //读取
//     let _is_shutdown_arc_clone = Arc::clone(&is_shutdown_arc);
//     let _tr = tokio::spawn(async move {
//         trace!("线程-数据读取-开启");
//         loop {
//             if let Some(data) = read_rx.next().await {
//                 trace!("读取数据data:{:?}",data)
//             }
//         }
//         // trace!("线程-数据读取-结束");
//     });
// 
//     //写数据
//     // let bool_v2_clone = Arc::clone(&is_shutdown_arc);
//     // let write_tx_clone = Arc::clone(&write_tx_am);
//     // let su = ws.get_subscription();
//     // let tw = tokio::spawn(async move {
//     //     trace!("线程-数据写入-开始");
//     //     loop {
//     //         tokio::time::sleep(Duration::from_millis(20 * 1000)).await;
//     //         // let close_frame = CloseFrame {
//     //         //     code: CloseCode::Normal,
//     //         //     reason: Cow::Borrowed("Bye bye"),
//     //         // };
//     //         // let message = Message::Close(Some(close_frame));
//     //
//     //
//     //         let message = Message::Text(su.clone());
//     //         AbstractWsMode::send_subscribe(write_tx_clone.clone(), message.clone()).await;
//     //         trace!("发送指令成功");
//     //     }
//     //     trace!("线程-数据写入-结束");
//     // });
// 
//     let t1 = tokio::spawn(async move {
//         //链接
//         let bool_v3_clone = Arc::clone(&is_shutdown_arc);
//         ws.ws_connect_async(bool_v3_clone, &write_tx_am, write_rx, read_tx).await.expect("链接失败(内部一个心跳线程应该已经关闭了)");
//         trace!("test 唯一线程结束--");
//     });
//     tokio::try_join!(t1).unwrap();
//     trace!("当此结束");
//     trace!("重启！");
//     trace!("参考交易所关闭");
//     return;
// }
// 
// //ws-订阅私有频道信息
// #[tokio::test(flavor = "multi_thread", worker_threads = 5)]
// async fn ws_custom_subscribe_pr() {
//     global::log_utils::init_log_with_trace();
// 
//     let (write_tx, write_rx) = futures_channel::mpsc::unbounded();
//     let (read_tx, mut read_rx) = futures_channel::mpsc::unbounded();
// 
//     let btree_map = OkxSwapLogin {
//         api_key: ACCESS_KEY.to_string(),
//         secret_key: SECRET_KEY.to_string(),
//         passphrase: PASS_KEY.to_string(),
//     };
//     let mut ws = get_ws(Option::from(btree_map), OkxSwapWsType::Private).await;
//     ws.set_symbols(vec!["BTC-USDT".to_string()]);
//     ws.set_subscribe(vec![
//         OkxSwapSubscribeType::PrAccount("USDT".to_string()),
//         OkxSwapSubscribeType::PrOrders,
//         OkxSwapSubscribeType::PrPositions,
//         OkxSwapSubscribeType::PrBalanceAndPosition,
//     ]);
// 
// 
//     let write_tx_am = Arc::new(Mutex::new(write_tx));
//     let is_shutdown_arc = Arc::new(AtomicBool::new(true));
// 
//     //读取
//     let _is_shutdown_arc_clone = Arc::clone(&is_shutdown_arc);
//     let _tr = tokio::spawn(async move {
//         trace!("线程-数据读取-开启");
//         loop {
//             if let Some(data) = read_rx.next().await {
//                 trace!("读取数据data:{:?}",data)
//             }
//         }
//         // trace!("线程-数据读取-结束");
//     });
// 
//     //写数据
//     // let bool_v2_clone = Arc::clone(&is_shutdown_arc);
//     // let write_tx_clone = Arc::clone(&write_tx_am);
//     // let su = ws.get_subscription();
//     // let tw = tokio::spawn(async move {
//     //     trace!("线程-数据写入-开始");
//     //     loop {
//     //         tokio::time::sleep(Duration::from_millis(20 * 1000)).await;
//     //         // let close_frame = CloseFrame {
//     //         //     code: CloseCode::Normal,
//     //         //     reason: Cow::Borrowed("Bye bye"),
//     //         // };
//     //         // let message = Message::Close(Some(close_frame));
//     //
//     //
//     //         let message = Message::Text(su.clone());
//     //         AbstractWsMode::send_subscribe(write_tx_clone.clone(), message.clone()).await;
//     //         trace!("发送指令成功");
//     //     }
//     //     trace!("线程-数据写入-结束");
//     // });
// 
//     let t1 = tokio::spawn(async move {
//         //链接
//         let bool_v3_clone = Arc::clone(&is_shutdown_arc);
//         ws.ws_connect_async(bool_v3_clone, &write_tx_am, write_rx, read_tx).await.expect("链接失败(内部一个心跳线程应该已经关闭了)");
//         trace!("test 唯一线程结束--");
//     });
//     tokio::try_join!(t1).unwrap();
//     trace!("当此结束");
//     trace!("重启！");
//     trace!("参考交易所关闭");
//     return;
//     //
//     // let mut is_shutdown_arc = Arc::new(AtomicBool::new(true));
//     // let mut btree_map: BTreeMap<String, String> = BTreeMap::new();
//     //
//     // btree_map.insert("access_key".to_string(), ACCESS_KEY.to_string());
//     // btree_map.insert("secret_key".to_string(), SECRET_KEY.to_string());
//     // btree_map.insert("pass_key".to_string(), PASS_KEY.to_string());
//     //
//     // let (tx, mut rx) = channel(1024);
//     // let mut ws = get_ws(btree_map, OkxWsType::Private, tx).await;
//     // ws.set_subscribe(vec![
//     //     OkxSubscribeType::PrBalanceAndPosition,
//     //     // OkxSubscribeType::PrAccount("USDT".to_string()),
//     //     OkxSubscribeType::PrOrders,
//     //     OkxSubscribeType::PrPositions,
//     // ]);
//     //
//     // let t1 = tokio::spawn(async move {
//     //     ws.custom_subscribe(is_shutdown_arc, vec!["BTC-USDT".to_string()]).await;
//     // });
//     //
//     // let t2 = tokio::spawn(async move {
//     //     loop {
//     //         if let Ok(received) = rx.try_recv() {
//     //             trace!( "age: {:?}", received);
//     //         }
//     //     }
//     // });
//     // try_join!(t1,t2).unwrap();
// }
// 
// 
// //rest-订单查询
// #[tokio::test]
// async fn rest_get_order_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_order("BTC-USDT".to_string(), "3333".to_string(), "".to_string()).await;
//     println!("okx--订单查询--{:?}", req_data);
// }
// 
// 
// //rest-未完成的订单
// #[tokio::test]
// async fn rest_get_incomplete_order_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_incomplete_order("BTC-USDT".to_string()).await;
//     println!("okx--未完成的订单--{:?}", req_data);
// }
// 
// //rest-获取系统时间
// #[tokio::test]
// async fn rest_get_server_time_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_server_time().await;
//     println!("okx--获取系统时间--{:?}", req_data);
// }
// 
// //rest-查看持仓信息
// #[tokio::test]
// async fn rest_get_positions_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_positions("SWA1P".to_string()).await;
//     println!("okx--查看持仓信息--{:?}", req_data);
// }
// 
// //rest-获取单个产品行情信息
// #[tokio::test]
// async fn rest_get_ticker_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_ticker("BTC-USDT".to_string()).await;
//     println!("okx--获取单个产品行情信息--{:?}", req_data);
// }
// 
// //rest-查看账户余额
// #[tokio::test]
// async fn rest_get_balance_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_balance("BTC,ETH".to_string()).await;
//     println!("okx--查看账户余额--{:?}", req_data);
// }
// 
// //rest-获取交易产品基础信息
// #[tokio::test]
// async fn rest_get_instruments_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_instruments().await;
//     println!("okx--获取交易产品基础信息--{:?}", req_data);
// }
// 
// //rest-获取成交明细（近三天）
// #[tokio::test]
// async fn rest_get_trade_fills_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_trade_fills("".to_string()).await;
//     println!("okx--获取成交明细（近三天）--{:?}", req_data);
// }
// 
// //rest-撤单
// #[tokio::test]
// async fn rest_cancel_order_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.cancel_order("BTC-USD".to_string(), "1111".to_string(), "".to_string()).await;
//     println!("okx--撤单--{:?}", req_data);
// }
// 
// //rest-设置杠杆倍数
// #[tokio::test]
// async fn rest_set_leverage_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.set_leverage("BTC-USDT".to_string(), "5".to_string()).await;
//     println!("okx--设置杠杆倍数--{:?}", req_data);
// }
// 
// //rest-设置持仓模式
// #[tokio::test]
// async fn rest_set_position_mode_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.set_position_mode().await;
//     println!("okx--设置持仓模式--{:?}", req_data);
// }
// 
// //rest-获取历史订单记录（近七天)
// #[tokio::test]
// async fn rest_get_orders_history_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_orders_history("".to_string(),
//                                           "".to_string(),
//                                           "filled".to_string(),
//                                           "".to_string(),
//                                           "".to_string(),
//                                           "".to_string(),
//     ).await;
//     println!("okx--获取历史订单记录--{:?}", req_data);
// }
// 
// //rest-获取历史成交数据（近七天）
// #[tokio::test]
// async fn rest_get_trades_test() {
//     global::log_utils::init_log_with_trace();
// 
//     let mut ret = get_rest();
//     let req_data = ret.get_trades("".to_string(),
//                                   "".to_string(),
//                                   "".to_string(),
//                                   "".to_string(),
//                                   "".to_string(),
//                                   "100".to_string(),
//     ).await;
//     println!("okx--获取历史成交数据--{:?}", req_data);
// }
// 
// 
// async fn get_ws(btree_map: Option<OkxSwapLogin>, type_v: OkxSwapWsType) -> OkxSwapWs {
//     let ku_ws = OkxSwapWs::new(false, btree_map, type_v);
//     ku_ws
// }
// 
// fn get_rest() -> OkxSwapRest {
//     let mut btree_map: BTreeMap<String, String> = BTreeMap::new();
//     btree_map.insert("access_key".to_string(), ACCESS_KEY.to_string());
//     btree_map.insert("secret_key".to_string(), SECRET_KEY.to_string());
//     btree_map.insert("pass_key".to_string(), PASS_KEY.to_string());
// 
//     let okx_exc = OkxSwapRest::new(false, btree_map.clone());
//     okx_exc
// }