rust多个属性宏叠加_rust-参数校验宏实现

0x00 关于参数校验

开题: 在Spring开发中, 我们习惯了, 在参数Bean中添加JSR标准的参数校验Annotation,

那么, 我们在rust接口开发时, 同样也会面临这样的开发需求.

如果没有相应的开发框架, 那么接口的参数校验将会变得异常痛苦.

经过一个星期左右的学习和实践,终于完成此篇成果.

0x01 使用体验

使用体验上, 与Java开发并无二致.

#[derive(BeanCheck)]

struct UserDO {

#[Min(30)]

pub min_age: u32,

#[Max(20)]

#[Range(1, 100)]

pub age: u16,

#[Length(1, 13)]

pub username: String,

#[Pattern(r"^\d{1,5}$")]

pub password: String,

#[Email]

pub email: String,

pub mobile: String,

}

fn main() {

let u = UserDO {

min_age: 325,

age: 20,

username: "gorey".to_string(),

password: "12345".to_string(),

email: "[email protected]".to_string(),

mobile: "13812341234".to_string()

};

match u.validate() {

Ok(_) => { println!("check pass "); },

Err(e) => { println!("{}", e); },

}

}

目前移植了数字,字符串参数的常用的校验方法.

使用时, 只需要指定过程宏的名称, 然后就可以在相应字段上面, 添加 校验属性即可.

0x02 原理解释

这里使用过程宏+属性宏, 来实现需求, 首先我们需要定义一个trait, 用于宏生成的方法:

trait定义: bean_check_lib :

pub trait BeanCheck {

fn validate(&self) -> Result;

}

在生在代码中, 只需要调用此方法就可以完成参数校验工作.

宏实现: bean_check:

#[proc_macro_derive(BeanCheck, attributes(NotEmpty, Length, Min, Max, Range, Email, Pattern))]

pub fn hello_macro_derive(input: TokenStream) -> TokenStream {

let ast = syn::parse(input).unwrap();

generate_validate(&ast)

}

留意此方法的宏调用:#[proc_macro_derive(BeanCheck, attributes(NotEmpty, Length, Min, Max, Range, Email, Pattern))]

这里定义了过程宏的名称, 以及其支持的属性.

在此方法中, 首先通过参数的TokenStream, 构建一个语法树对象ast , 然后再通过解析 ast,

提取相应的字段及属性信息, 最后完成目标校验方法代码的生成.

工程目录 :

src

0x03 开发过程宏应用, 需要知道的

其实, 前面的帖子也聊过rust宏的基本写法, 里面提到一些基础模式匹配来提取信息.

对于复杂代码的信息提取分类, 我们可以使用库 syn 和 quote 来帮助我们构造方法代码 .

[dependencies]

syn = "1.0"

quote = "1.0"

简单说明一下, syn的作用是通过解析源码(TokenStream),生成语法树对象 , 然后就可以通过面向对象的处理方式 , 进行解析, 使用十分方便.

quote库的作用是: 构建用于生成TokenStream代码的辅助工具.

二者结合起来, 可以高效的创造方法代码了.

3.1 syn 说明

从帮助理解代码的角度来说明一下syn相关知识

syn::DeriveInput

最大颗粒的语法树对象 , 我们进行解析操作的数据来源

syn::Data

语法树对象的 data 属性, 通过match机制, 可以匹配处理我们需要的数据类型.

syn::Data::Struct

本例中, 我们只处理 struct, 所以代码中, 也仅关注此类型.

syn::Fields::Named

对于 struct, 我们接下来就需要匹配struct的field信息. 那么问题来了, 有的struct可以不包含field.

只有定义了field的场景下, 才会匹配到此类型.

syn::Field

这个才是我们需要处理的对象 , 通过对上一个节点对象的field进行遍历.

那么, 这个Field包含哪些信息呢?

包含:

pub struct Field {

/// Attributes tagged on the field.

pub attrs: Vec,

/// Visibility of the field.

pub vis: Visibility,

/// Name of the field, if any.

///

/// Fields of tuple structs have no names.

pub ident: Option,

pub colon_token: Option,

/// Type of the field.

pub ty: Type,

}

其中, 对我们有用的信息为:

attrs : 属性列表, 也就是说, 一个字段, 可以支持多行(个)属性定义.

ident : field变量名

ty : field变量类型

通过上述三种信息, 我们就可以在代码中创建校验代码了.

举例: 这里描述的是对字符串(String)进行长度(Length)校验的例子.

如果校验失败, 方法会返回一个自定义异常(CheckError), 用于向业务处理层反馈失败原因.

validate_quote = quote! {

#validate_quote

if !( self.#ident.len() >= #min && self.#ident.len() <= #max ) {

return Err(bean_check_lib::CheckError::Simple(stringify!(check failed: #ident #at).to_string()));

}

}

0x04 小结

话说回来, 与java的自定义annotation相比, 复杂度也差不多.

然后比较苦闷的是, clion的宏, 本身就是预编译时使用的, 所以, 在开发的时候 , IDE支持性比较差,

很多时候 还是要靠脑补.

欢迎+星, 提问,pull request, 一起成长.