Rust : 声明宏在不同K线bar类型中的应用

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);
相关推荐
芊寻(嵌入式)12 分钟前
C转C++学习笔记--基础知识摘录总结
开发语言·c++·笔记·学习
AskHarries15 分钟前
Java字节码增强库ByteBuddy
java·后端
一颗松鼠21 分钟前
JavaScript 闭包是什么?简单到看完就理解!
开发语言·前端·javascript·ecmascript
有梦想的咸鱼_22 分钟前
go实现并发安全hashtable 拉链法
开发语言·golang·哈希算法
海阔天空_201328 分钟前
Python pyautogui库:自动化操作的强大工具
运维·开发语言·python·青少年编程·自动化
佳佳_29 分钟前
Spring Boot 应用启动时打印配置类信息
spring boot·后端
天下皆白_唯我独黑35 分钟前
php 使用qrcode制作二维码图片
开发语言·php
夜雨翦春韭39 分钟前
Java中的动态代理
java·开发语言·aop·动态代理
小远yyds41 分钟前
前端Web用户 token 持久化
开发语言·前端·javascript·vue.js
何曾参静谧1 小时前
「C/C++」C/C++ 之 变量作用域详解
c语言·开发语言·c++