如何在Rust中写入长度不超过缓冲区大小的格式化字符串？

你遇到的问题是因为Rust标准库的Write trait默认在写入失败时会中止操作并返回错误，而ArrayString的Write实现一旦空间不足就会返回错误，导致write!宏在处理第二个参数时失败，最终只保留了第一个参数的内容。要实现像snprintf那样尽可能写入直到填满缓冲区的功能，我们可以通过自定义一个支持截断写入的Write包装器来解决。

解决方案：自定义截断式Writer #

我们可以创建一个包装ArrayString的结构体，并为它实现Write trait，在写入时自动截断内容以适应剩余空间，同时保证UTF-8的有效性：

use std::fmt::Write;
use arrayvec::ArrayString;

// 包装ArrayString，实现截断式写入
struct TruncatingWriter<'a, const N: usize>(&'a mut ArrayString<N>);

impl<'a, const N: usize> Write for TruncatingWriter<'a, N> {
    fn write_str(&mut self, s: &str) -> std::fmt::Result {
        let remaining_bytes = self.0.capacity() - self.0.len();
        if remaining_bytes == 0 {
            return Ok(());
        }

        // 计算可以安全截取的字节数，保证UTF-8有效性
        let mut take_bytes = 0;
        let mut chars = s.chars();
        while take_bytes < remaining_bytes {
            match chars.next() {
                Some(c) => {
                    let c_len = c.len_utf8();
                    if take_bytes + c_len > remaining_bytes {
                        break;
                    }
                    take_bytes += c_len;
                }
                None => break,
            }
        }

        // 截取有效子串并写入
        let truncated = &s[0..take_bytes];
        self.0.push_str(truncated)
            .map_err(|_| std::fmt::Error)?;

        Ok(())
    }
}

fn main() {
    const SIZE: usize = 16;
    let mut s = ArrayString::<SIZE>::new();
    let mut writer = TruncatingWriter(&mut s);

    // 使用自定义writer执行格式化写入，忽略返回值（因为我们已经处理了截断）
    let _ = write!(writer, "{}{}", "0123456789", "0123456789");

    println!("{}", s); // 输出：0123456789012345
}

为什么这个方案有效？ #

1. 截断逻辑：我们的write_str实现会先计算缓冲区剩余的字节空间，然后从输入字符串中截取尽可能多的有效UTF-8字符（不会截断到半个字符），确保写入的内容始终是合法的UTF-8。
2. 忽略写入终止错误：当缓冲区填满后，后续的写入请求会直接返回Ok(())，不会中断整个格式化流程，这样所有能写入的内容都会被保留。
3. 兼容write!宏：因为我们实现了标准的Write trait，所以可以直接使用write!宏来进行格式化，和原来的代码用法几乎一致。

对比C的snprintf #

• C的snprintf会为终止符\0预留一个字节，而Rust的ArrayString不需要终止符（它通过内部长度跟踪字符串边界），所以我们的实现会填满整个缓冲区的容量。
• 我们的实现天然支持UTF-8，不会出现截断导致的乱码问题，这比C的snprintf更安全（C中直接截断字节可能导致多字节字符损坏）。

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