Можно ли разрешить макросу расширяться до поля структуры?
Я хотел бы сделать следующее, но макросы в этой позиции, похоже, не работают (я получаю error: expected `:`, found `!`. Как я могу сопоставить отдельные элементы структуры с шаблоном и прикрепить к ним атрибуты на основе совпадения?
use serde_derive::Serialize;
macro_rules! optional_param {
($name:ident : Option<$type:ty>) => { #[serde(skip_serializing_if = "Option::is_none")] pub $name: Option<$ty> };
($name:ident : Vec <$type:ty>) => { #[serde(skip_serializing_if = "Vec::is_empty" )] pub $name: Vec <$ty> };
($name:ident : bool ) => { #[serde(skip_serializing_if = "bool::not" )] pub $name: bool };
}
macro_rules! impl_extra {
( $name:ident { $( $param:ident : $type:ty ),* $(,)* } ) => (
#[derive(Default,Debug,Serialize)]
pub struct $name {
$( optional_param!($param : $type), )*
}
);
}
impl_extra!(MyStruct { member: Option<String> });
Ответы 1
В самом деле, вызовы макросов недопустимы в середине определения структуры. Однако там мы можем использовать метапеременные. Хитрость заключается в том, чтобы анализировать параметры постепенно по ходу строил токены для определений полей, а когда больше нет ввода для обработки, генерировать определение структуры с определениями полей, исходящими из метапеременной.
В качестве первого шага давайте посмотрим, как конкретно выглядит макрос, который не обрабатывает типы полей:
macro_rules! impl_extra {
( @ $name:ident { } -> ($($result:tt)*) ) => (
#[derive(Default, Debug, Serialize)]
pub struct $name {
$($result)*
}
);
( @ $name:ident { $param:ident : $type:ty, $($rest:tt)* } -> ($($result:tt)*) ) => (
impl_extra!(@ $name { $($rest)* } -> (
$($result)*
pub $param : $type,
));
);
( $name:ident { $( $param:ident : $type:ty ),* $(,)* } ) => (
impl_extra!(@ $name { $($param : $type,)* } -> ());
);
}
Единственное, что делает этот макрос, - это добавляет pub в каждое поле и определяет pub struct с атрибутом #[derive]. Первое правило обрабатывает конечный случай, то есть когда больше нет полей для обработки. Второе правило обрабатывает рекурсивный регистр, а третье правило обрабатывает «общедоступный» синтаксис макроса и преобразует его в синтаксис «обработки».
Обратите внимание, что я использую @ в качестве начального токена для внутренних правил, чтобы отличать их от «общедоступных» правил. Если этот макрос не предназначен для экспорта в другие ящики, вы также можете переместить внутренние правила в другой макрос. Если же макрос экспортируется, то, возможно, также потребуется экспортировать отдельный макрос для внутренних правил.
Теперь давайте разберемся с различными типами полей:
macro_rules! impl_extra {
( @ $name:ident { } -> ($($result:tt)*) ) => (
#[derive(Default, Debug, Serialize)]
pub struct $name {
$($result)*
}
);
( @ $name:ident { $param:ident : Option<$type:ty>, $($rest:tt)* } -> ($($result:tt)*) ) => (
impl_extra!(@ $name { $($rest)* } -> (
$($result)*
#[serde(skip_serializing_if = "Option::is_none")]
pub $param : Option<$type>,
));
);
( @ $name:ident { $param:ident : Vec<$type:ty>, $($rest:tt)* } -> ($($result:tt)*) ) => (
impl_extra!(@ $name { $($rest)* } -> (
$($result)*
#[serde(skip_serializing_if = "Vec::is_empty")]
pub $param : Vec<$type>,
));
);
( @ $name:ident { $param:ident : bool, $($rest:tt)* } -> ($($result:tt)*) ) => (
impl_extra!(@ $name { $($rest)* } -> (
$($result)*
#[serde(skip_serializing_if = "bool::not")]
pub $param : bool,
));
);
( $name:ident { $( $param:ident : $($type:tt)* ),* $(,)* } ) => (
impl_extra!(@ $name { $($param : $($type)*,)* } -> ());
);
}
Обратите внимание, что есть различие в последнем правиле: вместо сопоставления на ty мы теперь сопоставляем последовательность tt. Это потому, что после того, как макрос проанализировал ty, он не может быть разбит, поэтому, когда мы делаем рекурсивный вызов макроса, ty не может соответствовать чему-то вроде Option<$type:ty>.
Большое тебе спасибо! В частности, замечание
ty, вероятно, сэкономило мне время на отладку! Хотелось бы, чтобы был более элегантный и менее повторяющийся способ, но, очевидно, он будет идти только с макросами 2.0, которые afaik разрешат макросы в позиции мета-элемента.