为何Polars的SerWriter trait的finish方法需要可变引用？

针对Polars SerWriter trait中finish方法要求可变引用但实际不修改数据的问题，除了克隆DataFrame外，有以下几种可行方案：

1. 用RefCell实现内部可变性适配 #

利用Rust的std::cell::RefCell提供内部可变性，将不可变的DataFrame包装后，安全地获取可变引用传递给finish方法——因为你确认方法不会修改数据，这种操作是安全的。

示例代码（结合serde序列化场景）：

use polars::prelude::*;
use serde::ser::{Serialize, Serializer};
use std::cell::RefCell;

struct MySerializer<W: Write> {
    writer: RefCell<W>,
}

impl<W: Write> MySerializer<W> {
    fn new(writer: W) -> Self {
        Self { writer: RefCell::new(writer) }
    }
}

impl Serialize for DataFrame {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        // 包装DataFrame为RefCell
        let df_ref = RefCell::new(self);
        // 获取可变引用传递给finish
        let mut writer = MySerializer::new(serializer.into_inner());
        writer.finish(&mut df_ref.borrow_mut()).map_err(serde::ser::Error::custom)?;
        Ok(serializer.serialize_unit()?)
    }
}

2. 定义适配Trait封装可变引用调用 #

自定义一个新Trait，将接受不可变引用的逻辑封装起来，避免重复编写RefCell相关代码：

use polars::prelude::*;
use std::cell::RefCell;

trait ImmutableSerWriter<W: Write>: SerWriter<W> {
    fn finish_immutable(&mut self, df: &DataFrame) -> Result<(), PolarsError> {
        let df_ref = RefCell::new(df);
        self.finish(&mut df_ref.borrow_mut())
    }
}

// 为所有实现SerWriter的类型自动实现ImmutableSerWriter
impl<W: Write, T: SerWriter<W>> ImmutableSerWriter<W> for T {}

// 使用时直接调用finish_immutable
let df = DataFrame::new(...).unwrap();
let mut csv_writer = CsvWriter::new(std::io::stdout());
csv_writer.finish_immutable(&df)?;

3. 谨慎使用unsafe转换（仅限确认无修改场景） #

如果能通过查看Polars源码确认finish方法绝对不会修改传入的DataFrame（仅读取数据），可以用unsafe将不可变引用强制转换为可变引用。这种方式要非常谨慎，一旦后续Polars修改finish的实现开始修改数据，会导致未定义行为。

示例代码：

use polars::prelude::*;

fn finish_with_immutable<W: Write, T: SerWriter<W>>(writer: &mut T, df: &DataFrame) -> Result<(), PolarsError> {
    // 安全前提：确认SerWriter的finish方法不会修改DataFrame
    let df_mut = unsafe { &mut *(df as *const DataFrame as *mut DataFrame) };
    writer.finish(df_mut)
}

// 使用
let df = DataFrame::new(...).unwrap();
let mut csv_writer = CsvWriter::new(std::io::stdout());
finish_with_immutable(&mut csv_writer, &df)?;

4. 长期方案：提交PR修改Polars的SerWriter trait #

如果finish方法确实不需要修改DataFrame，可以向Polars项目提交PR，将finish的参数从&mut DataFrame改为&DataFrame。这能从根源解决兼容性问题，但需要等待社区审核和合并。

内容的提问来源于stack exchange，提问作者Bartek