如何创建具有 RAII 属性的订阅服务器对象

Question

我正在通过带有发布者/订阅者模型的链接与一些硬件交谈。C++，我订阅了 RAII 以记住始终取消订阅，但我似乎无法在 rust 中正确获得所有权/借用。

天真地，这就像我想做的事情。 send和receive可能需要&mut self，因此据我了解，Subscription需要对Transport的可变访问。

struct Transport;
impl Transport {
    pub fn send(&mut self, cmd: &str) { unimplemented!() }
    pub fn subscribe(&mut self, cmd: &str) -> Subscription {
        self.send("subscribe-with-params");
        Subscription { trans: &mut self }
    }
}
struct Subscription {
    trans: &mut Transport,
}
impl Drop for Subscription {
    fn drop(&mut self) {
        self.trans.send("unsubscribe-with params");
    }
}
impl Subscription {
    fn receive(&mut self) -> &[u8] { /*blocking wait for data*/ }
}
fn test(t: Transport) {
    // Need to subscribe before command, as command might generate status messages
    let mut status_sub = t.subscribe("status-message");
    {
        let mut short_lived_sub = t.subscribe("command_reply");
        t.send("command");
        short_lived_sub.receive(); // Wait for ack
    }
    loop {
        println!("{:?}", status_sub.receive());
        /*processing of status */
    }
}

这里至少有两个问题。一个是Subscription应该如何保留对它的"父"，Transport 的一些引用，另一个是fn test的问题，我不能为两个不同的订阅借用两次Transport。

我

有一种感觉，我在这里问错了问题，所以也许有一种好方法可以完全以不同的方式解决这个问题？

Answer 1

对于您的Subscription来说，持有对Transport的可变引用是有问题的，因为正如您所发现的，您一次只能持有一个，并且在此期间您将无法对传输执行任何其他操作。

相反，您可以使用Rc（用于共享所有权）和RefCell（用于内部可变性）：

use std::rc::Rc;
use std::cell::RefCell;
struct TransportInner;
pub struct Transport {
    inner: Rc<RefCell<TransportInner>>,
}
pub struct Subscription { 
    trans: Rc<RefCell<TransportInner>>
}
impl TransportInner {
   pub fn send(&mut self, cmd: &str) { }
}
impl Transport {
   pub fn send(&mut self, cmd: &str) { 
       self.inner.borrow_mut().send(cmd)
   }
   pub fn subscribe(&mut self, cmd: &str) -> Subscription {
      self.send("subscribe-with-params");
      Subscription { trans: Rc::clone(&self.inner) }
   }
}
impl Drop for Subscription {
   fn drop(&mut self) {
      self.trans.borrow_mut().send("unsubscribe-with params");
   }
}

您可以执行此操作，而无需将其拆分为内部和外部结构，但这也需要用户通过Rc访问Transport，这可能很笨拙。

如果需要它跨线程工作，则应改用Arc<Mutex>。