快捷導(dǎo)航

Rust聲明宏在不同K線bar類(lèi)型中的應(yīng)用小結(jié)

更新時(shí)間：2024年05月11日 11:01:53 作者：songroom

在K線bar中,往往有很多不同分時(shí)k線圖,比如1,2,3,5,,,,,60,120,250,300…,,不同分鐘類(lèi)型,如果不用宏,那么手寫(xiě)會(huì)比較麻煩,下面就試用一下宏來(lái)實(shí)現(xiàn)不同類(lèi)型的bar,感興趣的朋友一起看看吧

Rust的宏功能博大精深。在K線bar中，往往有很多不同分時(shí)k線圖，比如1，2，3，5，。。。。60，120，250，300…。。不同分鐘類(lèi)型。
如果不用宏，那么手寫(xiě)會(huì)比較麻煩。下面就試用一下宏來(lái)實(shí)現(xiàn)不同類(lèi)型的bar.

一、數(shù)據(jù)和功能

bar的結(jié)構(gòu)

為了簡(jiǎn)單起見(jiàn)，我們把不同分時(shí)的Bar抽象成下面的結(jié)構(gòu)。

 struct Bar{
     open :f64,
     close:f64,
     high:f64,
     low:f64,
 }

同時(shí)這個(gè)結(jié)構(gòu)要實(shí)現(xiàn)一個(gè)trait

trait BarPrint{
    fn print_self(&self);
}

二、單一bar的實(shí)現(xiàn)

我們先考慮，impl_single_bar接受單一的類(lèi)型參數(shù)，比如Bar1,Bar3,Bar5,…

trait BarPrint{
    fn print_self(&self);
}
// Bar1,Bar2,Bar3,Bar5,Bar10,Bar15,Bar30,Bar45,Bar60,.....
macro_rules!  impl_single_bar {
    ($bar:ident) => (
        #[derive(Debug)]
        struct $bar{
            open:f64,
            close:f64,
            high:f64,
            low:f64,
        }
        impl $bar{
            fn new() -> Self{
                $bar{
                    open:0.0,
                    close:0.0,
                    high:0.0,
                    low:0.0,
               }
            }
        }
        impl BarPrint for $bar {
            fn print_self(&self){
                println!("impl_single_bar =>close:{:?} open: {:?}, high:{:?}, low:{:?}",&self.close,&self.open,&self.high,&self.low);
            }
        }
    );
}
fn main(){
    impl_single_bar!(Bar1); //這個(gè)可以放在main()函數(shù)外，不影響
    let  bar = Bar1::new();
    println!("bar:{:?}",bar);
    bar.print_self(); 
    impl_single_bar!(Bar2);
    let  bar2 = Bar2::new();
    println!("bar:{:?}",bar2);
    bar2.print_self(); 
}

輸出：

bar:Bar1 { open: 0.0, close: 0.0, high: 0.0, low: 0.0 }
impl_single_bar =>close:0.0 open: 0.0, high:0.0, low:0.0
bar:Bar2 { open: 0.0, close: 0.0, high: 0.0, low: 0.0 }
impl_single_bar =>close:0.0 open: 0.0, high:0.0, low:0.0

這樣的確方便了一些，但是因?yàn)閰?shù)是一個(gè)個(gè)輸入，需要

impl_single_bar!(Bar1);
impl_single_bar!(Bar2);

每一個(gè)類(lèi)型，寫(xiě)一行函數(shù)，還是不太方便。
注意：
1、impl_single_bar!(Bar1)，可以放在main()函數(shù)外，不受影響；
2、$bar:ident，也可以是 $bar:tt。tt是分語(yǔ)樹(shù)，比ident概念要大。

三、實(shí)現(xiàn)多類(lèi)型參數(shù)輸入

這里就需要用到rust宏的重復(fù)的寫(xiě)法。這里不特別展開(kāi)，相關(guān)的資料很多。

1、試寫(xiě)一下生成多個(gè)類(lèi)型的宏

macro_rules! create_bars{
    ($($bar:ident),*) => {
        $(
            #[derive(Debug)]
            struct $bar{
                open:f64,
                close:f64,
                high:f64,
                low:f64,
            }
        )*
    }
}

2、上面也可以跳過(guò)，直接

trait BarPrint{
    fn print_self(&self);
}
macro_rules! impl_multi_bars{
    ($($bar:ident),*) => {
        $(
            #[derive(Debug)]
            struct $bar{
                open:f64,
                close:f64,
                high:f64,
                low:f64,
            }
            impl $bar{
                fn new() -> Self{
                    $bar{
                        open:0.0,
                        close:0.0,
                        high:0.0,
                        low:0.0,
                    }
                }
            }
            impl BarPrint for $bar {
                fn print_self(&self){
                    println!("impl_multi_bars => close:{:?} open: {:?}, high:{:?}, low:{:?}",&self.close,&self.open,&self.high,&self.low);
                }
            }
        )*
    }      
}
fn main(){
    create_bars!(Bar3,Bar4);
    let bar3 =Bar3{open:0.0,close:0.0,high:0.0,low:0.0};
    println!("bar3:{:?}",bar3);
    let bar4 =Bar4{open:0.0,close:0.0,high:0.0,low:0.0};
    println!("bar4:{:?}",bar4);
    // 測(cè)試生成多個(gè)struct Bar5,Bar6,Bar7,同時(shí)測(cè)試其實(shí)現(xiàn)的方法
    impl_multi_bars!(Bar5,Bar6,Bar7);//可以放在main()函數(shù)外，在main()函數(shù)內(nèi)，直接調(diào)用即可。
    let  bar5 = Bar5::new();
    println!("bar5:{:?}",bar5);
    bar5.print_self();
}

輸出：

bar3:Bar3 { open: 0.0, close: 0.0, high: 0.0, low: 0.0 }
bar4:Bar4 { open: 0.0, close: 0.0, high: 0.0, low: 0.0 }
bar5:Bar5 { open: 0.0, close: 0.0, high: 0.0, low: 0.0 }
impl_multi_bars => close:0.0 open: 0.0, high:0.0, low:0.0

和2相比，你可需要把多個(gè)類(lèi)型寫(xiě)到一行中就行了，即：