詳解rust?自動化測試、迭代器與閉包、智能指針、無畏并發(fā)
編寫測試可以讓我們的代碼在后續(xù)迭代過程中不出現(xiàn)功能性缺陷問題;理解迭代器、閉包的函數(shù)式編程特性;Box<T>
智能指針在堆上存儲數(shù)據(jù),Rc<T>
智能指針開啟多所有權模式等;理解并發(fā),如何安全的使用線程,共享數(shù)據(jù)。
自動化測試
編寫測試以方便我們在后續(xù)的迭代過程中,不會改壞代碼。保證了程序的健壯性。
測試函數(shù)通常進行如下操作:
- 設置需要的數(shù)據(jù)或狀態(tài)
- 運行需要測試的代碼
- 斷言其結(jié)果是我們期望的
在 rust 中,通過test
屬性、斷言宏和一些屬性設置來測試代碼。
$> cargo new ifun-grep --lib
創(chuàng)建項目時,通過--lib
表明創(chuàng)建一個庫,會默認生成一個測試示例,在src/lib.rs
中
pub fn add(left: usize, right: usize) -> usize { left + right } #[cfg(test)] mod tests { use super::*; #[test] fn it_works() { let result = add(2, 2); assert_eq!(result, 4); } }
進入到項目中,執(zhí)行cargo test
就會看到執(zhí)行完測試的詳細信息。包括了測試數(shù)量、通過測試數(shù)、失敗測試數(shù)等等維度
首先使用mod tests
定義了一個 tests 模塊,內(nèi)部函數(shù)需要使用外部方法,在最頂部調(diào)用了use super::*;
。這在包的一節(jié)里已有說明。
#[cfg(test)]
標注測試模塊。它可以告訴 rust 在編譯時不需要包含該測試代碼。
#[test]
表明是測試函數(shù),通過 assert_eq!()
斷言結(jié)果值是否相同。
可以手動改動一下斷言值assert_eq!(result, 5)
,再次執(zhí)行可以看到測試不通過,并給出了結(jié)果的不同之處。
由 rust 標準庫提供的斷言測試宏,幫助我們處理結(jié)果值。結(jié)果與預期相同時,則測試會通過;不一樣時,則會調(diào)用panic!
宏,導致測試失敗。
assert!()
一個必傳參數(shù),true
是測試通過;false
測試失敗。assert_eq!()
兩個必傳參數(shù),比對它們是否相同。assert_ne!
兩個必傳參數(shù),比對它們是否不相同。
assert_eq!
和assert_ne
斷言失敗時,會打印出兩個值,便于觀察為什么失敗。因為會打印輸出,所以兩個值必須實現(xiàn)PartialEq
和Debug trait
可以被比較和輸出調(diào)試。
如果是我們自定義的結(jié)構體或枚舉類型,則可以直接增加#[derive(PartialEq, Debug)]
注解。如果是復雜的類型,則需要派生宏trait
,這在后面的文章會講。
#[derive(PartialEq,Debug)] struct User { name: String, }
宏除了它們必須的參數(shù)之外,也可以傳遞更多的參數(shù),這些參數(shù)會被傳遞給format!()
打印輸出。這樣我們可以增加一些輸出,方便解決斷言失敗的問題
assert_eq!(result, 5, "hello rust!");
測試程序處理錯誤
除了測試程序正常執(zhí)行邏輯的結(jié)果,也需要測試程序發(fā)生錯誤時,是否按照我們的錯誤處理邏輯 處理了錯誤。
假設我們的被測試函數(shù)接受的參數(shù)不能大于100
,大于時panic
錯誤 信息
pub fn add(left: usize, right: usize) -> usize { if right > 100 { panic!("the value exceeds 100!"); } left + right } #[cfg(test)] mod tests { use super::*; #[test] fn it_works() { let result = add(2, 102); assert_eq!(result, 104); } }
執(zhí)行測試cargo test
,就算斷言結(jié)果時邏輯正確的,但是我們的函數(shù)限制了參數(shù)最大值,測試不通過。
增加測試用例來測試這種場景,通過增加#[should_panic]
來處理程序確實有這種限制,并panic!
。
#[test] #[should_panic] fn value_exceed_100() { add(5, 120); }
執(zhí)行cargo test
,可以看到測試示例通過了。如果我們符合參數(shù)要求,測試示例就會是失敗
但如果我們代碼中有多個錯誤panic!()
,就會有同樣的多個測試示例不通過,打印輸出并沒有給我們足夠的信息去找到問題所在。
通過should_panic
可選擇參數(shù)expected
提供一個錯誤描述信息,
pub fn add(left: usize, right: usize) -> usize { if right > 100 { panic!("the value exceeds 100!,got {}", right) } else if right < 50 { panic!("the value does not less than 50!,got {}", right) } left + right } #[cfg(test)] mod tests { use super::*; #[test] #[should_panic(expected = "exceeds 100!")] fn value_exceed_100() { add(5, 99); } #[test] #[should_panic(expected = "less than 50!")] fn value_not_less_50() { add(59, 59); } }
也可以通過Result<T,E>
編寫測試,在程序失敗時,返回Err
而不是panic
;
#[test] fn add_equal() -> Result<(), String> { if add(5, 105) == 111 { Ok(()) } else { Err(String::from("Error in add")) } }
此時不能使用#[should_panic()]
注解。也不能使用表達式?
控制測試運行
cargo test
在測試模式下編譯代碼并發(fā)運行生成的測試二進制文件。
1.可以通過設置測試線程,單次只執(zhí)行一個測試示例
$> cargo test -- --test-threads=1
測試線程為 1,程序不會使用任何并行機制。
2.默認的測試在測試示例通過時,不會打印輸出。通過設置在測試成功時也輸出程序中的打印
$> cargo test -- --show-output
3.默認的cargo test
會運行所有測試,通過指定名稱來運行部分測試
$> cargot test add_equal
過濾運行多個測試,可以通過指定測試名稱的一部分,只要匹配這個名稱的測試都會被運行。
$> cargot test value
通過#[ignore]
標記忽略該測試。
#[test] #[ignore] fn add_equal() -> Result<(), String> { if add(5, 105) == 110 { Ok(()) } else { Err(String::from("Error in add")) } }
測試被忽略,但是可以通過cargot test -- --ignored
來運行被忽略的測試。
如果想運行所有的測試,可以通過cargot test -- --include-ignored
集成測試
單元測試可以在指定的模塊中書寫測試實例,每次測試一個模塊,也可以測試私有接口。
集成測試對庫來說是外部的,只能測試公有接口,可測試多個模塊。通過創(chuàng)建tests
目錄編寫獨立的測試文件。
tests/lib.rs
use ifun_grep; #[test] #[should_panic(expected = "exceeds")] fn value_exceed_100() { ifun_grep::add(5, 99); }
隨著集成測試模塊的增多,我們需要更好的組織它們,可以根據(jù)測試的功能將測試分組。將一些測試公共模塊抽離出來,作為其他測試功能組的測試函數(shù)調(diào)用
比如tests/common.rs
pub fn init(){ // something init }
再執(zhí)行cargo test
,會看到運行了tests/common.rs
運行了 0 個測試。這顯然是我們不需要的,可以改寫文件目錄tests/common/mod.rs
,這會告訴 rust 不要將common
看作一個集成測試文件。
迭代器與閉包
rust 類似函數(shù)式編程語言的特性??梢詫⒑瘮?shù)作為參數(shù)值或返回值、將函數(shù)賦值給變量等。
閉包
可以儲存在變量里的類似函數(shù)的結(jié)構。保存在一個變量中或作為參數(shù)傳遞給其他函數(shù)的匿名函數(shù)。
閉包允許捕獲被定義時所在作用域中的值。
#[derive(Debug)] enum Name { Admin, Test, } #[derive(Debug)] struct User {} impl User { fn get_name(&self, name: Option<Name>) -> Name { name.unwrap_or_else(|| self.random_name()) } fn random_name(&self) -> Name { Name::Admin } } fn main(){ let user = User {}; println!("{:?}", user.get_name(Some(Name::Test))); println!("{:?}", user.get_name(None)); }
unwrap_or_else
方法接受一個閉包函數(shù),當一個Some
值存在時直接返回,如果不存在則執(zhí)行其傳入的閉包函數(shù)計算一個值返回。
閉包不需要在參數(shù)或返回值上注明類型。閉包通常只關聯(lián)小范圍的上下文而非任意情景,所以編譯器可以推導出參數(shù)和返回值類型。
也可以顯示定義閉包的參數(shù)和返回值的類型:
fn main(){ let get_age = |age: i8| -> i8 { age }; // let get_age = |age| age; println!("{}", get_age(32)); }
相對于增加參數(shù)或返回值類型使得書寫更加的繁瑣。而對于未標注類型的閉包,在第一次調(diào)用后就確定其參數(shù)和返回值類型,再傳其他類型時就會報錯。
fn main(){ let get_age = |age| age; println!("{}", get_age(String::from("admin"))); // 調(diào)用出錯,已經(jīng)確定了參數(shù)和返回值類型為String println!("{}", get_age(32)); }
捕獲引用或移動所有權
在傳遞給閉包參數(shù)時,需要考慮參數(shù)的傳遞方式:不可變借用、可變借用和獲取所有權。這是根據(jù)傳遞的值決定的。
對于不可變借用,變量可以在任何情形下被訪問。
let str = String::from("hboot"); let print_str = || println!("{:?}", str); println!("{str}"); print_str(); println!("{str}");
而對于可變借用,則只能在借用結(jié)束后調(diào)用.聲明的閉包函數(shù)也需要mut
聲明
let mut str = String::from("hboot"); let mut print_str = || str.push_str("-rust"); // println!("{str}"); print_str(); println!("{str}");
通過move
關鍵字將變量的所有權轉(zhuǎn)移閉包所在的環(huán)境中。
use std::thread; fn main(){ let mut str = String::from("hboot"); println!("{str}"); thread::spawn(move || { str.push_str("-rust"); println!("{str}") }) .join() .unwrap(); }
此時,將變量str
值的所有權轉(zhuǎn)移到了新線程中,主線程則不能再使用。
將被捕獲的值移出閉包和 Fn trait
在閉包環(huán)境中,捕獲和處理值的方式會影響閉包 trait 的實現(xiàn)。trait 是函數(shù)或結(jié)構體指定它們可以使用什么類型的閉包。
從閉包如何任何處理值、閉包自動、漸進實現(xiàn)一個、多個 Fn
trait
FnOnce
適用于調(diào)用一次的閉包。所有閉包都是實現(xiàn)這個 trait,它會將捕獲的值移除閉包。FnMut
不會將捕獲的值移除閉包,可能會修改值。會被調(diào)用 多次。Fn
不會移除捕獲的值,也不修改捕獲的值。會被調(diào)用多次而不改變其環(huán)境。
這是Option<T>
的unwrap_or_else()
方法定義
impl<T> Option<T> { pub fn unwrap_or_else<F>(self, f: F) -> T where F: FnOnce() -> T { match self { Some(x) => x, None => f(), } } }
F
就是閉包指定的類型,T
是返回值類型。FnOnce()->T
表明了閉包會被調(diào)用一次,有值時Some
,返回值;沒有值時None
,f
調(diào)用一次。
在使用閉包時,如果我們不需要捕獲其環(huán)境中的值,則可以不使用閉包,而使用傳遞函數(shù)作為參數(shù)。
迭代器
迭代器是處理元素序列的方式。遍歷序列中的每一項以及決定序列何時結(jié)束的邏輯。
fn main(){ let arr = [1, 2, 3, 4]; for val in arr { println!("{}", val) } }
迭代器都定義了Iterator
trait,并實現(xiàn)next
方法。調(diào)用next
返回迭代器的一個項,封裝在Some
中,結(jié)束后返回None
pub trait Iterator { type Item; fn next(&mut self) -> Option<Self::Item>; }
type Item
和Self::Item
定義了 trait 的關聯(lián)類型。表明了迭代器返回值類型為Item
可以通過next()
方法迭代獲取值:
fn main(){ let arr = [1, 2, 3, 4]; let mut iter = arr.iter(); println!("{:?}", iter.next()); println!("{:?}", iter.next()); }
iter()
生成一個不可變引用的迭代器。對于迭代器實例iter
必須是mut
可變的。
into_ter()
獲取到 arr 所有權的迭代器。iter_mut()
可以獲取到可變引用迭代器。 消費適配器
調(diào)用next()
方法的方法被稱為消費適配器。
fn main() { let arr = [1, 2, 3, 4]; let total: i8 = arr.iter().sum(); println!("{}", total); }
這些方法總是會獲取迭代器的所有權并反復調(diào)用 next
來遍歷迭代器。sum()
方法返回調(diào)用next
方法獲取值,最終返回和值。
迭代器適配器
將當前迭代器變?yōu)椴煌愋偷牡???梢枣準秸{(diào)用多個迭代器適配器,但是每次調(diào)用都必須調(diào)用消費適配器來獲取調(diào)用結(jié)果。
fn main(){ let arr = [1, 2, 3, 4]; let arr2: Vec<_> = arr.iter().map(|val| val + 1).collect(); for val in arr2 { println!("{}", val) } }
map()
方法接受一個閉包函數(shù),可以在遍歷元素上執(zhí)行任何操作。進行了一次迭代適配器操作,然后通過collect()
方法獲取調(diào)用的結(jié)果值。
智能指針
指針是一個包含內(nèi)存地址的變量。智能指針是一類數(shù)據(jù)結(jié)構,表現(xiàn)同指針,并擁有額外的元數(shù)據(jù)和功能。
智能指針通常使用結(jié)構體實現(xiàn),實現(xiàn)了Deref
和Drop
trait。deref trait
允許智能指針結(jié)構體實例表現(xiàn)的像引用一樣;drop trait
允許智能指針離開作用域時自定義運行代碼
標準庫中常用的智能指針:
Box<T>
用于在堆上分配值Rc<T>
引用計數(shù)類型,其數(shù)據(jù)可以有多個所有者Ref<T>、RefMut<T>
通過RefCell<T>
訪問,這是一個在運行時執(zhí)行借用規(guī)則的類型。
Box<T>
智能指針 box 允許將一個值放在堆上而不是棧上。留在棧上的則是指向堆數(shù)據(jù)的指針。
在以下情況下可以考慮使用:
- 編譯時未知大小的類型,又想在確切大小的上下文中使用這個類型的值。
- 當有大量數(shù)據(jù)不被拷貝的情況下轉(zhuǎn)移所有權的時候
- 當有一個值只關心它的類型是否實現(xiàn)特定 trait,而不是具體類型的時候
fn main(){ let b = Box::new(100); println!("{}", b); }
直接聲明創(chuàng)建 box 類型變量,并分配了一個值100
存儲在堆上, 可以直接訪問變量訪問值。
通過cons
list 數(shù)據(jù)結(jié)構定義遞歸數(shù)據(jù)類型
它是construct function
的縮寫,利用兩個參數(shù)構造一個新的列表.最后一項值包含了Nil
值,標識結(jié)束
enum List { Cons(i32, Box<List>), Nil, } use crate::List::{Cons, Nil}; fn main(){ let list = Cons(1, Box::new(Cons(2, Box::new(Cons(3, Box::new(Nil)))))); }
Cons
可能會無限嵌套下去,為了保證 rust 編譯時計算需要的大小,只能通過Box
來幫助 rust 計算出List
需要的大小。
Deref
trait 重載解引用運算符*
之前已經(jīng)使用過*
解引用值,可以獲取到指針指向引用的值。
fn main(){ let mut s = String::from("hboot"); let s1 = &mut s; *s1 += " admin"; println!("{}", s) }
s1
是 s 的可變引用,再通過*
解引用后,可以修改存儲在堆上的數(shù)據(jù)。
也可以通過Box<T>
代替引用,和*
擁有相同的功能。
fn main(){ let s = String::from("hboot"); let mut s1 = Box::new(s); *s1 += " admin"; println!("{:?}", s1); }
Box
會拷貝s
在棧上的指針數(shù)據(jù),導致存儲在堆上的數(shù)據(jù)所有權被轉(zhuǎn)移,s
在后續(xù)變的不可用。
自定義實現(xiàn)一個智能指針MyBox
,它可以做到上面的解引用操作
#[derive(Debug)] struct MyBox<T>(T); impl<T> MyBox<T> { fn new(val: T) -> MyBox<T> { MyBox(val) } }
實現(xiàn)了一個元組結(jié)構體,自定義實例new
方法,接受一個參數(shù)進行初始化操作。還需要實現(xiàn)解引用功能,Deref
trait 由標準庫提供,實現(xiàn) deref 方法
use std::ops::Deref; impl<T> Deref for MyBox<T> { type Target = T; fn deref(&self) -> &Self::Target { &self.0 } }
上述的解引用例子,則可以由MyBox
代替實現(xiàn)。type Target = T
定義了 trait 的關聯(lián)類型,&self.0
訪問元組結(jié)構體的第一個元素。
fn main(){ let s = String::from("hboot"); let s1 = MyBox::new(s); // *s1 += " admin"; println!("{:?}", *s1); }
因為實現(xiàn)的是Deref
所以不能修改,修改時需要實現(xiàn)DerefMut
trait。
實現(xiàn)了Deref
trait 的數(shù)據(jù)類型,在函數(shù)傳參時,可做到隱式轉(zhuǎn)換,而不需要手動去轉(zhuǎn)換為參數(shù)需要的類型。
fn print(val: &str) { println!("{}", val) } fn main(){ // 輸出上面的示例 s1 print(&s1); }
對于數(shù)據(jù)的強制轉(zhuǎn)換,只能將可變引用轉(zhuǎn)為不可變引用;不能將不可變引用轉(zhuǎn)為可變引用。
Drop
trait 運行清理代碼
實現(xiàn)了Drop
trait 的數(shù)據(jù),在離開作用域時,會調(diào)用其實現(xiàn)的drop
方法,它獲取一個可變引用。
為上述的MyBox
實現(xiàn)Drop
,無需引入,Drop
trait 是 prelude 的。
impl<T> Drop for MyBox<T> { fn drop(&mut self) { println!("mybox drop value"); } }
再次調(diào)用執(zhí)行,可以看到最終在程序執(zhí)行完畢后,打印輸出了mybox drop value
. drop
會自動執(zhí)行,而無需手動調(diào)用。
如果想要提前銷毀資源,則需要std::mem::drop
,可以調(diào)用drop
方法
fn main(){ drop(s1); // 手動清理后,后續(xù)不能再使用s1 // print(&s1); }
Rc<T>
引用計數(shù)啟用多所有權模式
在圖形結(jié)構中,每個節(jié)點都有多個邊指向,所以每個節(jié)點都會擁有指向它的邊的所有權。
通過使用Rc<T>
類型,記錄被引用的數(shù)量,來確定這個值有沒有被引用。如果為 0 沒有被引用,則會被清理。
Rc<T>
只適用于單線程
創(chuàng)建Rc
類型的變量s
,然后通過Rc::clone
克隆變量s
生成s1\s2
.
use std::rc::Rc; fn main(){ let s = Rc::new(String::from("hboot")); let s1 = Rc::clone(&s); let s2 = Rc::clone(&s); println!("s:{},s1:{},s2:{}", s, s1, s2) }
這里可以看到s1、s2
沒有獲取s
的所有權,它們?nèi)匀煌瑫r生效。Rc::clone
不同于深拷貝,只會增加引用計數(shù)。
可以通過strong_count()
方法查看被引用次數(shù)
fn main(){ let s = Rc::new(String::from("hboot")); println!("s create - {}", Rc::strong_count(&s)); let s1 = Rc::clone(&s); println!("s1 create - {}", Rc::strong_count(&s)); { let s2 = Rc::clone(&s); println!("s2 create - {}", Rc::strong_count(&s)); } println!("s2 goes out of scope - {}", Rc::strong_count(&s)); }
執(zhí)行測試輸出為
通過不可變引用,Rc<T>
允許程序在多個部分之間只讀地共享數(shù)據(jù)。
RefCell<T>
允許修改不可變引用
根據(jù) rust 的不可變引用規(guī)則,被引用的變量是不允許修改。但是在某些模式下,可以做到修改,也就是內(nèi)部可變性模式。
內(nèi)部可變性通過在數(shù)據(jù)結(jié)構中使用unsafe
代碼來模糊 rust 的不可變性和借用規(guī)則。unsafe
不安全代碼表明我們需要手動去檢查代碼而不是讓編譯器檢查。
RefCell<T>
類型是在代碼運行時作用檢測不可變或可變借用規(guī)則,而通常的規(guī)則檢測是在編譯階段。
特點:
可以在允許出現(xiàn)特定內(nèi)存安全的場景中使用。需要確認你的代碼遵守了借用規(guī)則,但是 rust 無法理解只能用于單線程
RefCell<T>
在運行時記錄借用,通過borrow()
和borrow_mut()
方法,會返回Ref<T>
和RefMut<T>
智能指針,并實現(xiàn)了Deref
trait.
定義一個MixName
trait,然后結(jié)構體User
實現(xiàn)了它,并實現(xiàn)它的方法mix
.
use std::cell::RefCell; pub trait MixName { fn mix(&self, suffix: &str); } struct User { name: RefCell<String>, } impl User { fn new() -> User { User { name: RefCell::new(String::from("hboot")), } } } impl MixName for User { fn mix(&self, suffix: &str) { self.name.borrow_mut().push_str(suffix); } }
mix
方法修改了 self 內(nèi)部屬性name
的值,但是我們可以看到&self
時不可變引用,這歸功于RefCell<T>
創(chuàng)建值,使得不可變借用可以修改其內(nèi)部值。
fn main(){ let user = User::new(); user.mix(" hello"); println!("{:?}", user.name.borrow()); }
執(zhí)行程序可以看到內(nèi)部的值已經(jīng)被修改了。RefCell<T>
會在調(diào)用borrow
時,記錄借用次數(shù),當離開了作用域時,借用次數(shù)減一。
RefCell<T>
只能有一個所有者,結(jié)合Rc<T>
使其擁有多個可變數(shù)據(jù)所有者。
use std::cell::RefCell; use std::rc::Rc; fn main(){ let s = Rc::new(RefCell::new(String::from("hboot"))); let s1 = Rc::clone(&s); let s2 = Rc::clone(&s); *s.borrow_mut() += " good"; println!("{:?}", s); }
通過RefCell
來創(chuàng)建變量,然后通過Rc
開啟多所有權,這樣在*s.borrow_mut() += " good";
,修改后,變量s、s1、s2
的值都發(fā)生了變更。
但是這只能在單線中使用,如果想要多線程使用,則需要使用并發(fā)安全的Mutex<T>
類型。
無畏并發(fā)
并發(fā)編程 - 代表程序的不同部分相互獨立的運行。
并行編程 - 代表程序不同部分同時執(zhí)行。
thread
多線程運行代碼
多線程運行代碼可以提高程序的執(zhí)行效率。也會造成一些問題
- 多個線程在不同時刻訪問同一數(shù)據(jù)資源,形成競爭
- 相互等待對方,造成死鎖
- 一些情況下出現(xiàn)的難以修復的 bug
使用thread::spawn
創(chuàng)建一個線程,它接受一個閉包函數(shù)
use std::thread; fn main() { thread::spawn(|| { println!("hello!"); }); println!("rust!"); }
可以看到輸出,先是rust!
,也就是主線程先執(zhí)行??梢远啻螆?zhí)行cargo run
以觀察結(jié)果,會出現(xiàn)新線程沒有打印輸出,這是因為主線程結(jié)束,新線程也會結(jié)束,而不會等待新線程是否執(zhí)行完畢。
可以通過線程休眠,展示這一特點
use std::thread; use std::time::Duration; fn main() { thread::spawn(|| { thread::sleep(Duration::from_millis(2)); println!("hello!"); }); println!("rust!"); }
程序基本沒有什么機會切換到新線程去執(zhí)行,也看不到新線程的打印輸出。
可以通過thread::spawn
的返回值線程實例,然后調(diào)用join()
方法,來等待線程結(jié)束
let thread = thread::spawn(|| { thread::sleep(Duration::from_millis(2)); println!("hello!"); }); println!("rust!"); thread.join().unwrap();
再次執(zhí)行,可以看到新線程的打印輸出。join()
會阻塞當前線程,知道線程實例thread
執(zhí)行完畢。
可以將thread.join().unwrap();
放在主線程輸出之前,優(yōu)先執(zhí)行
thread.join().unwrap(); println!("rust!");
通過move
關鍵字強制閉包獲取其所有權,thread::spawn
創(chuàng)建線程給的閉包函數(shù)沒有任何參數(shù),需要使用主線程里的變量
let name = String::from("hboot"); let thread = thread::spawn(move || { thread::sleep(Duration::from_millis(2)); println!("hello! - {}", name); });
新線程強制獲取了環(huán)境中變量的所有權,保證了新線程執(zhí)行不會出錯。如果是引用,那么由于新線程的執(zhí)行順序,可能會在主線程執(zhí)行過程使引用失效,從而導致新線程執(zhí)行報錯
線程間消息傳遞
通過channel
實現(xiàn)線程間消息的傳遞并發(fā)。
通過mpsc::channel
創(chuàng)建通信通道,這個通道可以有多個發(fā)送端,但只能有一個接收端.
use std::sync::mpsc; fn main(){ let (send, receive) = mpsc::channel(); thread::spawn(move || { let name = String::from("rust"); send.send(name).unwrap(); }); let receive_str = receive.recv().unwrap(); println!("get thread msg :{}", receive_str); }
mpsc::channel()
生成一個通過,返回一個元組,第一個是發(fā)送者,第二個是接收者。然后創(chuàng)建一個新線程,通過實例對象send
發(fā)送一條信息;在主線程中通過實例對象receive
接受數(shù)據(jù)。
不管是send()
發(fā)送方法還是recv()
方法,它們都返回Result<T,E>
類型,如果接受端或發(fā)送端被清除了,則會返回錯誤。
接受recv()
方法是阻塞線程的,也就是必須接收到一個值。還有一個方法try_recv()
方法則不會阻塞,需要頻繁去調(diào)用,在有可用消息時進行處理。
新線程將變量name
發(fā)送出去,那么它的所有權也被轉(zhuǎn)移 出去了,后續(xù)不能使用它
send.send(name).unwrap(); // 在發(fā)送后,不能再使用改變量 println!("{}", name);
當在子線程中連續(xù)多次發(fā)送多個值時,可以通過迭代器遍歷receive
獲取值
fn main(){ let (send, receive) = mpsc::channel(); thread::spawn(move || { send.send(1).unwrap(); send.send(10).unwrap(); send.send(100).unwrap(); }); for receive_str in receive { println!("{}", receive_str); } }
上述例子只是單發(fā)送者,可以通過clone()
方法克隆send
發(fā)送對象,然后傳給另一個線程
fn main(){ let (send, receive) = mpsc::channel(); let send_str = send.clone(); thread::spawn(move || { send_str.send("hello").unwrap(); send_str.send("rust").unwrap(); }); thread::spawn(move || { send.send("good").unwrap(); send.send("hboot").unwrap(); }); for receive_str in receive { println!("{}", receive_str); } }
創(chuàng)建兩個線程,一個線程傳入時克隆的send_str
,它們都發(fā)送消息,然后在主線程中,接收到所有消息。
多個線程由于執(zhí)行順序?qū)е麓蛴≥敵龅捻樞蛞膊槐M相同。這依賴于系統(tǒng),我們可以通過線程休眠做實驗,觀察到輸出的順序不同
線程間共享狀態(tài)
除了相互之間發(fā)送消息外, 還可以通過共享數(shù)據(jù),來傳遞數(shù)據(jù)狀態(tài)變化。
通過Mutex<T>
創(chuàng)建共享數(shù)據(jù),在需要使用的線程中通過lock()
獲取鎖,以訪問數(shù)據(jù)。
use std::sync::{Mutex}; fn main()[ let name = Mutex::new(String::from("hboot")); { let mut name = name.lock().unwrap(); *name += " good!"; } println!("{:?}", name.lock().unwrap()); ]
新創(chuàng)建的數(shù)據(jù)hboot
,在局部作用域中獲取鎖,然后解引用后變更值,最終打印輸出可以看到變更后的數(shù)據(jù)。
Mutext<T>
是一個智能指針,調(diào)用lock()
返回了一個MutexGuard
智能指針,它實現(xiàn)了Deref
來指向內(nèi)部數(shù)據(jù),同時也提供Drop
實現(xiàn)了當離開作用域時自動釋放鎖。
正因為這樣,我們在編碼時,不會因為忘記釋放鎖而導致其他線程訪問不了數(shù)據(jù)。
如果想要在多個線程中訪問共享數(shù)據(jù),因為線程需要轉(zhuǎn)移所有權,這樣導致共享數(shù)據(jù)每次只能在一個線程中使用,通過Arc<T>
來創(chuàng)建多所有者,使得共享數(shù)據(jù)可被多個線程同時訪問。
use std::sync::{Arc, Mutex}; use std::thread; fn main(){ let name = Arc::new(Mutex::new(String::from("hboot"))); let mut thread_arr = vec![]; for val in ["admin", "test", "hello", "rust"] { let name = Arc::clone(&name); let thread = thread::spawn(move || { let mut name = name.lock().unwrap(); *name += val; }); thread_arr.push(thread); } for thread in thread_arr { thread.join().unwrap(); } println!("{:?}", name.lock().unwrap()) }
Arc<T>
擁有和Rc<T>
相同的 api,它可以用于并發(fā)環(huán)境的類型。這是一個原子引用計數(shù)類型。
Mutex<T>
同RefCell<T>
一樣,提供了內(nèi)部可變性,通過獲取內(nèi)布值的可變引用修改值。當然,Mutex<T>
也會有出現(xiàn)相互引用鎖死的風險,兩個線程需要鎖住兩個資源而各自已經(jīng)鎖了一個,造成了互相等待的問題。
Sync
和Send trait
擴展并發(fā)
除了使用 rust 標準庫提供的處理并發(fā)問題,還可以使用別人編寫的并發(fā)功能
當嘗試編寫并發(fā)功能時,有兩個并發(fā)概念:
- 通過
Send trait
表明實現(xiàn)了Send
的類型值的所有權可以在線程間傳遞。rust 幾乎所有類型都是Send
, 還有一些不能Send
,比如Rc<T>
,它只能用于單線程, - 通過
Sync trait
表明實現(xiàn)了Sync
的類型可以安全的在多個線程中擁有其值的引用。Rc<T>、RefCell<T>
都不是Sync
類型的。
根據(jù)這兩個概念,可以手動創(chuàng)建用于并發(fā)功能的并發(fā)類型,在使用時需要多加小心,以維護其安全保證。
到此這篇關于rust 自動化測試、迭代器與閉包、智能指針、無畏并發(fā)的文章就介紹到這了,更多相關rust 自動化測試內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
Rust開發(fā)環(huán)境搭建到運行第一個程序HelloRust的圖文教程
本文主要介紹了Rust開發(fā)環(huán)境搭建到運行第一個程序HelloRust的圖文教程,文中通過圖文介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2023-12-12rust?創(chuàng)建多線程web?server的詳細過程
web?server?中主要的兩個協(xié)議是?http?和?tcp,tcp?是底層協(xié)議,http?是構建在?tcp?之上的,本篇文章重點給大家介紹rust?創(chuàng)建多線程web?server的詳細過程,感興趣的朋友跟隨小編一起看看吧2023-11-11Go調(diào)用Rust方法及外部函數(shù)接口前置
這篇文章主要為大家介紹了Go調(diào)用Rust方法及外部函數(shù)接口前置示例實現(xiàn),有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2022-06-06rust多個mod文件引用和文件夾mod使用注意事項小結(jié)
在 Rust 項目中,可以使用 mod 關鍵字將一個文件夾或一個 rs 文件作為一個模塊引入到當前文件中,本文給大家介紹rust多個mod文件引用和文件夾mod使用注意事項小結(jié),感興趣的朋友跟隨小編一起看看吧2024-03-03