AskOverflow.Dev

AskOverflow.Dev Logo AskOverflow.Dev Logo

AskOverflow.Dev Navigation

  • 主页
  • 系统&网络
  • Ubuntu
  • Unix
  • DBA
  • Computer
  • Coding
  • LangChain

Mobile menu

Close
  • 主页
  • 系统&网络
    • 最新
    • 热门
    • 标签
  • Ubuntu
    • 最新
    • 热门
    • 标签
  • Unix
    • 最新
    • 标签
  • DBA
    • 最新
    • 标签
  • Computer
    • 最新
    • 标签
  • Coding
    • 最新
    • 标签
主页 / coding / 问题 / 79153514
Accepted
BallpointBen
BallpointBen
Asked: 2024-11-04 03:50:57 +0800 CST2024-11-04 03:50:57 +0800 CST 2024-11-04 03:50:57 +0800 CST

如何在包装对“write!”的多个调用的宏中正确捕获 writer!

  • 772

我有一个宏,它包装了对的多个调用write!。举个例子,假设我们有这个宏:

use std::fmt::Write;

macro_rules! write_twice {
    ($writer:expr, $($fmt:tt)*) => {{
        (|| {
            for _ in 0..2 {
                if let err @ ::core::result::Result::Err(_) = write!($writer, $($fmt)*) {
                    return err;
                }
            }
            Ok(())
        })()
    }};
}

fn main() {
    let v = vec![1, 2];
    let mut buf = String::new();
    write_twice!(&mut buf, "{v:?}").unwrap();
    println!("{buf:?}");  // [1, 2][1, 2]
}

现在,我想正确捕捉作者,但我正在努力让它在所有情况下都发挥作用。即:

  1. 如果传递了一个表达式,则应该只对其进行一次求值,这样就可以正常write_twice(fs::File::create(path)?, args...)工作而无需第二次写入重新截断文件。
  2. 宏调用之后,编写器应该仍然可用,即$writer不应被移动。
  3. 如果 的实例可以write!(w, args...)编译,那么在替换 时它也应该可以编译write_twice!(w, args...)。

write_twice!现在,上述失败 #1的版本write!($writer, $($fmt)*)将被评估两次。我们可以用以下代码替换它:

macro_rules! write_twice {
    ($writer:expr, $($fmt:tt)*) => {{
        (|| {
            let mut writer = $writer;  // evaluate only once
            for _ in 0..2 {
                if let err @ ::core::result::Result::Err(_) = write!(writer, $($fmt)*) {
                    return err;
                }
            }
            Ok(())
        })()
    }};
}

但这违反了 #2 因为 a&mut T不是Copy:

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)?;
        write!(f, " <- second")  // borrow of moved value: `f`
    }
}
error[E0382]: borrow of moved value: `f`
  --> src/main.rs:23:10
   |
5  |         (|| {
   |          -- value moved into closure here
...
20 |     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
   |                   - move occurs because `f` has type `&mut Formatter<'_>`, which does not implement the `Copy` trait
21 |         write!(f, "first -> ")?;
22 |         write_twice!(f, "{}", self.0)?;
   |                      - variable moved due to use in closure
23 |         write!(f, " <- second")
   |                ^ value borrowed here after move

因此我尝试用 替换该行let writer = &mut $writer;以避免移动&mut T,但现在我们收到一个新的编译器错误,违反了 #3(尽管编译器的建议确实解决了该问题):

error[E0596]: cannot borrow `f` as mutable, as it is not declared as mutable
  --> src/main.rs:6:17
   |
6  |             let writer = &mut $writer;
   |                          ^^^^^^^^^^^^ cannot borrow as mutable
...
22 |         write_twice!(f, "{}", self.0)?;
   |         ----------------------------- in this macro invocation
   |
   = note: this error originates in the macro `write_twice` (in Nightly builds, run with -Z macro-backtrace for more info)
help: consider changing this to be mutable
   |
20 |     fn fmt(&self, mut f: &mut fmt::Formatter<'_>) -> fmt::Result {
   |                   +++

如果我们尝试let mut writer = &mut *$writer,我们将无法再调用,write_twice!(string, ...)因为我们现在尝试写入 a&mut str而不是 a &mut String。

那么,在多次写入的宏中捕获写入器的正确方法是什么?

rust
  • 1 1 个回答
  • 53 Views

1 个回答

  • Voted
  1. Best Answer
    Francis Gagné
    2024-11-04T08:18:08+08:002024-11-04T08:18:08+08:00
    use std::fmt::{self, Write};
    
    trait WithMut {
        fn with_mut<'a, R>(&'a mut self, f: impl FnMut(&'a mut Self) -> R) -> R;
    }
    
    impl<T> WithMut for T {
        fn with_mut<'a, R>(&'a mut self, mut f: impl FnMut(&'a mut Self) -> R) -> R {
            f(self)
        }
    }
    
    macro_rules! write_twice {
        ($writer:expr, $($fmt:tt)*) => {{
            use $crate::WithMut; // make sure the trait is in scope
            $writer.with_mut(|mut writer| {
                for _ in 0..2 {
                    if let err @ ::core::result::Result::Err(_) = write!(writer, $($fmt)*) {
                        return err;
                    }
                }
                Ok(())
            })
        }};
    }
    
    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)?;
            write!(f, " <- second")
        }
    }
    
    fn main() {
        let v = vec![1, 2];
        let mut buf = String::new();
        write_twice!(&mut buf, "{v:?}").unwrap();
        println!("{buf:?}");  // [1, 2][1, 2]
        
        println!("{}", Double("hello".into())); // first -> hellohello <- second
    }
    

    此解决方案依赖于辅助特征,我已将其称为WithMut并提供一种with_mut方法。由于write!宏依赖于 autoref 来调用write_fmt,因此我们需要在宏中执行相同操作。的目的WithMut::with_mut是强制$writer将强制转换为可变引用。编写器作为闭包的参数传回&mut Self。这确保$writer只评估一次,同时允许传入值和可变引用而不移动。

    • 4

相关问题

  • 在匹配内重用函数时,匹配臂具有预期的不兼容类型

  • match 语句中的 Rust 类型转换

  • 如何强制匹配的返回类型为()?

  • 原始表示中的 Rust 枚举

  • 有没有办法直接简化 Result<String, VarError> 中 Ok("VAL") 的匹配

Sidebar

Stats

  • 问题 205573
  • 回答 270741
  • 最佳答案 135370
  • 用户 68524
  • 热门
  • 回答
  • Marko Smith

    Vue 3:创建时出错“预期标识符但发现‘导入’”[重复]

    • 1 个回答
  • Marko Smith

    为什么这个简单而小的 Java 代码在所有 Graal JVM 上的运行速度都快 30 倍,但在任何 Oracle JVM 上却不行?

    • 1 个回答
  • Marko Smith

    具有指定基础类型但没有枚举器的“枚举类”的用途是什么?

    • 1 个回答
  • Marko Smith

    如何修复未手动导入的模块的 MODULE_NOT_FOUND 错误?

    • 6 个回答
  • Marko Smith

    `(表达式,左值) = 右值` 在 C 或 C++ 中是有效的赋值吗?为什么有些编译器会接受/拒绝它?

    • 3 个回答
  • Marko Smith

    何时应使用 std::inplace_vector 而不是 std::vector?

    • 3 个回答
  • Marko Smith

    在 C++ 中,一个不执行任何操作的空程序需要 204KB 的堆,但在 C 中则不需要

    • 1 个回答
  • Marko Smith

    PowerBI 目前与 BigQuery 不兼容:Simba 驱动程序与 Windows 更新有关

    • 2 个回答
  • Marko Smith

    AdMob:MobileAds.initialize() - 对于某些设备,“java.lang.Integer 无法转换为 java.lang.String”

    • 1 个回答
  • Marko Smith

    我正在尝试仅使用海龟随机和数学模块来制作吃豆人游戏

    • 1 个回答
  • Martin Hope
    Aleksandr Dubinsky 为什么 InetAddress 上的 switch 模式匹配会失败,并出现“未涵盖所有可能的输入值”? 2024-12-23 06:56:21 +0800 CST
  • Martin Hope
    Phillip Borge 为什么这个简单而小的 Java 代码在所有 Graal JVM 上的运行速度都快 30 倍,但在任何 Oracle JVM 上却不行? 2024-12-12 20:46:46 +0800 CST
  • Martin Hope
    Oodini 具有指定基础类型但没有枚举器的“枚举类”的用途是什么? 2024-12-12 06:27:11 +0800 CST
  • Martin Hope
    sleeptightAnsiC `(表达式,左值) = 右值` 在 C 或 C++ 中是有效的赋值吗?为什么有些编译器会接受/拒绝它? 2024-11-09 07:18:53 +0800 CST
  • Martin Hope
    The Mad Gamer 何时应使用 std::inplace_vector 而不是 std::vector? 2024-10-29 23:01:00 +0800 CST
  • Martin Hope
    Chad Feller 在 5.2 版中,bash 条件语句中的 [[ .. ]] 中的分号现在是可选的吗? 2024-10-21 05:50:33 +0800 CST
  • Martin Hope
    Wrench 为什么双破折号 (--) 会导致此 MariaDB 子句评估为 true? 2024-05-05 13:37:20 +0800 CST
  • Martin Hope
    Waket Zheng 为什么 `dict(id=1, **{'id': 2})` 有时会引发 `KeyError: 'id'` 而不是 TypeError? 2024-05-04 14:19:19 +0800 CST
  • Martin Hope
    user924 AdMob:MobileAds.initialize() - 对于某些设备,“java.lang.Integer 无法转换为 java.lang.String” 2024-03-20 03:12:31 +0800 CST
  • Martin Hope
    MarkB 为什么 GCC 生成有条件执行 SIMD 实现的代码? 2024-02-17 06:17:14 +0800 CST

热门标签

python javascript c++ c# java typescript sql reactjs html

Explore

  • 主页
  • 问题
    • 最新
    • 热门
  • 标签
  • 帮助

Footer

AskOverflow.Dev

关于我们

  • 关于我们
  • 联系我们

Legal Stuff

  • Privacy Policy

Language

  • Pt
  • Server
  • Unix

© 2023 AskOverflow.DEV All Rights Reserve