You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何将Rust重试循环改写为闭包?避免使用mut变量

问题描述

我编写了如下Rust代码,用于重试调用some_func():

let mut result = some_func();
for s in some_iterator {
    if result.is_ok() {
        break;
    }
    thread::sleep(time::Duration::from_millis(10));
    result = some_func();
}
// use result

此处必须将result声明为mut,才能在重试过程中更新它。请问是否可以运用函数式编程技巧,避免将result声明为mut?

我曾考虑过以下实现方案,但并不理想,因为它会遍历some_iterator的所有元素,这不符合我的需求:

let result = some_iterator.fold(some_func(), |result, x| {
    if result.is_ok() {
        return result; 
    }
    // sleep and retry 
});
解决方案

当然可以,以下两种函数式方案都能满足需求,且成功后立即停止迭代,不会遍历所有元素:

方案一:使用try_fold

try_fold支持提前终止迭代,当闭包返回包含成功结果的Ok(result)时,会直接终止并返回该结果:

use std::thread;
use std::time;

let result = some_iterator.try_fold(some_func(), |result, _| {
    if result.is_ok() {
        Ok(result)
    } else {
        thread::sleep(time::Duration::from_millis(10));
        Ok(some_func())
    }
}).unwrap_or_else(|_| unreachable!());

方案二:使用iter::successors

通过successors生成包含重试结果的迭代器,取第一个成功的结果;如果需要限制重试次数,可以结合take:

use std::iter;
use std::thread;
use std::time;

// 无次数限制的重试(直到成功)
let result = iter::successors(Some(some_func()), |prev| {
    if prev.is_ok() {
        None
    } else {
        thread::sleep(time::Duration::from_millis(10));
        Some(some_func())
    }
}).last().unwrap();

// 限制重试次数(与原代码的some_iterator长度一致)
let max_retries = some_iterator.count();
let result = iter::successors(Some(some_func()), |prev| {
    if prev.is_ok() {
        None
    } else {
        thread::sleep(time::Duration::from_millis(10));
        Some(some_func())
    }
}).take(max_retries + 1) // 包含初始调用+max_retries次重试
.last().unwrap();

这两种方式都无需声明mut的result变量,完全通过函数式迭代器方法实现重试逻辑,且符合“成功即停止”的需求。

内容的提问来源于stack exchange,提问作者User12547645

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.04 19:10:40