Rust语言之多线程
文章目录
- 一、简介
- 二、创建线程
- 1.创建一个线程
- 2.创建多个线程
- 生成随机数
- 尝试让程序睡一会儿
- 引入多线程
- 三、线程返回值的处理
- 1.每个线程处理一个独立的值
- 2.多个线程处理一个值
- Arc(原子引用计数)
- Mutex(互斥锁)
- RwLock(读写锁)
一、简介
多线程 是一种并发执行的技术,它允许一个程序或进程同时执行多个线程。每个线程都是程序执行的一个独立路径,它们可以并行运行,共享进程的资源(如内存空间),但每个线程有自己的指令指针、堆栈和局部变量。多线程的主要目的是提高程序的执行效率,通过同时执行多个任务来充分利用计算机的多核处理器。
- 在Rust语言中,多线程编程是通过标准库中的
std::thread
模块来实现的。Rust提供了创建和管理线程的API,以及用于线程间同步和通信的机制,如互斥锁(Mutex)和通道(Channel)。
二、创建线程
1.创建一个线程
use std::thread; fn main() { // 创建一个新线程 let handle = thread::spawn(|| { // 在新线程中执行的代码 println!("Hello from a new thread!"); }); // 等待线程结束 handle.join().unwrap();
}
2.创建多个线程
use std::thread; fn main() { // 创建一个向量来存储线程的句柄 let mut threads = vec![]; // 创建多个线程 for i in 0..5 { // 使用闭包捕获变量i的值 let thread_number = i; let handle = thread::spawn(move || { // 在新线程中打印线程编号 println!("线程 {} 正在运行", thread_number); }); // 将线程句柄添加到向量中 threads.push(handle); } // 等待所有线程完成 for handle in threads { handle.join().unwrap(); } // 在主线程中打印一些信息 for i in 0..5 { println!("主线程打印数字: {}", i); }
}# 输出结果:
线程 0 正在运行
线程 1 正在运行
线程 2 正在运行
线程 3 正在运行
线程 4 正在运行
主线程打印数字: 0
主线程打印数字: 1
主线程打印数字: 2
主线程打印数字: 3
主线程打印数字: 4
- 从输出结果上看,仍然像是顺序执行,所以这里引入一个休眠,让线程执行的时候随机休眠0-3秒。
生成随机数
由于Rust核心语言中没有随机数生成的函数,需要使用rand库来进行
# 首先需要在Cargo.toml中添加以下内容
[dependencies]
rand = "0.8"# 然后在代码中用use 引入
use rand::Rng;
use rand::thread_rng; fn main() { let mut rng = thread_rng(); for _i in 0..10{let random_number = rng.gen_range(1..4); println!("随机数是: {}", random_number); }
}
# 结果:
随机数是: 3
随机数是: 1
随机数是: 3
随机数是: 1
随机数是: 3
随机数是: 1
随机数是: 3
随机数是: 2
随机数是: 3
随机数是: 1
随机数生成的区间与循环一样,是一个前闭后开的区间
尝试让程序睡一会儿
use rand::Rng;
use std::{thread::sleep, time::Duration}; fn main() { // 创建一个随机数生成器 let mut rng = rand::thread_rng(); // 生成一个0到3之间的随机秒数 let random_seconds: u64 = rng.gen_range(0..4); // 将秒数转换为Duration let duration = Duration::from_secs(random_seconds); // 让当前线程睡眠指定的时间 sleep(duration); // 之后的代码会在等待后执行 println!("等待了 {} 秒", random_seconds);
}
引入多线程
use rand::Rng;
use std::{thread::sleep, time::Duration};
use std::thread; fn main() { // 创建一个向量来存储线程的句柄 let mut threads = vec![]; // 创建多个线程 for i in 1..=10 { // 使用闭包捕获变量i的值 let thread_number = i; let handle = thread::spawn(move || { // 在新线程中打印线程编号 println!("线程 {} 正在运行", thread_number); // 创建一个随机数生成器 let mut rng = rand::thread_rng(); // 生成一个0到3之间的随机秒数 let random_seconds: u64 = rng.gen_range(0..4); // 将秒数转换为Duration let duration = Duration::from_secs(random_seconds); // 让当前线程睡眠指定的时间 sleep(duration); println!("线程 {} 运行结束,休息了{}秒.", thread_number,random_seconds); }); // 将线程句柄添加到向量中 threads.push(handle); } // 等待所有线程完成 for handle in threads { handle.join().unwrap(); } // 在主线程中打印一些信息 for i in 0..5 { println!("主线程打印数字: {}", i); }
}# 输出结果
线程 3 正在运行
线程 2 正在运行
线程 5 正在运行
线程 7 正在运行
线程 7 运行结束,休息了0秒.
线程 4 正在运行
线程 6 正在运行
线程 1 正在运行
线程 8 正在运行
线程 9 正在运行
线程 10 正在运行
线程 6 运行结束,休息了1秒.
线程 4 运行结束,休息了1秒.
线程 3 运行结束,休息了1秒.
线程 9 运行结束,休息了1秒.
线程 1 运行结束,休息了2秒.
线程 10 运行结束,休息了2秒.
线程 5 运行结束,休息了2秒.
线程 2 运行结束,休息了3秒.
线程 8 运行结束,休息了3秒.
主线程打印数字: 0
主线程打印数字: 1
主线程打印数字: 2
主线程打印数字: 3
主线程打印数字: 4
三、线程返回值的处理
对于有返回值的多线程来说有两种情况,一种是每个线程处理一个独立的值,用向量接收,另一种是多个线程处理一个值。
1.每个线程处理一个独立的值
use std::thread; fn main() { let mut handles = vec![]; for i in 0..5 { let handle = thread::spawn(move || { return i*i;}); handles.push(handle); } let mut results = vec![]; for handle in handles { match handle.join() { Ok(value) => results.push(value), Err(e) => println!("Thread panicked: {:?}", e), } } println!("Results: {:?}", results); //Results: [0, 1, 4, 9, 16]
}
2.多个线程处理一个值
由于多个线程处理一个值,可能造成条件竞争,属于线程不安全行为,Rust语言中提供了3种处理行为。
- Arc 只读访问,用于共享只读数据,通过原子引用计数管理生命周期。
- Mutex 互斥锁,用于保护数据,确保一次只有一个线程可以访问数据(提供独占访问)。
- RwLock 读写锁,用于保护数据,但允许多个读者同时访问,写者必须独占访问。
Arc(原子引用计数)
Arc
是一个提供共享所有权的智能指针。它用于在多个所有者之间共享数据,且只允许对这些数据进行只读访问。Arc通过原子操作维护一个引用计数,确保数据的生命周期至少与最长的所有者一样长。当最后一个Arc指针被丢弃时,其指向的数据也会被释放。
use std::sync::Arc;
use std::thread; fn main() { // 创建一个要在多个线程之间共享的值 let data = Arc::new(vec![1, 2, 3, 4, 5]); // 创建一个向量来存储线程的句柄 let mut handles = vec![]; println!("Thread {:?} is reading value: {:?}", thread::current().id(), &data); // 主线程的线程ID为 1// 创建几个线程来只读访问数据 for _i in 0..data.len() { let data = data.clone(); // 克隆Arc以便在线程中使用 let handle = thread::spawn(move || { // 获取Vec的引用以便索引 // 使用 {:?} 来打印 ThreadId println!("Thread {:?} is reading value: {:?}", thread::current().id(), &data); }); handles.push(handle); } // 等待所有线程完成 for handle in handles { handle.join().unwrap(); }
}
# 运行结果
Thread ThreadId(1) is reading value: [1, 2, 3, 4, 5]
Thread ThreadId(2) is reading value: [1, 2, 3, 4, 5]
Thread ThreadId(3) is reading value: [1, 2, 3, 4, 5]
Thread ThreadId(4) is reading value: [1, 2, 3, 4, 5]
Thread ThreadId(5) is reading value: [1, 2, 3, 4, 5]
Thread ThreadId(6) is reading value: [1, 2, 3, 4, 5]
Mutex(互斥锁)
Mutex
是一个提供互斥访问的智能指针。它用于保护数据,确保一次只有一个线程能够访问数据。当一个线程拥有Mutex的锁时,其他尝试获取锁的线程将被阻塞,直到锁被释放。
use std::sync::{Arc, Mutex};
use std::thread; fn main() { // 创建一个Arc包裹的互斥锁和值 let counter = Arc::new(Mutex::new(1)); let mut handles = vec![]; // 创建几个线程来增加计数器 for i in 1..10 { // 克隆Arc智能指针,而不是Mutex或它的值 let counter = Arc::clone(&counter); let handle = thread::spawn(move || { // 获取互斥锁以便修改值 let mut num = counter.lock().unwrap(); *num *=i;}); handles.push(handle); } // 等待所有线程完成 for handle in handles { handle.join().unwrap(); } // 输出最终计数器的值 println!("Result: {}", *counter.lock().unwrap()); // Result: 362880
}
RwLock(读写锁)
RwLock
是一个提供读写锁定的智能指针。与Mutex不同,RwLock允许多个读者同时访问数据,但写者必须独占锁。当写者拥有锁时,任何尝试获取读锁或写锁的线程都将被阻塞。当没有写者时,可以有多个读者同时访问数据。
use std::sync::{Arc, RwLock}; fn main() { let data = Arc::new(RwLock::new(0)); let mut handles = vec![]; // 创建多个读线程 for i in 0..5 { let data = Arc::clone(&data); let handle = std::thread::spawn(move || { let num = data.read().unwrap(); println!("Thread {} Reading value: {}", i,*num); }); handles.push(handle); } // 创建一个写线程 let data = Arc::clone(&data); let handle = std::thread::spawn(move || { let mut num = data.write().unwrap(); *num += 1; println!("Writing value: {}", *num); }); handles.push(handle); for handle in handles { handle.join().unwrap(); }
}
# 第一次执行结果
Thread 0 Reading value: 0
Thread 3 Reading value: 0
Thread 1 Reading value: 0
Thread 4 Reading value: 0
Thread 2 Reading value: 0
Writing value: 1
# 第二次执行结果
Thread 0 Reading value: 0
Thread 3 Reading value: 0
Thread 1 Reading value: 0
Thread 4 Reading value: 0
Thread 2 Reading value: 0
Writing value: 1
这里有一个问题,就是如果写线程最后执行,那么读线程读的都是原始数据,如果写线程先执行,那么读的就是修改后的数据,所以对读写顺序有要求的话应该做好时序的控制