Rust 宏 - 结构、枚举、转换

Question

有与（可能）rust 宏相关的问题，或者其他一些实现它的聪明方法。

1. 以下代码来自第三方，无法修改

   pub struct Message0;
   
   pub struct Message1;
   
   pub struct Message {
      pub payload: Option<Payload>,
   }
   
   pub enum Payload {
      PayloadMessage0(Message0),
      PayloadMessage1(Message1),
   }
   
   pub fn third_party_send(msg: Message) {
      // ...
   }
   

   2. 希望以某种方式实现此功能（或不需要为每个 Message0 和 Message1 类型实现单独功能的其他变体）

   pub fn send<T>(specific_msg: T) {
       third_party_send(Message {
           payload: Some(Payload::PayloadMessage???(specific_msg)),
       });
   }
   

   我是生锈的新手，仍然对宏和泛型感到困惑....

Answer 1

我不认为你可以绕过必须为每个变体创建一个函数，但是你 可以为你创建一个宏。 然后我将它们包装在特征中以将 Message? 转换为 Payload 并启用您的通用功能：

pub trait ToPayload {
    fn to_payload(self) -> Payload;
}
macro_rules! to_payload {
    ($message:ident, $payload:ident) => {
        impl ToPayload for $message {
            fn to_payload(self) -> Payload {
                Payload::$payload(self)
            }
        }
    }
}

to_payload!{ Message0, PayloadMessage0 }
to_payload!{ Message1, PayloadMessage1 }

pub fn send<T: ToPayload>(specific_msg: T) {
    third_party_send(Message {
        payload: Some(specific_msg.to_payload())
    });
}

fn main() {
    send(Message0);
    send(Message1);
}

Playground