Rust的宏功能博大精深。在K线bar中,往往有很多不同分时k线图,比如1,2,3,5,。。。。60,120,250,300...。。不同分钟类型。
如果不用宏,那么手写会比较麻烦。下面就试用一下宏来实现不同类型的bar.
一、数据和功能
bar的结构
为了简单起见,我们把不同分时的Bar抽象成下面的结构。
bash
struct Bar{
open :f64,
close:f64,
high:f64,
low:f64,
}
同时这个结构要实现一个trait
bash
trait BarPrint{
fn print_self(&self);
}
二、单一bar的实现
我们先考虑,impl_single_bar接受单一的类型参数,比如Bar1,Bar3,Bar5,...
bash
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); //这个可以放在main()函数外,不影响
let bar = Bar1::new();
println!("bar:{:?}",bar);
bar.print_self();
impl_single_bar!(Bar2);
let bar2 = Bar2::new();
println!("bar:{:?}",bar2);
bar2.print_self();
}
输出:
bash
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
这样的确方便了一些,但是因为参数是一个个输入,需要
bash
impl_single_bar!(Bar1);
impl_single_bar!(Bar2);
每一个类型,写一行函数,还是不太方便。
注意:
1、impl_single_bar!(Bar1),可以放在main()函数外,不受影响;
2、$bar:ident,也可以是 $bar:tt。tt是分语树,比ident概念要大。
三、实现多类型参数输入
这里就需要用到rust宏的重复的写法。这里不特别展开,相关的资料很多。
1、试写一下生成多个类型的宏
bash
macro_rules! create_bars{
($($bar:ident),*) => {
$(
#[derive(Debug)]
struct $bar{
open:f64,
close:f64,
high:f64,
low:f64,
}
)*
}
}
2、上面也可以跳过,直接
bash
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);
// 测试生成多个struct Bar5,Bar6,Bar7,同时测试其实现的方法
impl_multi_bars!(Bar5,Bar6,Bar7);//可以放在main()函数外,在main()函数内,直接调用即可。
let bar5 = Bar5::new();
println!("bar5:{:?}",bar5);
bar5.print_self();
}
输出:
bash
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相比,你可需要把多个类型写到一行中就行了,即:
bash
impl_multi_bars!(Bar5,Bar6,Bar7);