gpt4 book ai didi

rust - 如何使用macro_rules 定义带有可选#[cfg] 的结构?

转载 作者:行者123 更新时间:2023-12-03 11:28:24 24 4
gpt4 key购买 nike

我正在编写一个 Rust 程序,它从 C DLL 动态加载一些函数。我编写了一个辅助结构来加载它们:

// some function pointer definitions
// type func1_ptr = unsafe extern "C" func1() -> ();
// .....

pub struct FuncWrapper {
pub func1: func1_type,
}

impl FuncWrapper {
pub fn new() -> Self {
Self {
func1: unsafe { mem::transmute(load_function(lib, "func1")) },
}
}
}
由于要加载 100 多个函数,我编写了一个宏来完成这些脏活:
use paste::paste;

macro_rules! define_load_functions {
($v:vis $name:ident { $($func_name:ident),* $(,)?}) => {
paste! {
$v struct $name {
$(
pub $func_name: [<$func_name _ptr>],
)*
}

impl $name {
pub fn new() -> Self {
Self {
$(
$func_name: unsafe {
mem::transmute(load_function(ptr, stringify!($func_name)))
},
)*
}
}
}
}
}
}

macro_rules! define_load_functions {
pub FuncWrapper {
func1,
}
}
但是,有些函数是平台相关的,我想向它们添加 #[cfg(target_os)]。如何更改此宏以匹配可选的 #[cfg]?我想让它适用于以下示例代码:
macro_rules! define_load_functions {
pub FuncWrapper {
func1,
func2,
#[cfg(target_os = "windows")] func3,
#[cfg(target_os = "linux")] func4,
}
}
谢谢你。

最佳答案

以下会起作用吗?

macro_rules! define_load_functions {
(
$struct_vis: vis $struct_name: ident {
$( $(#[cfg($os_attr: meta)])? $fn_name: ident $(,)? )*
}
) => {
$struct_vis struct $struct_name {
$(
$(#[cfg($os_attr)])?
$fn_name: String,
)*
}

impl $struct_name {
pub fn new() -> Self {
Self {
$(
$(#[cfg($os_attr)])?
$fn_name: "Example".to_owned(),
)*
}
}
}
};
}

define_load_functions!(pub Example {
func1,
#[cfg(target_os = "linux")] func2,
#[cfg(target_os = "windows")] func3,
#[cfg(target_os = "macos")] func4,
#[cfg(target_os = "linux")] func5,
func6,
});
在基于 Linux 的系统上扩展上述宏给出以下内容:
pub struct Example {
func1: String,
#[cfg(target_os = "linux")]
func2: String,
#[cfg(target_os = "linux")]
func5: String,
func6: String,
}

impl Example {
pub fn new() -> Self {
Self {
func1: "Example".to_owned(),
#[cfg(target_os = "linux")]
func2: "Example".to_owned(),
#[cfg(target_os = "linux")]
func5: "Example".to_owned(),
func6: "Example".to_owned(),
}
}
}

关于rust - 如何使用macro_rules 定义带有可选#[cfg] 的结构?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66291962/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com