|
|
@@ -1,25 +1,28 @@
|
|
|
use std::collections::{BTreeMap};
|
|
|
-use std::{env};
|
|
|
+// use std::{env};
|
|
|
use std::io::{Error};
|
|
|
-use std::sync::Arc;
|
|
|
-use std::sync::atomic::AtomicBool;
|
|
|
-use std::time::Duration;
|
|
|
-use rust_decimal_macros::dec;
|
|
|
+// use std::sync::Arc;
|
|
|
+// use std::sync::atomic::AtomicBool;
|
|
|
+// use std::time::Duration;
|
|
|
+// use rust_decimal_macros::dec;
|
|
|
use tokio::sync::mpsc::{channel, Receiver, Sender};
|
|
|
-use tokio::try_join;
|
|
|
-use tracing::{error, trace};
|
|
|
+// use tokio::try_join;
|
|
|
+// use tracing::{error, trace};
|
|
|
+use tracing::{error};
|
|
|
+use exchanges::kucoin_spot_ws::KucoinSpotSubscribeType;
|
|
|
// use exchanges::bitget_spot_ws::{BitgetSpotWs, BitgetSubscribeType, BitgetWsType};
|
|
|
-use exchanges::kucoin_spot_ws::{KucoinSpotWs, KucoinSubscribeType, KucoinWsType};
|
|
|
+// use exchanges::kucoin_spot_ws::{KucoinSpotWs, KucoinSpotSubscribeType, KucoinSpotWsType};
|
|
|
// use exchanges::binance_spot_ws::{BinanceSpotSubscribeType, BinanceSpotWs, BinanceSpotWsType};
|
|
|
// use exchanges::okx_swap_ws::{OkxSubscribeType, OkxSwapWs, OkxWsType};
|
|
|
// use exchanges::kucoin_swap_ws::{KucoinSubscribeType, KucoinSwapWs, KucoinWsType};
|
|
|
-use exchanges::response_base::ResponseData;
|
|
|
+// use exchanges::response_base::ResponseData;
|
|
|
use standard::exchange::{Exchange, ExchangeEnum};
|
|
|
// use standard::{binance_handle, Order, Platform, utils};
|
|
|
// use standard::{okx_handle, Order, Platform, utils};
|
|
|
// use standard::{kucoin_handle, Order, Platform, utils};
|
|
|
-use standard::{kucoin_spot_handle, Order, Platform, utils};
|
|
|
+// use standard::{kucoin_spot_handle, Order, Platform, utils};
|
|
|
// use standard::{bitget_spot_handle, Order, Platform, utils};
|
|
|
+use standard::{Order, Platform, utils};
|
|
|
|
|
|
// 创建实体
|
|
|
#[allow(dead_code)]
|
|
|
@@ -106,7 +109,7 @@ pub async fn test_new_exchange(exchange: ExchangeEnum, symbol: &str) -> Box<dyn
|
|
|
}
|
|
|
|
|
|
#[allow(dead_code)]
|
|
|
-pub async fn test_new_exchange_wss<T>(exchange: ExchangeEnum, symbol: &str, subscriber_type: T, mold: &str) where Vec<KucoinSubscribeType>: From<T> {
|
|
|
+pub async fn test_new_exchange_wss<T>(exchange: ExchangeEnum, _symbol: &str, _subscriber_type: T, _mold: &str) where Vec<KucoinSpotSubscribeType>: From<T> {
|
|
|
utils::proxy_handle();
|
|
|
|
|
|
match exchange {
|
|
|
@@ -234,73 +237,73 @@ pub async fn test_new_exchange_wss<T>(exchange: ExchangeEnum, symbol: &str, subs
|
|
|
// try_join!(t1, t2).unwrap();
|
|
|
}
|
|
|
ExchangeEnum::KucoinSpot => {
|
|
|
- let symbol_format = utils::format_symbol(symbol.to_string(), "-").to_uppercase();
|
|
|
- let symbol_back = utils::format_symbol(symbol.to_string(), "_");
|
|
|
- trace!(symbol_format);
|
|
|
- let name = format!("kucoin_usdt_spot@{}", symbol.to_string().to_lowercase());
|
|
|
- let bool_v1 = Arc::new(AtomicBool::new(true));
|
|
|
-
|
|
|
- let (res_sender, mut res_receiver): (Sender<ResponseData>, Receiver<ResponseData>) = channel(1024);
|
|
|
- let mut params: BTreeMap<String, String> = BTreeMap::new();
|
|
|
- let access_key = env::var("kucoin_access_key").unwrap_or("".to_string());
|
|
|
- let secret_key = env::var("kucoin_secret_key").unwrap_or("".to_string());
|
|
|
- let pass_key = env::var("kucoin_pass_key").unwrap_or("".to_string());
|
|
|
- params.insert("access_key".to_string(), access_key);
|
|
|
- params.insert("secret_key".to_string(), secret_key);
|
|
|
- params.insert("pass_key".to_string(), pass_key);
|
|
|
-
|
|
|
- let mut exchange_wss = if ["depth", "ticker"].contains(&mold) {
|
|
|
- KucoinSpotWs::new_label(name, false, params, KucoinWsType::Public, res_sender).await
|
|
|
- } else {
|
|
|
- KucoinSpotWs::new_label(name, false, params, KucoinWsType::Private, res_sender).await
|
|
|
- };
|
|
|
-
|
|
|
- exchange_wss.set_subscribe(subscriber_type.into());
|
|
|
-
|
|
|
- let t1 = tokio::spawn(async move {
|
|
|
- exchange_wss.custom_subscribe(bool_v1, vec![symbol_format]).await;
|
|
|
- });
|
|
|
- let mold_arc = Arc::new(mold.to_string());
|
|
|
- let t2 = tokio::spawn(async move {
|
|
|
- let mold_clone = Arc::clone(&mold_arc);
|
|
|
- loop {
|
|
|
- tokio::time::sleep(Duration::from_millis(1)).await;
|
|
|
- if let Ok(received) = res_receiver.try_recv() {
|
|
|
- trace!(?received);
|
|
|
- match mold_clone.as_str() {
|
|
|
- "depth" => {
|
|
|
- if received.data != "" {
|
|
|
- let result = kucoin_spot_handle::handle_special_depth(received);
|
|
|
- trace!(?result)
|
|
|
- }
|
|
|
- }
|
|
|
- "ticker" => {
|
|
|
- if received.data != "" {
|
|
|
- let result = kucoin_spot_handle::handle_special_ticker(received);
|
|
|
- trace!(?result)
|
|
|
- }
|
|
|
- }
|
|
|
- "account" => {
|
|
|
- if received.data != "" {
|
|
|
- let result = kucoin_spot_handle::handle_account_info(received, symbol_back.clone());
|
|
|
- trace!(?result)
|
|
|
- }
|
|
|
- }
|
|
|
- "orders" => {
|
|
|
- if received.data != "" {
|
|
|
- let result = kucoin_spot_handle::handle_order(received, dec!(1));
|
|
|
- trace!(?result)
|
|
|
- }
|
|
|
- }
|
|
|
- _ => {
|
|
|
- error!("没有该命令!mode={}", mold_clone);
|
|
|
- panic!("没有该命令!mode={}", mold_clone)
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- });
|
|
|
- try_join!(t1, t2).unwrap();
|
|
|
+ // let symbol_format = utils::format_symbol(symbol.to_string(), "-").to_uppercase();
|
|
|
+ // let symbol_back = utils::format_symbol(symbol.to_string(), "_");
|
|
|
+ // trace!(symbol_format);
|
|
|
+ // let name = format!("kucoin_usdt_spot@{}", symbol.to_string().to_lowercase());
|
|
|
+ // let bool_v1 = Arc::new(AtomicBool::new(true));
|
|
|
+ //
|
|
|
+ // let (res_sender, mut res_receiver): (Sender<ResponseData>, Receiver<ResponseData>) = channel(1024);
|
|
|
+ // let mut params: BTreeMap<String, String> = BTreeMap::new();
|
|
|
+ // let access_key = env::var("kucoin_access_key").unwrap_or("".to_string());
|
|
|
+ // let secret_key = env::var("kucoin_secret_key").unwrap_or("".to_string());
|
|
|
+ // let pass_key = env::var("kucoin_pass_key").unwrap_or("".to_string());
|
|
|
+ // params.insert("access_key".to_string(), access_key);
|
|
|
+ // params.insert("secret_key".to_string(), secret_key);
|
|
|
+ // params.insert("pass_key".to_string(), pass_key);
|
|
|
+ //
|
|
|
+ // let mut exchange_wss = if ["depth", "ticker"].contains(&mold) {
|
|
|
+ // KucoinSpotWs::new_label(name, false, params, KucoinSpotWsType::Public).await
|
|
|
+ // } else {
|
|
|
+ // KucoinSpotWs::new_label(name, false, params, KucoinSpotWsType::Private).await
|
|
|
+ // };
|
|
|
+ //
|
|
|
+ // exchange_wss.set_subscribe(subscriber_type.into());
|
|
|
+ //
|
|
|
+ // let t1 = tokio::spawn(async move {
|
|
|
+ // exchange_wss.custom_subscribe(bool_v1, vec![symbol_format]).await;
|
|
|
+ // });
|
|
|
+ // let mold_arc = Arc::new(mold.to_string());
|
|
|
+ // let t2 = tokio::spawn(async move {
|
|
|
+ // let mold_clone = Arc::clone(&mold_arc);
|
|
|
+ // loop {
|
|
|
+ // tokio::time::sleep(Duration::from_millis(1)).await;
|
|
|
+ // if let Ok(received) = res_receiver.try_recv() {
|
|
|
+ // trace!(?received);
|
|
|
+ // match mold_clone.as_str() {
|
|
|
+ // "depth" => {
|
|
|
+ // if received.data != "" {
|
|
|
+ // let result = kucoin_spot_handle::handle_special_depth(received);
|
|
|
+ // trace!(?result)
|
|
|
+ // }
|
|
|
+ // }
|
|
|
+ // "ticker" => {
|
|
|
+ // if received.data != "" {
|
|
|
+ // let result = kucoin_spot_handle::handle_special_ticker(received);
|
|
|
+ // trace!(?result)
|
|
|
+ // }
|
|
|
+ // }
|
|
|
+ // "account" => {
|
|
|
+ // if received.data != "" {
|
|
|
+ // let result = kucoin_spot_handle::handle_account_info(received, symbol_back.clone());
|
|
|
+ // trace!(?result)
|
|
|
+ // }
|
|
|
+ // }
|
|
|
+ // "orders" => {
|
|
|
+ // if received.data != "" {
|
|
|
+ // let result = kucoin_spot_handle::handle_order(received, dec!(1));
|
|
|
+ // trace!(?result)
|
|
|
+ // }
|
|
|
+ // }
|
|
|
+ // _ => {
|
|
|
+ // error!("没有该命令!mode={}", mold_clone);
|
|
|
+ // panic!("没有该命令!mode={}", mold_clone)
|
|
|
+ // }
|
|
|
+ // }
|
|
|
+ // }
|
|
|
+ // }
|
|
|
+ // });
|
|
|
+ // try_join!(t1, t2).unwrap();
|
|
|
}
|
|
|
ExchangeEnum::BitgetSpot => {
|
|
|
// let symbol_format = utils::format_symbol(symbol.to_string(), "-").to_uppercase();
|