如何在多个“编码器”之间共享可变写入器?

问题描述 投票:0回答:1

我正在实现一个数据压缩接口:

pub trait NumericEncoder<V> {
    fn encode(&mut self, value: V) -> io::Result<()>;
}

编码器可以在某种输出中编码某个数字,其中输出可以是流(文件),字节缓冲器或甚至是另一个编码器。有人可能会调用这样的实现:

let f = File::create("out").unwrap();
// Delta encoder whose data is run-length-compressed
let mut enc = DeltaEncoder::new(RunLengthEncoder::new(f));
enc.encode(123).unwrap();

这一切都很好,但在某些情况下,我需要针对相同输出流的多个编码器。像(简化)的东西:

let f = File::create("out")?;
let mut idEnc = RunLengthEncoder::new(DeltaEncoder::new(f));
let mut dataEnc = LZEncoder::new(f);
for (id, data) in input.iter() {
    idEnc.encode(id);
    dataEnc.encode(data);
}

这里,两个编码器在写入数据时会交错数据。

这需要对同一文件的可变访问,这对于直接的&mut引用是不可能的。据我所知,实现这一目标的唯一方法是使用RefCell;有没有更好的办法?

据我所知,这将使所有编码器实现不那么干净。现在可以像这样声明编码器:

pub struct MySpecialEncoder<'a, V, W>
where
    W: io::Write,
{
    w: &'a mut W,
    phantom: std::marker::PhantomData<V>,
}

使用RefCell,每个编码器结构和构造函数都需要处理Rc<RefCell<W>>,这不是很好并且将编写器的共享泄漏到编码器中,编码器不应该知道编写器是共享的。

(我确实考虑过是否可以改变NumericEncoder特性来采取作家论证,这必须是std::io::Write。这不会起作用,因为有些编码器不会写入std::io::Write,而是写入另一个NumericEncoder。)

rust mutability interior-mutability
1个回答
1
投票

实现这一目标的唯一方法是使用RefCell

任何赋予内部可变性的类型都可以。例如,Mutex也足够了。

这将使所有编码器实现不那么干净

我不知道你为什么这么认为。创建一个使用内部可变性的类型,并在需要额外功能时仅使用该类型:

#[derive(Debug)]
struct Funnel<E>(Rc<RefCell<E>>);

impl<E> Funnel<E> {
    fn new(e: E) -> Self {
        Funnel(Rc::new(RefCell::new(e)))
    }
}

impl<E> Clone for Funnel<E> {
    fn clone(&self) -> Self {
        Funnel(self.0.clone())
    }
}

impl<V, E> NumericEncoder<V> for Funnel<E>
where
    E: NumericEncoder<V>,
{
    fn encode(&mut self, value: V) -> io::Result<()> {
        self.0.borrow_mut().encode(value)
    }
}
fn main() -> io::Result<()> {
    let s = Shared;

    let s1 = Funnel::new(s);
    let s2 = s1.clone();

    let mut e1 = Wrapper(s1);
    let mut e2 = Wrapper(s2);

    e1.encode(1)?;
    e2.encode(2)?;

    Ok(())
}

您还应该考虑按值使用W,我不确定为什么需要PhantomData - 我的代码没有。

也可以看看:

© www.soinside.com 2019 - 2024. All rights reserved.