多線程中的信號量

信號量

DispatchSemaphoreGCD提供的API,通過設(shè)置一個閾值,允許競爭許可信號最大的線程數(shù)

  • 詳細(xì)機制
  1. wait方法判斷和獲取信號量,當(dāng)前信號量>0,開始執(zhí)行任務(wù)(線程),獲取信號量(-1)
    任務(wù)(線程)執(zhí)行結(jié)束,調(diào)用signal方法,釋放信號量(+1)

2.wait方法判斷和獲取信號量,當(dāng)前信號量<=0,線程進入阻塞狀態(tài)

import Foundation

let maxConcurrentDownloads = 3
let semaphore = DispatchSemaphore(value: maxConcurrentDownloads)

func downloadFile(url: URL, completion: @escaping (Data?, Error?) -> Void) {
    semaphore.wait()  // 獲取信號量-1
    print("Starting download for \(url)")

    let task = URLSession.shared.dataTask(with: url) { data, response, error in
        if let error = error {
            print("Download failed: \(error)")
            completion(nil, error)
        } else {
            print("Download completed for \(url)")
            completion(data, nil)
        }
        
        semaphore.signal()  // 釋放信號量+1
    }
    task.resume()//啟動任務(wù)
}

let urls = [
    URL(string: "https://example.com/file1")!,
    URL(string: "https://example.com/file2")!,
    URL(string: "https://example.com/file3")!,
    URL(string: "https://example.com/file4")!
]

for url in urls {
    downloadFile(url: url) { data, error in
        if let data = data {
            // 處理下載的數(shù)據(jù)
            print("Downloaded data: \(data.count) bytes")
        } else if let error = error {
            // 處理錯誤
            print("Error: \(error)")
        }
    }
}

// 保持主線程運行,以便所有下載任務(wù)完成
RunLoop.main.run()

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容