如何在每一步将一些值转移到 Rust 生成器中？

Question

我将生成器用作长期异步线程（请参阅 How to implement a lightweight long-lived thread based on a generator or asynchronous function in Rust?) 在用户交互场景中。我需要在每一步将用户输入传递给生成器。我想我可以用RefCell 做到这一点，但不清楚如何在创建实例时将引用转移到生成器内部的RefCell？

fn user_scenario() -> impl Generator<Yield = String, Return = String> {
    || {
        yield format!("what is your name?");
        yield format!("{}, how are you feeling?", "anon");
        return format!("{}, bye !", "anon");
    }
}

UserData 结构包含用户输入，第二个结构包含由UserData 和生成器实例组成的用户会话。会话收集在HashMap。

struct UserData {
    sid: String,
    msg_in: String,
    msg_out: String,
}

struct UserSession {
    udata_cell: RefCell<UserData>,
    scenario: Pin<Box<dyn Generator<Yield = String, Return = String>>>,
}

type UserSessions = HashMap<String, UserSession>;

let mut sessions: UserSessions = HashMap::new();

UserData 是在接收用户输入时创建的——此时我需要在生成器中发送一个指向UserData 的链接，将其包装在RefCell 中，但我不知道该怎么做因为生成器的生命周期为 'static，而 RefCell 的生命周期较短！

let mut udata: UserData = read_udata(&mut stream);
let mut session: UserSession;

if udata.sid == "" { //new session
    let sid = rnd.gen::<u64>().to_string();
    udata.sid = sid.clone();
    sessions.insert(
        sid.clone(),
        UserSession {
            udata_cell: RefCell::new(udata),
            scenario: Box::pin(user_scenario())
        }
    );
    session = sessions.get_mut(&sid).unwrap();
} 

full code is here，但是这里的生成器看不到用户输入。

Answer 1

免责声明：恢复参数是生成器的计划扩展，因此在将来的某个时候可以使用&UserData 恢复参数。

---

目前，我建议共享所有权。成本相当小（一次内存分配，一次间接），并且会为您省去很多麻烦：

struct UserSession {
    user_data: Rc<RefCell<UserData>>,
    scenario: ..,
}

构建于：

let user_data = Rc::new(RefCell::new(udata));
UserSession {
    user_data: user_data.clone(),
    scenario: Box::pin(user_scenario(user_data))
}

然后，会话和生成器都可以轮流访问UserData，一切都很好。

有一个小问题：小心范围。如果您在yield 点上保持.borrow() 处于活动状态，这是可能的，那么当您尝试在生成器之外对其进行写入时，您将遇到运行时错误。

---

更复杂的解决方案是使用消息队列；这也将涉及内存分配等...我认为您的UserData 结构是一对队列的退化形式：它是两个具有一条消息容量的队列。您可以使用常规队列使其更明确，但这不会给您带来太多收益。