【社区投稿】给Rust的Struct自动实现trait

给Rust的Struct自动实现trait

我们通常使用

#[derive(Clone, Debug)]

这样的方式给struct自动实现相应的trait,从而让struct具备某些特性,但是如果我们想让编译器给struct自动实现自己定义的trait要怎么办?

首先我们需要有一个trait,假设如下面的定义:

pub trait Printable {pub fn print_me(&self);
}

我们定义这个trait给struct赋予一个行为是逐行打印struct的所有Field。当然如果是自己实现肯定是可以凭空乱写的,那么我们可以和Debug一样,在 derive 中让编译器自动添加默认的实现。

首先需要给crate添加一个子crate:

cargo new --lib printable

然后在当前crate的 Cargo.toml 中添加依赖

[workspace]
members = [".","printable"
]
[dependencies]
printable = { version = "*", path = "printable"}

在printable 的 Cargo.toml 里还需要添加依赖

[dependencies]
syn = { version = "1.0", features = ["full"] }
quote = "1.0"
proc-macro2 = "1.0.51"

我们需要这三个crate来简化代码生成的工作,这里proc-macro2提供了自动实现宏的功能,syn用来解析结构体,quote用来输出TokenStream。

在 printable 的lib.rs 文件中

#[proc_macro_derive(Printable)]
pub fn print_info_derive(input: TokenStream) -> TokenStream {
}

我们在函数 print_info_derive 中输出的TokenStream,就会在编译时动态注入到struct中,这里参数input就是struct本身的代码流。

我们通过解析input就可以分析出 Struct的名字,Field列表,所有Field的名字,类型…..

下面是简化后的代码:

#[proc_macro_derive(Printable)]
pub fn print_info_derive(input: TokenStream) -> TokenStream {let struct_name = to_snake_case(input.ident.to_string().as_str());let fields = match input.data.clone() {syn::Data::Struct(data) => data.fields,_ => panic!("Only structs are supported"),};let fields_name: Vec<Ident> = fields.iter().map(|field| {field.ident.as_ref().unwrap().clone()}).collect();
}

之后我们就需要构建输出的代码流,这里使用 quote! 这个宏来实现。

#[proc_macro_derive(Printable)]
pub fn print_info_derive(input: TokenStream) -> TokenStream {let struct_name = to_snake_case(input.ident.to_string().as_str());let fields = match input.data.clone() {syn::Data::Struct(data) => data.fields,_ => panic!("Only structs are supported"),};let fields_name: Vec<Ident> = fields.iter().map(|field| {field.ident.as_ref().unwrap().clone()}).collect();let output_token = quote! {impl Printable for #struct_name {pub fn print_me(&self) {//这里添加逐行打印Field的代码,因为quote里本来就是在输出代码流//所以不能直接访问fields_name,比如循环之类的,所以我们这里需要//把生成这部分代码提取到函数外}}}output_token.into()
}

为了简单演示我们就使用一个函数来实现:

fn gen_print(fileds: Vec<Ident>) -> TokenStream2 {let print_stmts =fields.iter().map(|field| {quote! {println!("field:{}", &self.#field);}});quote!{#(#print_stmts)*}
}

最后组装一下,lib.rs  的代码如下:

#[proc_macro_derive(Printable)]
pub fn print_info_derive(input: TokenStream) -> TokenStream {let struct_name = to_snake_case(input.ident.to_string().as_str());let fields = match input.data.clone() {syn::Data::Struct(data) => data.fields,_ => panic!("Only structs are supported"),};let fields_name: Vec<Ident> = fields.iter().map(|field| {field.ident.as_ref().unwrap().clone()}).collect();let print_code = gen_print(fields_name);let output_token = quote! {impl Printable for #struct_name {pub fn print_me(&self) {#output_token}}}output_token.into()
}

如此这般一通操作后,我们随便一定一个Struct:

#[derive(Debug, Printable)]
struct TestTb {id: String,name: String,ts: i32
}

就可以

TestTb{id: "123".to_string(), name: "alex", ts: 111}.print_me();

就可以逐行打印出所有的Field了。

那么灵活的使用这个玩法,我们可以根据Struct的Field,自动生成 insert, update, delete的SQL也是可以的。给每个Field自动生成getter,setter方法…… (这个Java味太浓了,だめ)

研究这个是为了给 sqlx 增加一个自动生成insert,update,delete方法的增强,因为不喜欢写超长的insert和update语句。

Amusez-vous tous!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/690617.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

蓝桥杯刷题--python-9(2023填空题2)

001串的熵 - 蓝桥云课 (lanqiao.cn) import mathn 23333333for i in range(1, n >> 1):j n - ia -(i / n) * (math.log2(i / n)) * i - (j / n) * (math.log2(j / n)) * ja round(a, 4)if a 11625907.5798:print(i)break0求和 - 蓝桥云课 (lanqiao.cn) n20230408 pr…

opencv进行人脸识别

目录 一:准备工作 二:人脸图片识别 三:视频人脸识别 一:准备工作 1:github网站下载开源人脸特征数据 haarcascade_frontalface_default.xml 下载链接:

如何确定分库还是 分表?

分库分表 分库分表使用的场景不一样&#xff1a; 分表因为数据量比较大&#xff0c;导致事务执行缓慢&#xff1b;分库是因为单库的性能无法满足要求。 分片策略 1、垂直拆分 水平拆分 3 范围分片&#xff08;range&#xff09; 垂直水平拆分 4 如何解决数据查询问题&a…

【Jvm】性能调优(拓展)Jprofiler如何监控和解决死锁、内存泄露问题

文章目录 Jprofiler简介1.安装及IDEA集成Jprofiler2.如何监控并解决死锁3.如何监控及解决内存泄露(重点)4.总结5.后话 Jprofiler简介 Jprofilers是针对Java开发的性能分析工具(免费试用10天), 可以对Java程序的内存,CPU,线程,GC,锁等进行监控和分析, 1.安装及IDEA集成Jprofil…

车载软件架构 —— Adaptive AUTOSAR软件架构中时间同步、网络管理和软件更新策略

车载软件架构 —— Adaptive AUTOSAR软件架构中时间同步、网络管理和软件更新策略 我是穿拖鞋的汉子&#xff0c;魔都中坚持长期主义的汽车电子工程师&#xff08;Wechat&#xff1a;gongkenan2013&#xff09;。 老规矩&#xff0c;分享一段喜欢的文字&#xff0c;避免自己成…

解决用IPV6+DDNS访问UNRAID webui周期性失效的问题,smb不能访问的问题

我使用的unraid系统使用ddns&#xff08;DDNSGO&#xff09;绑定域名&#xff08;阿里域名&#xff09;与主机的ipv6地址进行远程访问&#xff0c;unraid是6.12.8。 遇到的问题是&#xff0c;配置当时是没问题的&#xff0c;但是过几天就会失效&#xff0c;无法通过域名访问we…

linux查看磁盘占用命令

查看系统整体的磁盘占用情况 df -h查看当前文件夹下所有子文件夹的磁盘占用情况 du -ah --max-depth1按照文件大小从大到小排序 du -ah --max-depth1 .|sort -hr查看当前文件夹以及所有子文件夹的磁盘占用汇总 du -sh *按照文件大小从大到小排序 du -sh *|sort -nr

MongoDB聚合运算符:$arrayElemAt

MongoDB聚合运算符$arrayElemAt用于返回数组中指定位置的元素。 语法 { $arrayElemAt: [ <array>, <idx> ] }<array>可以是任何能被解析为数组的表达式。<idx>可以是任何可以被解析为整数的表达式。 使用 如果<idx>为0或正整数&#xff0c;则…

【SQL注入】基于extractvalue函数的报错注入原理

一、ExtractValue函数 ExtractValue()函数是MySQL数据库中用于提取XML数据中指定路径的值的函数。它基于XPath语法&#xff0c;允许用户根据特定的路径表达式从XML格式的数据中获取所需信息。以下是关于MySQL中ExtractValue()函数的详细讲解&#xff1a; 1.1语法 EXTRACTVAL…

【国产MCU】-CH32V307-通用定时器(GPTM)-编码模式与旋转编码器驱动

通用定时器(GPTM)-编码模式与旋转编码器驱动 文章目录 通用定时器(GPTM)-编码模式与旋转编码器驱动1、通用定时器编码模式介绍2、旋转编码器介绍3、驱动API介绍4、编码模式使用示例本文将详细介绍如何使用CH32V307通用定时器的编码模式。 1、通用定时器编码模式介绍 编码器…

Maven高级(一)

文章目录 Maven高级&#xff08;一&#xff09;1. 分模块设计与开发1.1 介绍1.2 实践1.2.1 分析1.2.2 实现 1.3 总结 2. 继承与聚合2.1 继承2.1.1 继承关系2.1.1.1 思路分析2.1.1.2 实现 2.1.2 版本锁定2.1.2.1 场景2.1.2.2 介绍2.1.2.3 实现2.1.2.4 属性配置 2.2 聚合2.2.1 介…

扫码即可快速协作:草料二维码底部协作面板功能详解

功能介绍 在二维码上添加 底部协作面板 功能后 &#xff0c;扫码后不仅可以阅读设备信息、产品资料等基本信息&#xff0c;还可以在二维码底部输入内容评论并他人快速协作&#xff0c;支持添加图文、语言、手写签名等操作。 底部协作面板是提供给组织内部成员快速协作的功能&…

《汇编语言》- 读书笔记 - 第10章-CALL 和 RET 指令

《汇编语言》- 读书笔记 - 第10章-CALL 和 RET 指令 10.1 ret 和 retf检测点 10.1 10.2 call 指令10.3 依据位移进行转移的 call 指令检测点 10.2 10.4 转移的目的地址在指令中的 call 指令检测点 10.3 10.5 转移地址在寄存器中的 call 指令10.6 转移地址在内存中的 call 指令检…

数据模型概念

一、概念 (1) 定义 在数据库系统中针对不同的使用对象和应用目的&#xff0c;采用不同的数据模型。根据模型的应用的不同目的&#xff0c;可以将这些模型划分为两类&#xff1a; (2) 分类 A&#xff1a;概念数据模型 它也称信息模型它是按用户的观点&#xff08;观念世界&…

Android 基础技术——Synchonized 关键字

笔者希望做一个系列&#xff0c;整理 Android 基础技术&#xff0c;本章是关于Synchonized 关键字 Synchronized 关键字的作用是什么&#xff1f; 原子性&#xff1a;确保线程互斥访问同步代码 可见性&#xff1a;保证共享变量的修改能够及时可见&#xff0c;就是通过 Java 内存…

JVM(2)实战篇

1 内存调优 1.1 内存溢出和内存泄漏 内存泄漏&#xff08;memory leak&#xff09;&#xff1a;在Java中如果不再使用一个对象&#xff0c;但是该对象依然在GC ROOT的引用链上&#xff0c;这个对象就不会被垃圾回收器回收&#xff0c;这种情况就称之为内存泄漏。 内存泄漏绝…

5G LAN

5G LAN定义 5G LAN 是企业专用蜂窝网络解决方案。5G LAN 集成到组织现有的 IT 基础设施中以提供 5G 的所有优势&#xff0c;以确定的性能和延迟为整个企业的关键任务数字计划创建高速可预测的无线连接。根据定义&#xff0c;5G LAN 消除了在企业自己的环境中采用私有 5G 无线技…

单调队列和优先队列

本篇记录下一下关于单调队列和优先队列&#xff08;堆&#xff09;的方法以及解题思路. 文章目录 一. 单调队列1. 绝对差不超过限制得最长连续子数组2. 跳跃游戏 VI3. 设计自助结算系统4. 和至少为k的最短子数组5. 满足不等式的最大值 二. 优先队列1. 最后一块石头的重量2. 数据…

SQL实现模糊查询的四种方法总结

目录 一、一般模糊查询 二、利用通配符查询 1. _ 表示任意的单个字符 2. % 表示匹配任意多个任意字符 3. [ ]表示筛选范围 4. 查询包含通配符的字符串 一、一般模糊查询 1. 单条件查询 //查询所有姓名包含“张”的记录select * from student where name like 张 2. 多条…

十二:枚举与注解

文章目录 01、枚举类的使用1.1、枚举类的理解1.2、自定义枚举类1.3、使用enum关键字定义枚举类1.4、Enum类中的常用方法1.5、使用enum关键字定义的枚举类实现接口 02、注解的使用2.1、注解的理解2.3、如何自定义注解2.4、jdk中4个基本的元注解的使用12.5、jdk中4个基本的元注解…