bitget_spot_test.rs 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515
  1. // use std::collections::BTreeMap;
  2. // use std::sync::Arc;
  3. // use std::sync::atomic::AtomicBool;
  4. //
  5. // use futures_util::StreamExt;
  6. // use tokio::sync::Mutex;
  7. // use tracing::trace;
  8. //
  9. // use exchanges::bitget_spot_rest::BitgetSpotRest;
  10. // use exchanges::bitget_spot_ws::{BitgetSpotLogin, BitgetSpotSubscribeType, BitgetSpotWs, BitgetSpotWsType};
  11. //
  12. // const ACCESS_KEY: &str = "";
  13. // const SECRET_KEY: &str = "";
  14. // const PASS_KEY: &str = "";
  15. //
  16. // //ws-订阅公共频道信息
  17. // #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
  18. // async fn ws_custom_subscribe_pu() {
  19. // global::log_utils::init_log_with_trace();
  20. //
  21. // let (write_tx, write_rx) = futures_channel::mpsc::unbounded();
  22. // let (read_tx, mut read_rx) = futures_channel::mpsc::unbounded();
  23. //
  24. // let is_shutdown_arc = Arc::new(AtomicBool::new(true));
  25. // let mut ws = get_ws(None, BitgetSpotWsType::Public).await;
  26. // ws.set_symbols(vec!["BTC_USDT".to_string()]);
  27. // ws.set_subscribe(vec![
  28. // BitgetSpotSubscribeType::PuTicker,
  29. // BitgetSpotSubscribeType::PuCandle1m,
  30. // BitgetSpotSubscribeType::PuTrade,
  31. // BitgetSpotSubscribeType::PuBooks5,
  32. // ]);
  33. //
  34. // let write_tx_am = Arc::new(Mutex::new(write_tx));
  35. // let is_shutdown_arc = Arc::new(AtomicBool::new(true));
  36. //
  37. // //读取
  38. // let _is_shutdown_arc_clone = Arc::clone(&is_shutdown_arc);
  39. // let _tr = tokio::spawn(async move {
  40. // trace!("线程-数据读取-开启");
  41. // loop {
  42. // if let Some(data) = read_rx.next().await {
  43. // trace!("读取数据data:{:?}",data)
  44. // }
  45. // }
  46. // // trace!("线程-数据读取-结束");
  47. // });
  48. //
  49. // //写数据
  50. // // let bool_v2_clone = Arc::clone(&is_shutdown_arc);
  51. // // let write_tx_clone = Arc::clone(&write_tx_am);
  52. // // let su = ws.get_subscription();
  53. // // let tw = tokio::spawn(async move {
  54. // // trace!("线程-数据写入-开始");
  55. // // loop {
  56. // // tokio::time::sleep(Duration::from_millis(20 * 1000)).await;
  57. // // // let close_frame = CloseFrame {
  58. // // // code: CloseCode::Normal,
  59. // // // reason: Cow::Borrowed("Bye bye"),
  60. // // // };
  61. // // // let message = Message::Close(Some(close_frame));
  62. // //
  63. // //
  64. // // let message = Message::Text(su.clone());
  65. // // AbstractWsMode::send_subscribe(write_tx_clone.clone(), message.clone()).await;
  66. // // trace!("发送指令成功");
  67. // // }
  68. // // trace!("线程-数据写入-结束");
  69. // // });
  70. //
  71. // let t1 = tokio::spawn(async move {
  72. // //链接
  73. // let bool_v3_clone = Arc::clone(&is_shutdown_arc);
  74. // ws.ws_connect_async(bool_v3_clone, &write_tx_am, write_rx, read_tx).await.expect("链接失败(内部一个心跳线程应该已经关闭了)");
  75. // trace!("test 唯一线程结束--");
  76. // });
  77. // tokio::try_join!(t1).unwrap();
  78. // trace!("当此结束");
  79. // trace!("重启!");
  80. // trace!("参考交易所关闭");
  81. // return;
  82. // }
  83. //
  84. // //ws-订阅私有频道信息
  85. // #[tokio::test(flavor = "multi_thread", worker_threads = 2)]
  86. // async fn ws_custom_subscribe_pr() {
  87. // global::log_utils::init_log_with_trace();
  88. //
  89. //
  90. // let (write_tx, write_rx) = futures_channel::mpsc::unbounded();
  91. // let (read_tx, mut read_rx) = futures_channel::mpsc::unbounded();
  92. //
  93. // let login_param = BitgetSpotLogin {
  94. // api_key: ACCESS_KEY.to_string(),
  95. // secret_key: SECRET_KEY.to_string(),
  96. // passphrase_key: PASS_KEY.to_string(),
  97. // };
  98. // let mut ws = get_ws(None, BitgetSpotWsType::Private).await;
  99. // ws.set_symbols(vec!["BTC_USDT".to_string()]);
  100. // ws.set_subscribe(vec![
  101. // BitgetSpotSubscribeType::PuTicker,
  102. // BitgetSpotSubscribeType::PuCandle1m,
  103. // BitgetSpotSubscribeType::PuTrade,
  104. // BitgetSpotSubscribeType::PuBooks5,
  105. // ]);
  106. //
  107. // let write_tx_am = Arc::new(Mutex::new(write_tx));
  108. // let is_shutdown_arc = Arc::new(AtomicBool::new(true));
  109. //
  110. // //读取
  111. // let _is_shutdown_arc_clone = Arc::clone(&is_shutdown_arc);
  112. // let _tr = tokio::spawn(async move {
  113. // trace!("线程-数据读取-开启");
  114. // loop {
  115. // if let Some(data) = read_rx.next().await {
  116. // trace!("读取数据data:{:?}",data)
  117. // }
  118. // }
  119. // // trace!("线程-数据读取-结束");
  120. // });
  121. //
  122. // //写数据
  123. // // let bool_v2_clone = Arc::clone(&is_shutdown_arc);
  124. // // let write_tx_clone = Arc::clone(&write_tx_am);
  125. // // let su = ws.get_subscription();
  126. // // let tw = tokio::spawn(async move {
  127. // // trace!("线程-数据写入-开始");
  128. // // loop {
  129. // // tokio::time::sleep(Duration::from_millis(20 * 1000)).await;
  130. // // // let close_frame = CloseFrame {
  131. // // // code: CloseCode::Normal,
  132. // // // reason: Cow::Borrowed("Bye bye"),
  133. // // // };
  134. // // // let message = Message::Close(Some(close_frame));
  135. // //
  136. // //
  137. // // let message = Message::Text(su.clone());
  138. // // AbstractWsMode::send_subscribe(write_tx_clone.clone(), message.clone()).await;
  139. // // trace!("发送指令成功");
  140. // // }
  141. // // trace!("线程-数据写入-结束");
  142. // // });
  143. //
  144. // let t1 = tokio::spawn(async move {
  145. // //链接
  146. // let bool_v3_clone = Arc::clone(&is_shutdown_arc);
  147. // ws.ws_connect_async(bool_v3_clone, &write_tx_am, write_rx, read_tx).await.expect("链接失败(内部一个心跳线程应该已经关闭了)");
  148. // trace!("test 唯一线程结束--");
  149. // });
  150. // tokio::try_join!(t1).unwrap();
  151. // trace!("当此结束");
  152. // trace!("重启!");
  153. // trace!("参考交易所关闭");
  154. // return;
  155. // }
  156. //
  157. //
  158. // //rest-获取系统时间
  159. // #[tokio::test]
  160. // async fn rest_get_server_time_test() {
  161. // global::log_utils::init_log_with_trace();
  162. //
  163. // let mut rest = get_rest();
  164. // let rep_data = rest.get_server_time().await;
  165. // trace!(?rep_data)
  166. // }
  167. //
  168. //
  169. // //rest-获取账户信息
  170. // #[tokio::test]
  171. // async fn rest_get_account_info_test() {
  172. // global::log_utils::init_log_with_trace();
  173. //
  174. // let mut rest = get_rest();
  175. // let rep_data = rest.get_account_info().await;
  176. // trace!(?rep_data)
  177. // }
  178. //
  179. // //rest-获取账户币种资产
  180. // #[tokio::test]
  181. // async fn rest_get_account_assets_test() {
  182. // global::log_utils::init_log_with_trace();
  183. //
  184. // let mut rest = get_rest();
  185. // let rep_data = rest.get_account_assets().await;
  186. // trace!(?rep_data)
  187. // }
  188. //
  189. // //rest-获取币种信息
  190. // #[tokio::test]
  191. // async fn rest_get_coins_test() {
  192. // global::log_utils::init_log_with_trace();
  193. //
  194. // let mut rest = get_rest();
  195. // let rep_data = rest.get_coins("USDT".to_string()).await;
  196. // trace!(?rep_data)
  197. // }
  198. //
  199. // //rest-获取交易对信息
  200. // #[tokio::test]
  201. // async fn rest_get_symbols_test() {
  202. // global::log_utils::init_log_with_trace();
  203. //
  204. // let mut rest = get_rest();
  205. // let rep_data = rest.get_symbols("BTCUSDT".to_string()).await;
  206. // trace!(?rep_data)
  207. // }
  208. //
  209. // //rest-获取现货VIP费率
  210. // #[tokio::test]
  211. // async fn rest_get_vip_fee_rate_test() {
  212. // global::log_utils::init_log_with_trace();
  213. //
  214. // let mut rest = get_rest();
  215. // let rep_data = rest.get_vip_fee_rate().await;
  216. // trace!(?rep_data)
  217. // }
  218. //
  219. // //rest-获取行情信息
  220. // #[tokio::test]
  221. // async fn rest_get_tickers_test() {
  222. // global::log_utils::init_log_with_trace();
  223. //
  224. // let mut rest = get_rest();
  225. // let rep_data = rest.get_tickers("BTCUSDT".to_string()).await;
  226. // trace!(?rep_data)
  227. // }
  228. //
  229. // //rest-获取合并交易深度
  230. // #[tokio::test]
  231. // async fn rest_get_merge_depth_test() {
  232. // global::log_utils::init_log_with_trace();
  233. //
  234. // let mut rest = get_rest();
  235. // let rep_data = rest.get_merge_depth("BTCUSDT".to_string()).await;
  236. // trace!(?rep_data)
  237. // }
  238. //
  239. // //rest-获取K线数据
  240. // #[tokio::test]
  241. // async fn rest_get_candles_test() {
  242. // global::log_utils::init_log_with_trace();
  243. //
  244. // let mut rest = get_rest();
  245. // let rep_data = rest.get_candles("BTCUSDT".to_string(), "1min".to_string(), "1697701550192".to_string(), "1697701556192".to_string(), "100".to_string()).await;
  246. // trace!(?rep_data)
  247. // }
  248. //
  249. // //rest-获取历史K线数据
  250. // #[tokio::test]
  251. // async fn rest_get_history_candles_test() {
  252. // global::log_utils::init_log_with_trace();
  253. //
  254. // let mut rest = get_rest();
  255. // let rep_data = rest.get_history_candles("BTCUSDT".to_string(), "1min".to_string(), "1697701556192".to_string(), "100".to_string()).await;
  256. // trace!(?rep_data)
  257. // }
  258. //
  259. // //rest-获取最近成交数据
  260. // #[tokio::test]
  261. // async fn rest_get_market_fills_test() {
  262. // global::log_utils::init_log_with_trace();
  263. //
  264. // let mut rest = get_rest();
  265. // let rep_data = rest.get_market_fills("BTCUSDT".to_string(), "100".to_string()).await;
  266. // trace!(?rep_data)
  267. // }
  268. //
  269. // //rest-获取历史成交数据
  270. // #[tokio::test]
  271. // async fn rest_get_fills_history_test() {
  272. // global::log_utils::init_log_with_trace();
  273. //
  274. // let mut rest = get_rest();
  275. // let rep_data = rest.get_market_fills_history("BTCUSDT".to_string(), "1697701550192".to_string(), "1697701556192".to_string(), "100".to_string()).await;
  276. // trace!(?rep_data)
  277. // }
  278. //
  279. // //rest-下单
  280. // #[tokio::test]
  281. // async fn rest_spot_order_test() {
  282. // global::log_utils::init_log_with_trace();
  283. //
  284. // //市价单
  285. // let mut rest = get_rest();
  286. // let params = serde_json::json!({
  287. // // "symbol":"CELRUSDT",
  288. // // "side":"sell",
  289. // // "orderType":"market",
  290. // // "force":"fok",
  291. // // "size":"887",
  292. // // "clientOid":"7d8zd4d_3",
  293. // });
  294. //
  295. // //限价单
  296. // let params = serde_json::json!({
  297. // // "symbol":"CELRUSDT",
  298. // // "side":"buy",
  299. // // "orderType":"limit",
  300. // // "force":"gtc",
  301. // // "price":"0.01001",
  302. // // "size":"10",
  303. // // "clientOid":"7d8zd4d_z1",
  304. // });
  305. //
  306. // let rep_data = rest.spot_order(params).await;
  307. // trace!(?rep_data)
  308. // }
  309. //
  310. //
  311. // //rest-撤单
  312. // #[tokio::test]
  313. // async fn rest_spot_cancel_order_test() {
  314. // global::log_utils::init_log_with_trace();
  315. // let mut rest = get_rest();
  316. // let rep_data = rest.spot_cancel_order("CELRUSDT".to_string(), "".to_string(), "1".to_string()).await;
  317. // trace!(?rep_data)
  318. // }
  319. //
  320. // //rest-批量撤单
  321. // #[tokio::test]
  322. // async fn rest_spot_cancel_orders_test() {
  323. // global::log_utils::init_log_with_trace();
  324. // let mut rest = get_rest();
  325. // let v = serde_json::json!({
  326. // "orderId":"1073370944162058240",
  327. // "clientOid":"1073370944162058240"
  328. // });
  329. // let rep_data = rest.spot_cancel_orders("CELRUSDT".to_string(), vec![v]).await;
  330. // trace!(?rep_data)
  331. // }
  332. //
  333. // //rest-按币对撤单
  334. // #[tokio::test]
  335. // async fn rest_spot_cancel_symbol_orders_test() {
  336. // global::log_utils::init_log_with_trace();
  337. // let mut rest = get_rest();
  338. // let rep_data = rest.spot_cancel_symbol_orders("CELRUSDT".to_string()).await;
  339. // trace!(?rep_data)
  340. // }
  341. //
  342. // //rest-获取订单详情
  343. // #[tokio::test]
  344. // async fn rest_get_order_test() {
  345. // global::log_utils::init_log_with_trace();
  346. // let mut rest = get_rest();
  347. // let rep_data = rest.get_order("".to_string(), "1".to_string()).await;
  348. // trace!(?rep_data)
  349. // }
  350. //
  351. // //rest-获取当前委托列表
  352. // #[tokio::test]
  353. // async fn rest_get_unfilled_orders_test() {
  354. // global::log_utils::init_log_with_trace();
  355. // let mut rest = get_rest();
  356. // let rep_data = rest.get_unfilled_orders("CELRUSDT".to_string(),
  357. // "".to_string(),
  358. // "".to_string(),
  359. // "".to_string(),
  360. // "".to_string(),
  361. // "".to_string(),
  362. // ).await;
  363. // trace!(?rep_data)
  364. // }
  365. //
  366. // //rest-获取历史委托列表
  367. // #[tokio::test]
  368. // async fn rest_get_history_orders_test() {
  369. // global::log_utils::init_log_with_trace();
  370. // let mut rest = get_rest();
  371. // let rep_data = rest.get_history_orders("CELRUSDT".to_string(),
  372. // "".to_string(),
  373. // "".to_string(),
  374. // "".to_string(),
  375. // "".to_string(),
  376. // "".to_string(),
  377. // ).await;
  378. // trace!(?rep_data)
  379. // }
  380. //
  381. // //rest-获取成交明细
  382. // #[tokio::test]
  383. // async fn rest_get_fills_test() {
  384. // global::log_utils::init_log_with_trace();
  385. // let mut rest = get_rest();
  386. // let rep_data = rest.get_fills("CELRUSDT".to_string(),
  387. // "1".to_string(),
  388. // "".to_string(),
  389. // "".to_string(),
  390. // "".to_string(),
  391. // "".to_string(),
  392. // ).await;
  393. // trace!(?rep_data)
  394. // }
  395. //
  396. // //rest-获取成交明细
  397. // #[tokio::test]
  398. // async fn rest_spot_place_plan_order_test() {
  399. // global::log_utils::init_log_with_trace();
  400. //
  401. // let mut rest = get_rest();
  402. // //限价-委托单
  403. // let params = serde_json::json!({
  404. // });
  405. //
  406. // let rep_data = rest.spot_place_plan_order(params).await;
  407. // trace!(?rep_data)
  408. // }
  409. //
  410. // //rest-修改计划委托
  411. // #[tokio::test]
  412. // async fn rest_update_place_plan_order_test() {
  413. // global::log_utils::init_log_with_trace();
  414. //
  415. // let mut rest = get_rest();
  416. // //限价-委托单
  417. // let params = serde_json::json!({
  418. // });
  419. //
  420. // let rep_data = rest.update_place_plan_order(params).await;
  421. // trace!(?rep_data)
  422. // }
  423. //
  424. // //rest-撤销计划委托
  425. // #[tokio::test]
  426. // async fn rest_cancel_plan_order_test() {
  427. // global::log_utils::init_log_with_trace();
  428. //
  429. // let mut rest = get_rest();
  430. // let rep_data = rest.cancel_plan_order("32131".to_string(), "3211".to_string()).await;
  431. // trace!(?rep_data)
  432. // }
  433. //
  434. // //rest-获取当前计划委托
  435. // #[tokio::test]
  436. // async fn rest_get_current_plan_order_test() {
  437. // global::log_utils::init_log_with_trace();
  438. //
  439. // let mut rest = get_rest();
  440. // let rep_data = rest.get_current_plan_order("CELRUSDT".to_string(),
  441. // "1".to_string(),
  442. // "".to_string(),
  443. // "".to_string(),
  444. // "".to_string(),
  445. // ).await;
  446. // trace!(?rep_data)
  447. // }
  448. //
  449. // //rest-获取历史计划委托
  450. // #[tokio::test]
  451. // async fn rest_get_history_plan_order_test() {
  452. // global::log_utils::init_log_with_trace();
  453. //
  454. // let mut rest = get_rest();
  455. // let rep_data = rest.get_history_plan_order("CELRUSDT".to_string(),
  456. // "1697701550192".to_string(),
  457. // "1697701580192".to_string(),
  458. // "100".to_string(),
  459. // ).await;
  460. // trace!(?rep_data)
  461. // }
  462. //
  463. // //rest-批量撤销计划委托
  464. // #[tokio::test]
  465. // async fn rest_cancel_plan_orders_test() {
  466. // global::log_utils::init_log_with_trace();
  467. //
  468. // let mut rest = get_rest();
  469. // let rep_data = rest.cancel_plan_orders(vec![]).await;
  470. // trace!(?rep_data)
  471. // }
  472. //
  473. // //rest-划转
  474. // #[tokio::test]
  475. // async fn rest_wallet_transfer_test() {
  476. // global::log_utils::init_log_with_trace();
  477. // let mut rest = get_rest();
  478. // let rep_data = rest.wallet_transfer("".to_string(),
  479. // "".to_string(),
  480. // "".to_string(),
  481. // "".to_string(),
  482. // "".to_string(),
  483. // "".to_string()).await;
  484. // trace!(?rep_data)
  485. // }
  486. //
  487. // //rest-获取账单流水
  488. // #[tokio::test]
  489. // async fn rest_get_account_bills_test() {
  490. // global::log_utils::init_log_with_trace();
  491. // let mut rest = get_rest();
  492. // let rep_data = rest.get_account_bills("".to_string(),
  493. // "".to_string(),
  494. // "".to_string(),
  495. // "".to_string(), ).await;
  496. // trace!(?rep_data)
  497. // }
  498. //
  499. //
  500. // async fn get_ws(btree_map: Option<BitgetSpotLogin>, type_v: BitgetSpotWsType) -> BitgetSpotWs {
  501. // let mut ku_ws = BitgetSpotWs::new(false, btree_map.clone(), type_v);
  502. // ku_ws
  503. // }
  504. //
  505. // fn get_rest() -> BitgetSpotRest {
  506. // let mut btree_map: BTreeMap<String, String> = BTreeMap::new();
  507. // // btree_map.insert("access_key".to_string(), ACCESS_KEY.to_string());
  508. // // btree_map.insert("secret_key".to_string(), SECRET_KEY.to_string());
  509. // btree_map.insert("access_key".to_string(), ACCESS_KEY.to_string());
  510. // btree_map.insert("secret_key".to_string(), SECRET_KEY.to_string());
  511. // btree_map.insert("pass_key".to_string(), PASS_KEY.to_string());
  512. //
  513. // let mut ku_exc = BitgetSpotRest::new(false, btree_map);
  514. // ku_exc
  515. // }