as-rust/exchanges/tests/kucoin_spot_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::kucoin_spot_rest::KucoinSpotRest;
use exchanges::kucoin_spot_ws::{KucoinSpotLogin, KucoinSpotSubscribeType, KucoinSpotWs, KucoinSpotWsType};

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, KucoinSpotWsType::Public).await;
    ws.set_symbols(vec!["BTC-USDT".to_string()]);
    ws.set_subscribe(vec![
        KucoinSpotSubscribeType::PuSpotMarketLevel2Depth50,
        KucoinSpotSubscribeType::PuMarketTicker,
        KucoinSpotSubscribeType::PuMarketMatch,
    ]);

    let write_tx_am = Arc::new(Mutex::new(write_tx));
    let bool_v1 = Arc::new(AtomicBool::new(true));

    //读取
    let _bool_v1_clone = Arc::clone(&bool_v1);
    let _tr = tokio::spawn(async move {
        trace!("线程-数据读取-开启");
        loop {
            if let Some(data) = read_rx.next().await {
                trace!("读取数据data:{:?}",data)
            }
        }
    });

    //写数据
    let _bool_v2_clone = Arc::clone(&bool_v1);
    let _write_tx_clone = Arc::clone(&write_tx_am);
    // 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 close_message = Message::Close(Some(close_frame));
    //         // AbstractWsMode::send_subscribe(write_tx_clone.clone(), Message::Text("32313221".to_string()));
    //         AbstractWsMode::send_subscribe(write_tx_clone.clone(), close_message);
    //         trace!("发送指令成功");
    //     }
    //     trace!("线程-数据写入-结束");
    // });

    // loop {
    let t1 = tokio::spawn(async move {
        //链接
        let bool_v3_clone = Arc::clone(&bool_v1);
        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_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 = KucoinSpotLogin {
        access_key: ACCESS_KEY.to_string(),
        secret_key: SECRET_KEY.to_string(),
        pass_key: PASS_KEY.to_string(),
    };
    let mut ws = get_ws(Option::from(btree_map), KucoinSpotWsType::Public).await;
    ws.set_symbols(vec!["BTC-USDT".to_string()]);
    ws.set_subscribe(vec![
        KucoinSpotSubscribeType::PrAccountBalance,
        KucoinSpotSubscribeType::PrSpotMarketTradeOrders,
    ]);

    let write_tx_am = Arc::new(Mutex::new(write_tx));
    let bool_v1 = Arc::new(AtomicBool::new(true));

    //读取
    let _bool_v1_clone = Arc::clone(&bool_v1);
    let _tr = tokio::spawn(async move {
        trace!("线程-数据读取-开启");
        loop {
            if let Some(data) = read_rx.next().await {
                trace!("读取数据data:{:?}",data)
            }
        }
    });

    //写数据
    let _bool_v2_clone = Arc::clone(&bool_v1);
    let _write_tx_clone = Arc::clone(&write_tx_am);
    // 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 close_message = Message::Close(Some(close_frame));
    //         // AbstractWsMode::send_subscribe(write_tx_clone.clone(), Message::Text("32313221".to_string()));
    //         AbstractWsMode::send_subscribe(write_tx_clone.clone(), close_message);
    //         trace!("发送指令成功");
    //     }
    //     trace!("线程-数据写入-结束");
    // });

    // loop {
    let t1 = tokio::spawn(async move {
        //链接
        let bool_v3_clone = Arc::clone(&bool_v1);
        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;
}


//rest-获取成交记录
#[tokio::test]
async fn rest_get_fills_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_fills("BTC-USDT".to_string(),
                                  "".to_string(),
                                  "".to_string(),
                                  -1,
                                  -1,
                                  500,
    ).await;
    trace!(?rep_data)
}


//rest-获取订单
#[tokio::test]
async fn rest_get_order_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_order().await;
    trace!(?rep_data)
}

//rest-獲取行情
#[tokio::test]
async fn rest_get_level1_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_level1("BTC-USDT".to_string()).await;
    trace!(?rep_data)
}

//rest-通過orderId获取訂單詳情
#[tokio::test]
async fn rest_get_order_by_order_id_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_order_by_order_id("3123123".to_string()).await;
    trace!(?rep_data)
}

//rest-通過clientOid獲取訂單詳情
#[tokio::test]
async fn rest_get_order_by_client_id_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_order_by_client_id("3123123".to_string()).await;
    trace!(?rep_data)
}

//rest-獲取賬戶列表 - 現貨/槓桿/現貨高頻
#[tokio::test]
async fn rest_get_accounts_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_accounts("USDT".to_string()).await;
    trace!(?rep_data)
}

//rest-獲取交易對列表
#[tokio::test]
async fn rest_get_symbols_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.get_symbols().await;
    trace!(?rep_data)
}

//rest-通過orderId撤單- 没权限需要查看
#[tokio::test]
async fn rest_cancel_order_by_order_id_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.cancel_order_by_order_id("dddd123131".to_string()).await;
    trace!(?rep_data)
}

//rest-通過clientOid撤單- 没权限需要查看
#[tokio::test]
async fn rest_cancel_order_by_client_id_test() {
    global::log_utils::init_log_with_trace();

    let mut rest = get_rest();
    let rep_data = rest.cancel_order_by_client_id("dddd123131".to_string()).await;
    trace!(?rep_data)
}


async fn get_ws(btree_map: Option<KucoinSpotLogin>, type_v: KucoinSpotWsType) -> KucoinSpotWs {
    let ku_ws = KucoinSpotWs::new(false, btree_map,
                                  type_v).await;
    ku_ws
}

fn get_rest() -> KucoinSpotRest {
    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("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 ku_exc = KucoinSpotRest::new(false, btree_map);
    ku_exc
}