Rust闭包详解 - 技术栈

文章目录

- 闭包

闭包

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

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

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

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

rust 复制代码

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

捕获外部变量

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

rust 复制代码

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

捕获变量有三种方法

传引用捕获（默认的，类似于 &T）
传值捕获（类似 T）
借用捕获（类似 &mut T）

rust 复制代码

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关键字，闭包会获取他捕获的变量的所有权，此时，变量的所有权会转移到内部，外部作用域无法使用这些变量

闭包的特性

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

例如

使用 impl Fn 返回闭包

rust 复制代码

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>返回闭包

rust 复制代码

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 的生命周期系统确保闭包不会比它们捕获的任何变量活得更长。