as-rust/standard/src/exchange.rs

use std::collections::{BTreeMap};
use std::io::Error;
use tokio::sync::mpsc::Sender;
use crate::{Order, Platform};
use crate::binance_swap::BinanceSwap;
use crate::bitget_swap::BitgetSwap;
use crate::bybit_swap::BybitSwap;
use crate::coinex_swap::CoinexSwap;
use crate::gate_swap::GateSwap;

/// 交易所交易模式枚举
/// - `BinanceSwap`: Binance交易所期货;
/// - `BinanceSpot`: Binance交易所现货;
/// - `GateSwap`: Gate交易所期货;
/// - `GateSpot`: Gate交易所现货;
/// - `KucoinSwap`: kucoin交易所期货;
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum ExchangeEnum {
    BinanceSwap,
    CoinexSwap,
    // BinanceSpot,
    GateSwap,
    // GateSpot,
    // KucoinSwap,
    // KucoinSpot,
    // OkxSwap,
    // BitgetSpot,
    BitgetSwap,
    BybitSwap,
    // HtxSwap,
    // BingxSwap,
    // MexcSwap,
    // BitMartSwap,
    // CoinsphSwap,
    // PhemexSwap,
    // WooSwap,
    // CointrSwap
}

/// Exchange结构体
///
/// 方法:
/// - 创建Exchange:
///
/// new(platform: [ExchangeEnum], symbol: `String`, is_colo: `bool`, params: `BTreeMap<String, String>`, order_sender: `Sender<Order>`, error_sender: `Sender<Error>`) -> `Box<dyn Platform + Send + Sync>`
/// - `platform(`[ExchangeEnum]`)`: 交易所平台枚举
/// - `symbol(String)`: 币对
/// - `is_colo(bool)`: 是否开始告诉模式
/// - `params(BTreeMap<String, String>)`: 登录所需参数
/// - `order_sender(Sender<Order>)`: 订单消息发送者
/// - `error_sender(Sender<Error>)`: 错误消息发送者
///
/// 示例参数值:
///
/// | 交易所枚举 | params参数示例 BTreeMap<String, String> |
/// | --- | --- |
/// | BinanceSwap | {"access_key":"your_access_key","secret_key":"your_secret_key"} |
/// | BinanceSpot | {"access_key":"your_access_key","secret_key":"your_secret_key"} |
/// | GateSwap | {"access_key":"your_access_key","secret_key":"your_secret_key"} |
/// | GateSpot | {"access_key":"your_access_key","secret_key":"your_secret_key"} |
/// | KucoinSwap | {"access_key":"your_access_key","secret_key":"your_secret_key","pass_key":"your_pass_key"} |
///  ```rust
/// use std::collections::BTreeMap;
/// use std::io::Error;
/// use tokio::sync::mpsc;
/// use standard::exchange::{Exchange, ExchangeEnum};
/// use standard::Order;
///
/// let mut params:BTreeMap<String,String> = BTreeMap::new();
/// params.insert("access_key".to_string(), "your_access_key".to_string());
/// params.insert("secret_key".to_string(), "your_secret_key".to_string());
/// let (order_sender, _order_receiver): (mpsc::Sender<Order>, mpsc::Receiver<Order>) = mpsc::channel(1024);
/// let (error_sender, _error_receiver): (mpsc::Sender<Error>, mpsc::Receiver<Error>) = mpsc::channel(1024);
///
/// let exchange = Exchange::new(ExchangeEnum::GateSwap, "BTC_USDT".to_string(), false, params, order_sender, error_sender);
#[derive(Debug, Clone)]
pub struct Exchange;

impl Exchange {
    pub async fn new(exchange: ExchangeEnum, symbol: String, is_colo: bool, params: BTreeMap<String, String>, order_sender: Sender<Order>, error_sender: Sender<Error>) -> Box<dyn Platform + Send + Sync> {
        match exchange {
            ExchangeEnum::BinanceSwap => {
                Box::new(BinanceSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            }
            ExchangeEnum::GateSwap => {
                Box::new(GateSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            }
            // ExchangeEnum::GateSpot => {
            //     Box::new(GateSpot::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::KucoinSwap => {
            //     Box::new(KucoinSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::KucoinSpot =>{
            //     Box::new(KucoinSpot::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::OkxSwap => {
            //     Box::new(OkxSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::BitgetSpot => {
            //     Box::new(BitgetSpot::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            ExchangeEnum::BitgetSwap => {
                Box::new(BitgetSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            }
            ExchangeEnum::BybitSwap => {
                Box::new(BybitSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            }
            ExchangeEnum::CoinexSwap => {
                Box::new(CoinexSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            }
            // ExchangeEnum::HtxSwap => {
            //     Box::new(HtxSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::BingxSwap => {
            //     Box::new(BingxSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::MexcSwap => {
            //     Box::new(MexcSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::BitMartSwap => {
            //     Box::new(BitMartSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::CoinsphSwap => {
            //     Box::new(CoinsphSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::PhemexSwap => {
            //     Box::new(PhemexSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::WooSwap => {
            //     Box::new(WooSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
            // ExchangeEnum::CointrSwap => {
            //     Box::new(CointrSwap::new(symbol, is_colo, params, order_sender, error_sender).await)
            // }
        }
    }
}