Rust中的后台工作线程和同步

Marius 发表于 Dev

马吕斯

我正在尝试编写一个简单的库，该库具有一个后台工作线程，该线程在调用库函数时处理命令。

我通常在C语言中执行此操作的方式是拥有一个将被工作人员阻塞的全局信号量句柄。函数在发送命令后会发出信号，此时工作人员将取消阻塞等。还有其他方法，但这仅是示例。

关于如何使用Rust实现类似的功能，我有几个问题。

如何防止线程在创建线程的函数返回后关闭？例如，线程将在我调用时创建init()，但在init()返回时退出，如何防止这种情况？
如何在工作线程和函数调用之间具有全局同步方法？我一直在使用通道，但是如何rx从线程访问和tx从不同的函数访问多个？例如send_cmd_a()，send_cmd_b()到同一线程

我要完成的伪代码：

static (tx, rx) = mpsc::channel(); //how to do something like this?

fn init() {
    thread::spawn(|| {
        loop {
            let cmd = rx.recv().unwrap(); //blocks till there is data
                                          //process data....
            if cmd == "exit" {
                return;
            }
        }
    });
}

fn send_cmd_a() {
    //Do a bunch of other stuff...
    tx.send("cmd_a").unwrap();
}

fn exit() {
    tx.send("exit").unwrap();
}

我是否必须创建一个封装所有这些内容的大对象，从而拥有同步机制？（仍然无法回答问题1）

在Rust中做这样的事情的首选方式是什么？

马吕斯

我想我想出了一种无需使用全局变量即可在Rust中实现所需功能的方法。

struct Device {
    sender: Sender<u8>, //other stuff
}

trait New {
    fn new() -> Self;
}

trait SendCommand {
    fn send_command(&self, u8);
}

impl New for Device {
    fn new() -> Device {
        let (tx, rx) = channel();
        let device = Device { sender: tx };
        thread::spawn(move || {
            loop {
                let cmd = rx.recv().unwrap();
                println!("Command: {}", cmd); //process commands here
            }
        });
        return device;
    }
}

impl SendCommand for Device {
    fn send_command(&self, cmd: u8) {
        self.sender.send(cmd).unwrap();
    }
}

fn main() {
    let dev = Device::new(); //create the device
    for i in 0..10 {
        dev.send_command(i); //send commands
        sleep(Duration::from_millis(50));
    }
    loop {}
}

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。