熊熊出没 发表于 2024-10-18 10:27:51

Rust宏之derive的设计及实战

Rust宏可以极大的简化编写的难度,学习好宏可以更好的减少冗余代码。
宏的基本概念

Rust中的宏可以分为两大类:声明宏(Declarative Macros)和过程宏(Procedural Macros)。

[*]声明宏:也称为macro_rules!宏,使用macro_rules!关键字界说。它是一种基于模式匹配的文本替换宏,雷同于C语言中的宏界说。声明宏在编译期展开,用匹配的代码片段替换宏调用处的代码。
[*]过程宏:是一种更为高级的宏,它通过编写Rust代码来处理输入的代码,并在编译期间生成新的代码。过程宏重要用于属性宏(Attribute Macros)、类函数宏(Function-Like Macros)和派生宏(Derive Macros)等场景。
宏的现实应用


[*]声明宏在Rust中的应用,我们最常接触的宏界说vec!大概println!都是尺度库里提供的,他可以在编译阶段就进行宏展开,在一定水平上牺牲编译速率有错误实时发现从而保证步伐运行稳定。
[*]过程宏在Rust中也是极为常见,就好比某个类,我们必要clone方法,但是声明的类并不支持clone,那么我们就可以在此类声明derive(Clone)如果必要默认的构造方法,那么同样可以声明derive(Default)
#
struct HcluaMacro {
    field: u32,
}此时我们就可以使用:
let obj = HcluaMacro::default();
let obj_clone = obj.clone();雷同的还要在序列化的宏等。
过程宏的实战

目录为Rust中的lua库hclua做对象的绑定,可以快速的实现Rust对象在Lua中的快速使用绑定。
新建库

由于过程宏只能在单独的库中使用,所以此时我们必要新建单独的一个项目cargo new hclua-macro,并在新项目的Cargo.toml中添加

proc-macro = true声明该项目为过程宏项目。
界说宏ObjectMacro

起首我们得界说ObjectMacro宏,那么我们必要声明:
#
pub fn object_macro_derive(input: TokenStream) -> TokenStream {
    TokenStream::new()
}此处我们就可以在这基础上实现额外的代码,他将在声明该宏文件中自动添加代码。
我们做以下测试:
#
pub fn object_macro_derive(input: TokenStream) -> TokenStream {
    quote! {
      fn this_is_macro_auto() {
            println!("this_is_macro_auto auto func");
      }
    }.into()
}其中quote!可以快速的生成代码块。
展开宏cargo-expand

接下我们必要宏在这个过期中帮我们生成了什么,我们借助以下工具cargo-expand,通过cargo install cargo-expand进行安装。
此时用cargo expand可以发现宏展开后的代码如下:
#!
#
use std::prelude::rust_2021::*;
#
extern crate std;
use hclua_macro::ObjectMacro;
struct HcluaMacro {
    field: u32,
}
fn this_is_macro_auto() {
    {
      ::std::io::_print(format_args!("this_is_macro_auto auto func\n"));
    };
}
fn main() {
    this_is_macro_auto();
}此时我们并没有处理跟类相关的任何东西,我们可以用parse_macro_input!将输入转成ItemStruct大概DeriveInput
#
pub fn object_macro_derive(input: TokenStream) -> TokenStream {
    let ItemStruct {
      ident,
      fields,
      attrs,
      ..
    } = parse_macro_input!(input);

    let name = ident.to_string();
    quote! {
      fn this_is_macro_auto() {
            println!("struct name {}", #name);
      }
    }.into()
}在quote中可以用#来序列化局数的变量数据。那么此时我们运行步伐,将会输出:
struct name HcluaMacro类名正确的被打印出来。
字段处理

界说

pub struct Field {
    pub attrs: Vec<Attribute>,

    pub vis: Visibility,

    pub mutability: FieldMutability,

    pub ident: Option<Ident>,

    pub colon_token: Option<Token![:]>,

    pub ty: Type,
}
[*]vis表现是否公开,就是表现pub大概pub(super) or pub(crate) or pub(in some::module)大概不公开模式
[*]attrs表现在该字段上的各种属性
[*]mutability表现是否可编辑
[*]ident变量的名字,当enum时只有类型没著名字
我们就可以通过处理变量的各种情况然后进行操纵,好比添加get_#ident大概set_#ident等方法。
属性处理

在此处我们界说了两种属性名称,hclua_field及 hclua_cfg,一种设置名称,一种设置是否可以在Lua中直接访问的字段名称,此时的宏界说:
#
pub fn object_macro_derive(input: TokenStream) -> TokenStream {
   
}如果没有在此处界说的attrib,在类型里直接添加会报编译错误。
此处我们判断是否为hclua_field字段进行相应的加工。
let functions: Vec<_> = fields
    .iter()
    .map(|field| {
      let field_ident = field.ident.clone().unwrap();
      if field.attrs.iter().any(|attr| attr.path().is_ident("hclua_field")) {
            quote! {}
      } else {
            quote! {}
      }
    })
    .collect();接下来将自动实现get及set方法。此处functions为TokenStream的数组,我们将用
#(#functions)*将此部门内容做展开。
完整宏代码:
use proc_macro::TokenStream;
use quote::{format_ident, quote};
use syn::{self, ItemStruct};

use syn::parse_macro_input;

#
pub fn object_macro_derive(input: TokenStream) -> TokenStream {
    let ItemStruct {
      ident,
      fields,
      attrs,
      ..
    } = parse_macro_input!(input);

    let functions: Vec<_> = fields
      .iter()
      .map(|field| {
            let field_ident = field.ident.clone().unwrap();
            if field.attrs.iter().any(|attr| attr.path().is_ident("hclua_field")) {
                let get_name = format_ident!("get_{}", field_ident);
                let set_name = format_ident!("set_{}", field_ident);
                let ty = field.ty.clone();
                quote! {
                  fn #get_name(&mut self) -> &#ty {
                        &self.#field_ident
                  }

                  fn #set_name(&mut self, val: #ty) {
                        self.#field_ident = val;
                  }
                }
            } else {
                quote! {}
            }
      })
      .collect();

    let name = ident.to_string();
    quote! {
      fn this_is_macro_auto() {
            println!("struct name {}", #name);
      }

      impl #ident {
            #(#functions)*
      }
    }.into()
}将示例代码进行如下书写:
use hclua_macro::ObjectMacro;

#
struct HcluaMacro {
    #
    field: u32,

    not_field: u32,
}

fn main() {
    this_is_macro_auto();
}通过cargo expand将得到如下的代码:
#!
#
use std::prelude::rust_2021::*;
#
extern crate std;
use hclua_macro::ObjectMacro;
struct HcluaMacro {
    #
    field: u32,
    not_field: u32,
}
fn this_is_macro_auto() {
    {
      ::std::io::_print(format_args!("struct name {0}\n", "HcluaMacro"));
    };
}
impl HcluaMacro {
    fn get_field(&mut self) -> &u32 {
      &self.field
    }
    fn set_field(&mut self, val: u32) {
      self.field = val;
    }
}
fn main() {
    this_is_macro_auto();
}自动实现了get及set方法,符合我们的要求。
注意事项


[*]学习曲线:难度相对较高,必要理解block,expr, ident, item, literal, pat, path, stmt, tt, ty, vis等相关内容。
[*]调试难度:由于宏是在编译时实行的,因此调试起来可能比力困难。对于严重依赖调试会相对吃力。
[*]滥用风险:虽然宏提供了强大的代码生成能力,但滥用宏也可能导致代码难以理解和维护。因此,在使用宏时尽量的做好规划及说明。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: Rust宏之derive的设计及实战