as-rust/derive/src/export_excel.rs

use std::collections::BTreeMap;
use crate::binance_swap_export::BinanceSwapExport;
use crate::bybit_swap_export::BybitSwapExport;
// use crate::bitget_spot_export::BitgetSpotExport;
use crate::ExportConnector;
use crate::gate_swap_export::GateSwapExport;
// use crate::kucoin_spot_export::KucoinSpotExport;
// use crate::kucoin_swap_export::KucoinSwapExport;
// use crate::okx_swap_export::OkxSwapExport;

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum ExportEnum {
    BinanceSwap,
    // KucoinSwap,
    // KucoinSpot,
    GateSwap,
    BybitSwap,
    // BitgetSpot,
    // OkxSwap,
}

#[derive(Debug, Clone)]
pub struct ExportExcel;

impl ExportExcel {
    pub async fn new(export_enum: ExportEnum, is_colo: bool, params: BTreeMap<String, String>) -> Box<dyn ExportConnector> {
        match export_enum {
            ExportEnum::BinanceSwap => {
                Box::new(BinanceSwapExport::new(is_colo, params).await)
            }
            // ExportEnum::KucoinSwap => {
            //     Box::new(KucoinSwapExport::new(is_colo, params).await)
            // }
            // ExportEnum::KucoinSpot => {
            //     Box::new(KucoinSpotExport::new(is_colo, params).await)
            // }
            ExportEnum::GateSwap => {
                Box::new(GateSwapExport::new(is_colo, params).await)
            }
            ExportEnum::BybitSwap => {
                Box::new(BybitSwapExport::new(is_colo, params).await)
            }
            // ExportEnum::BitgetSpot => {
            //     Box::new(BitgetSpotExport::new(is_colo, params).await)
            // }
            // ExportEnum::OkxSwap => {
            //     Box::new(OkxSwapExport::new(is_colo, params).await)
            // }
        }
    }
}