我正在实现一个数据压缩接口:
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
。)
实现这一目标的唯一方法是使用
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
- 我的代码没有。
也可以看看: