使用 Swift 的并發(fā)系統(tǒng)并行運行多個任務(wù)

前言

Swift 內(nèi)置并發(fā)系統(tǒng)的好處之一是它可以更輕松地并行執(zhí)行多個異步任務(wù),這反過來又可以使我們顯著加快可以分解為單獨部分的操作。

在本文中,讓我們看一下幾種不同的方法,以及這些技術(shù)中的每一種何時特別有用。

從異步到并發(fā)

首先,假設(shè)我們正在開發(fā)某種形式的購物應(yīng)用程序來顯示各種產(chǎn)品,并且我們已經(jīng)實現(xiàn)了一個ProductLoader允許我們使用一系列異步 API 加載不同產(chǎn)品集合的應(yīng)用程序,如下所示:

class ProductLoader {
    ...

    func loadFeatured() async throws -> [Product] {
        ...
    }
    
    func loadFavorites() async throws -> [Product] {
        ...
    }
    
    func loadLatest() async throws -> [Product] {
        ...
    }
}

盡管大多數(shù)情況下上述每個方法都可能會被單獨調(diào)用,但假設(shè)在我們應(yīng)用程序的某些部分中,我們還希望形成一個Recommendations包含這三個ProductLoader方法的所有結(jié)果的組合模型:

extension Product {
    struct Recommendations {
        var featured: [Product]
        var favorites: [Product]
        var latest: [Product]
    }
}

一種方法是使用await關(guān)鍵字調(diào)用每個加載方法,然后使用這些調(diào)用的結(jié)果來創(chuàng)建我們Recommendations模型的實例——如下所示:

extension ProductLoader {
    func loadRecommendations() async throws -> Product.Recommendations {
        let featured = try await loadFeatured()
let favorites = try await loadFavorites()
let latest = try await loadLatest()
        
        return Product.Recommendations(
            featured: featured,
            favorites: favorites,
            latest: latest
        )
    }
}

上面的實現(xiàn)確實有效——然而,即使我們的三個加載操作都是完全異步的,它們目前正在按順序執(zhí)行,一個接一個。因此,盡管我們的頂級loadRecommendations方法相對于我們應(yīng)用程序的其他代碼正在并發(fā)執(zhí)行,但實際上它還沒有利用并發(fā)來執(zhí)行其內(nèi)部操作集。

由于我們的產(chǎn)品加載方法不以任何方式相互依賴,因此實際上沒有理由按順序執(zhí)行它們,所以讓我們看看如何讓它們完全同時執(zhí)行。

關(guān)于如何做到這一點的初步想法可能是將上述代碼簡化為單個表達(dá)式,這將使我們能夠使用單個await關(guān)鍵字來等待我們的每個操作完成:

extension ProductLoader {
    func loadRecommendations() async throws -> Product.Recommendations {
        try await Product.Recommendations(
            featured: loadFeatured(),
            favorites: loadFavorites(),
            latest: loadLatest()
        )
    }
}

然而,即使我們的代碼現(xiàn)在看起來是并發(fā)的,它實際上仍會像以前一樣完全按順序執(zhí)行。

相反,我們需要利用 Swift 的async let綁定來告訴并發(fā)系統(tǒng)并行執(zhí)行我們的每個加載操作。使用該語法使我們能夠在后臺啟動異步操作,而無需我們立即等待它完成。

await如果我們在實際使用加載的數(shù)據(jù)時(即形成模型時)將其與單個關(guān)鍵字組合Recommendations,那么我們將獲得并行執(zhí)行加載操作的所有好處,而無需擔(dān)心狀態(tài)管理或數(shù)據(jù)競爭之類的事情:

extension ProductLoader {
    func loadRecommendations() async throws -> Product.Recommendations {
        async let featured = loadFeatured()
async let favorites = loadFavorites()
async let latest = loadLatest()
        
        return try await Product.Recommendations(
            featured: featured,
            favorites: favorites,
            latest: latest
        )
    }
}

很整齊!因此async let,當(dāng)我們有一組已知的、有限的任務(wù)要執(zhí)行時,它提供了一種同時運行多個操作的內(nèi)置方法。但如果不是這樣呢?

任務(wù)組

現(xiàn)在假設(shè)我們正在開發(fā)一個ImageLoader可以讓我們通過網(wǎng)絡(luò)加載圖像的工具。要從給定的 加載單個圖像URL,我們可以使用如下所示的方法:

class ImageLoader {
    ...

    func loadImage(from url: URL) async throws -> UIImage {
        ...
    }
}

為了使一次加載一系列圖像變得簡單,我們還創(chuàng)建了一個方便的 API,它接受一個 URL 數(shù)組并異步返回一個圖像字典,該字典由下載圖像的 URL 鍵控:

extension ImageLoader {
    func loadImages(from urls: [URL]) async throws -> [URL: UIImage] {
        var images = [URL: UIImage]()
        
        for url in urls {
            images[url] = try await loadImage(from: url)
        }
        
        return images
    }
}

現(xiàn)在讓我們說,就像我們ProductLoader之前的工作一樣,我們想讓上面的loadImages方法并發(fā)執(zhí)行,而不是按順序下載每個圖像(目前是這種情況,因為我們await在調(diào)用時直接使用loadImage我們的for環(huán)形)。

但是,這次我們將無法使用async let,因為我們需要執(zhí)行的任務(wù)數(shù)量在編譯時是未知的。值得慶幸的是,Swift 并發(fā)工具箱中還有一個工具可以讓我們并行執(zhí)行動態(tài)數(shù)量的任務(wù)——任務(wù)組。

要形成一個任務(wù)組,我們可以調(diào)用withTaskGroupwithThrowingTaskGroup,這取決于我們是否希望可以選擇在我們的任務(wù)中拋出錯誤。在這種情況下,我們將選擇后者,因為我們的底層loadImage方法是用throws關(guān)鍵字標(biāo)記的。

然后我們將遍歷每個 URL,就像以前一樣,只是這次我們將每個圖像加載任務(wù)添加到我們的組中,而不是直接等待它完成。相反,我們將await在添加每個任務(wù)之后單獨分組結(jié)果,這將允許我們的圖像加載操作完全并發(fā)執(zhí)行:

extension ImageLoader {
    func loadImages(from urls: [URL]) async throws -> [URL: UIImage] {
        try await withThrowingTaskGroup(of: (URL, UIImage).self) { group in
            for url in urls {
                group.addTask{
    let image = try await self.loadImage(from: url)
    return (url, image)
} 
            }
            
            var images = [URL: UIImage]()
            
            for try await (url, image) in group {
    images[url] = image
}
            
            return images
        }
    }
}

要了解有關(guān)上述for try await語法和一般異步序列的更多信息,請查看“異步序列、流和組合”

就像使用 時一樣async let,以我們的操作不會直接改變?nèi)魏螤顟B(tài)的方式編寫并發(fā)代碼的一個巨大好處是,這樣做可以讓我們完全避免任何類型的數(shù)據(jù)競爭問題,同時也不需要我們引入任何鎖定或序列化代碼混合在一起。

await因此,在可能的情況下,讓我們的每個并發(fā)操作返回一個完全獨立的結(jié)果,然后依次返回這些結(jié)果以形成我們的最終數(shù)據(jù)集,這通常是一種很好的方法。

在以后的文章中,我們將更仔細(xì)地研究避免數(shù)據(jù)競爭的其他方法(例如通過使用 Swift 的新actor類型)。

結(jié)論

重要的是要記住,僅僅因為給定的函數(shù)被標(biāo)記為async并不一定意味著它同時執(zhí)行它的工作。相反,如果這是我們想要做的,我們必須故意讓我們的任務(wù)并行運行,這只有在執(zhí)行一組可以獨立運行的操作時才有意義。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,825評論 6 546
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 99,814評論 3 429
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事?!?“怎么了?”我有些...
    開封第一講書人閱讀 178,980評論 0 384
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 64,064評論 1 319
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 72,779評論 6 414
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 56,109評論 1 330
  • 那天,我揣著相機(jī)與錄音,去河邊找鬼。 笑死,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 44,099評論 3 450
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 43,287評論 0 291
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 49,799評論 1 338
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 41,515評論 3 361
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,750評論 1 375
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,221評論 5 365
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,933評論 3 351
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,327評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,667評論 1 296
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,492評論 3 400
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 48,703評論 2 380

推薦閱讀更多精彩內(nèi)容