如何在封装多次`write!`调用的宏中正确捕获writer？

咱们先来梳理下核心需求：你想要实现一个write_twice!宏，把内容写入目标writer两次，同时得满足三个关键要求：

1. 传入的writer表达式只求值一次（比如创建文件的代码不会执行两次导致重复截断）
2. 宏调用后原始writer还能继续使用，不能被移动走
3. 完全兼容write!的所有合法场景——只要write!(w, ...)能编译，换成write_twice!(w, ...)也得能跑

之前的几个版本要么重复求值writer、要么移动了原始writer、要么兼容性有问题，下面咱们来写一个能满足所有要求的宏实现，再一步步说清它的工作原理。

正确的宏实现 #

use std::fmt::Write;

macro_rules! write_twice {
    ($writer:expr, $($fmt:tt)*) => {{
        (|| {
            // 关键1：将writer捕获为可变引用，确保$writer只被求值一次
            let mut writer = &mut $writer;
            for _ in 0..2 {
                if let Err(err) = write!(writer, $($fmt)*) {
                    return Err(err);
                }
                // 关键2：每次write后重新借用writer，避免可变引用被"锁定"
                writer = &mut *writer;
            }
            Ok(())
        })()
    }};
}

为什么这个版本能满足所有要求？ #

咱们逐个对应你的三个条件拆解：

1. 表达式仅求值一次 #

$writer只会在let mut writer = &mut $writer;这一行被求值一次，后续循环里用的都是这个预存的引用。比如你写write_twice!(fs::File::create(path)?, "content")时，fs::File::create只会执行一次，不会第二次截断文件，完全符合预期。

2. 原始writer不会被移动，后续仍可使用 #

我们只是创建了原始$writer的可变引用，原始绑定（比如fmt方法里的f、main里的&mut buf）根本没被移动。而且每次循环后的writer = &mut *writer;会重新借用，不会让原始引用的生命周期被固定，所以宏调用结束后，你还能继续用原来的writer做其他操作。

比如这个场景现在完全正常：

use std::fmt;

struct Double(String);

impl fmt::Display for Double {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "first -> ")?;
        write_twice!(f, "{}", self.0)?;
        // 这里f还能正常调用write!，不会报"borrow of moved value"错误
        write!(f, " <- second")
    }
}

fn main() {
    let d = Double("test".to_string());
    println!("{}", d); // 输出：first -> testtest <- second
}

3. 完全兼容write!的所有场景 #

不管你传给write!的是可变引用（比如&mut String、&mut Formatter）还是拥有所有权的实例（比如File、Vec<u8>），这个宏都能正常工作：

• 如果传入的是拥有所有权的值：&mut $writer会创建它的可变引用，和write!的用法完全一致。
• 如果传入的已经是可变引用：&mut $writer会生成&mut &mut T，Rust会自动通过DerefMut解引用成&mut T，满足write!的要求。
• 每次循环后的重新借用操作，能避免可变引用的生命周期冲突，确保两次write!都能拿到有效的可变引用。

比如这个传入&mut String的场景：

fn main() {
    let mut buf = String::new();
    write_twice!(&mut buf, "{} {}", 1, 2).unwrap();
    // 宏调用后还能继续往buf里写内容
    write!(&mut buf, " and more").unwrap();
    println!("{}", buf); // 输出：1 21 2 and more
}

为什么之前的版本失败？ #

• 最初直接循环调用write!($writer, ...)：每次循环都会重新求值$writer，导致像File::create这样的操作执行两次，破坏了第一个要求。
• 直接let mut writer = $writer;：会移动原始writer，导致后续无法使用，违反了第二个要求。
• 只写let mut writer = &mut $writer;但没有重新借用：第一次write!会占用这个可变引用，第二次循环时无法再借用，导致编译错误；加了writer = &mut *writer;后，每次都能拿到新的可变引用，解决了这个问题。

备注：内容来源于stack exchange，提问作者BallpointBen