Rust闭包详解

发布于:2025-04-02 ⋅ 阅读:(17) ⋅ 点赞:(0)

闭包

Rust中的闭包是一种匿名函数,可以捕获并存储环境中的变量,有点类似于Lambda表达式

闭包允许在其定义的作用域之外访问变量,并且可以在需要时将其移动或者借用给闭包

闭包在Rust中被广泛应用于函数式编程、并发编程和事件驱动编程等领域

比较适合用于短小的自定义逻辑的场景

let name = |参数列表| 表达式或语句块;
let add = |x: i32| x + 1;
let cal = |x, y, z| x * y + z;
let res = cal(1, 2, 3);

捕获外部变量

闭包可以捕获环境中的变量

let x = 5;
let add = |num| num + x;
println!("{}", add(1)); // 输出 6

捕获变量有三种方法

  • 传引用捕获(默认的,类似于 &T)
  • 传值捕获(类似 T)
  • 借用捕获(类似 &mut T)
fn main(){
  let mut num = 114;
  
  let print_num = || println!("num = {}", num);
  print_num(); // 直接输出内容
  
  let take_num = move || println!("num taken = {}", num);
  take_num(); // 直接输出内容
  // println!("{}", num); // 取消注释会报错,因为num的所有权被转移了
  
  let mut change_num = || num += 1;
  change_num();
  println!("num = {}", num);
}
  • 默认是按照引用捕获外部变量的
  • 使用move关键字可以强制按值捕获,但是所有权会转移到闭包内部,这里关于所有权的问题,我们会在后面讲解
  • 如果闭包需要修改外部变量,需要显示声明为mut闭包

移动和借用

闭包可以通过move关键字获取外部变量的所有权,或者通过借用的方式获取外部变量的引用

借用:默认情况下,闭包会借用他捕获的环境中的变量,闭包可以使用这些变量,但是不能修改他们的所有权,此时闭包内部和外部的作用域都可以使用这个变量

获得所有权:在闭包之前添加move关键字,闭包会获取他捕获的变量的所有权,此时,变量的所有权会转移到内部,外部作用域无法使用这些变量

闭包的特性

  1. 闭包可以作为参数传递给函数,例如迭代器的map,filter方法都需要用到闭包
  2. 闭包还可以作为返回值,由于闭包是匿名的,我们需要使用impl Trait 或者

例如

使用 impl Fn 返回闭包

fn apply_to_value<F>(val: i32, f: F) -> i32
where
    F: Fn(i32) -> i32,
{
    f(val)
}

fn main() {
    let double = |x| x * 2;
    let result = apply_to_value(5, double);
    println!("Result: {}", result); // 输出: Result: 10
}

使用 Box<dyn Fn>返回闭包

fn make_adder(x: i32) -> Box<dyn Fn(i32) -> i32> {
    Box::new(move |y| x + y)
}

fn main() {
    let add_ten = make_adder(10);
    println!("10 + 2 = {}", add_ten(2)); // 输出: 10 + 2 = 12
}

闭包和性能

Rust 的闭包是轻量级的,并且 Rust 的编译器会进行优化,使得闭包的调用接近于直接调用函数。

闭包和生命周期

闭包的生命周期与它们所捕获的变量的生命周期相关。Rust 的生命周期系统确保闭包不会比它们捕获的任何变量活得更长。


网站公告

今日签到

点亮在社区的每一天
去签到