C++11各種鎖的具體使用

Mutex(互斥鎖)

什麼是互斥量(鎖)?

這樣比喻:單位上有一臺打印機(共享數據a),你要用打印機(線程1要操作數據a),同事老王也要用打印機(線程2也要操作數據a),但是打印機同一時間隻能給一個人用,此時,規定不管是誰,在用打印機之前都要向領導申請許可證(lock),用完後再向領導歸還許可證(unlock),許可證總共隻有一個,沒有許可證的人就等著在用打印機的同事用完後才能申請許可證(阻塞,線程1lock互斥量後其他線程就無法lock,隻能等線程1unlock後,其他線程才能lock)。那麼,打印機就是共享數據,訪問打印機的這段代碼就是臨界區,這個必須互斥使用的許可證就是互斥量(鎖)。

互斥量是為瞭解決數據共享過程中可能存在的訪問沖突的問題。這裡的互斥量保證瞭使用打印機這一過程不被打斷。

死鎖

多線程編程時要考慮多個線程同時訪問共享資源所造成的問題,因此可以通過加鎖解鎖來保證同一時刻隻有一個線程能訪問共享資源;使用鎖的時候要註意,不能出現死鎖的狀況;

死鎖就是多個線程爭奪共享資源導致每個線程都不能取得自己所需的全部資源,從而程序無法向下執行。

產生死鎖的四個必要條件(面試考點):

互斥(資源同一時刻隻能被一個進程使用)請求並保持(進程在請資源時,不釋放自己已經占有的資源)不剝奪(進程已經獲得的資源,在進程使用完前,不能強制剝奪)循環等待(進程間形成環狀的資源循環等待關系) 互斥量mutex

互斥量mutex就是互斥鎖,加鎖的資源支持互斥訪問

直接操作 mutex,即直接調用 mutex 的 lock / unlock 函數

#include <iostream>
#include <mutex>
#include <thread>
#include <vector>


std::mutex g_mutex;
int g_count = 0;

void Counter() {
  g_mutex.lock();

  int i = ++g_count;
  std::cout << "count: " << i << std::endl;

  // 前面代碼如有異常,unlock 就調不到瞭。
  g_mutex.unlock();
}

int main() {
  const std::size_t SIZE = 4;

  // 創建一組線程。
  std::vector<std::thread> v;
  v.reserve(SIZE);

  for (std::size_t i = 0; i < SIZE; ++i) {
    v.emplace_back(&Counter);
  }

  // 等待所有線程結束。
  for (std::thread& t : v) {
    t.join();
  }

  return 0;
}

lock_guard

使用 lock_guard 自動加鎖、解鎖。原理是 RAII,和智能指針類似。

#include <iostream>
#include <mutex>
#include <thread>
#include <vector>

std::mutex g_mutex;
int g_count = 0;

void Counter() {
  // lock_guard 在構造函數裡加鎖,在析構函數裡解鎖。
  std::lock_guard<std::mutex> lock(g_mutex);

  int i = ++g_count;
  std::cout << "count: " << i << std::endl;
}

int main() {
  const std::size_t SIZE = 4;

  std::vector<std::thread> v;
  v.reserve(SIZE);

  for (std::size_t i = 0; i < SIZE; ++i) {
    v.emplace_back(&Counter);
  }

  for (std::thread& t : v) {
    t.join();
  }

  return 0;
}

unique_lock

使用 unique_lock 自動加鎖、解鎖。
unique_lock 與 lock_guard 原理相同,但是提供瞭更多功能(比如可以結合條件變量使用)。
註意:mutex::scoped_lock 其實就是 unique_lock 的 typedef。

#include <iostream>
#include <mutex>
#include <thread>
#include <vector>

std::mutex g_mutex;
int g_count = 0;

void Counter() {
  std::unique_lock<std::mutex> lock(g_mutex);

  int i = ++g_count;
  std::cout << "count: " << i << std::endl;
}

int main() {
  const std::size_t SIZE = 4;

  std::vector<std::thread> v;
  v.reserve(SIZE);

  for (std::size_t i = 0; i < SIZE; ++i) {
    v.emplace_back(&Counter);
  }

  for (std::thread& t : v) {
    t.join();
  }

  return 0;
}

std::recursive_mutex

就像互斥鎖(mutex)一樣,遞歸互斥鎖(recursive_mutex)是可鎖定的對象,但它允許同一線程獲得對互斥鎖對象的多級所有權(多次lock)。

這允許從已經鎖定它的線程鎖定(或嘗試鎖定)互斥對象,從而獲得對互斥對象的新所有權級別:互斥對象實際上將保持對該線程的鎖定,直到調用其成員 unlock 的次數與此所有權級別的次數相同。

#include <iostream>
#include <thread>
#include <mutex> 

std::recursive_mutex mtx;           

void print_block (int n, char c) {
  mtx.lock();
  mtx.lock();
  mtx.lock();
  
  for (int i=0; i<n; ++i) { std::cout << c; }
  std::cout << '\n';
  
  mtx.unlock();
  mtx.unlock();
  mtx.unlock();
}

int main ()
{
  std::thread th1 (print_block,50,'*');
  std::thread th2 (print_block,50,'$');

  th1.join();
  th2.join();

  return 0;
}

std::timed_mutex

定時互斥鎖是一個可時間鎖定的對象,旨在通知何時關鍵代碼需要獨占訪問,就像常規互斥鎖一樣,但還支持定時嘗試鎖定請求。

lock 調用線程將鎖定timed_mutex,並在必要時進行阻塞(其行為與 mutex 完全相同)
try_lock 嘗試鎖定 timed_mutex,而不進行阻塞(其行為與互斥鎖完全相同)
try_lock_for 嘗試鎖定 timed_mutex, 最多阻塞 rel_time 時間
try_lock_until 嘗試鎖定 timed_mutex,最多阻塞到 abs_time 時間點
unlock 解鎖 timed_mutex,釋放對其的所有權(其行為與 mutex 相同)

std::recursive_timed_mutex

遞歸定時互斥鎖將 recursive_timed 和 timed_mutex 的功能結合到一個類中:它既支持通過單個線程獲
取多個鎖定級別又支持定時的 try_lock 請求。

成員函數與 timed_mutex 相同。

once_flag、call_once使用

在多線程中,有一種場景是某個任務隻需要執行一次,可以用C++11中的std::call_once函數配合std::once_flag來實現。多個線程同時調用某個函數,std::call_once可以保證多個線程對該函數隻調用一次

實現線程安全的單例模式

//h文件
#pragma once
#include <thread>
#include <iostream>
#include <mutex>
#include <memory>

class Task
{
private:
	Task();
public:
	static Task* task;
	static Task* getInstance();
	void fun();
};
//cpp文件
Task* Task::task;
Task::Task()
{
	std::cout << "構造函數" << std::endl;
}

Task* Task::getInstance()
{
	static std::once_flag flag;
	std::call_once(flag, []
	{
		task = new Task();
	});
	return task;
}

void Task::fun()
{
	std::cout << "hello world!"<< std::endl;
}

條件變量condition_variable:

需要#include<condition_variable>,該頭文件中包含瞭條件變量相關的類,其中包括std::condition_variable類

如何使用?std::condition_variable類搭配std::mutex類來使用,std::condition_variable對象(std::condition_variable cond;)的作用不是用來管理互斥量的,它的作用是用來同步線程,它的用法相當於編程中常見的flag標志(A、B兩個人約定flag=true為行動號角,默認flag為false,A不斷的檢查flag的值,隻要B將flag修改為true,A就開始行動)。

類比到std::condition_variable,A、B兩個人約定notify_one為行動號角,A就等著(調用wait(),阻塞),隻要B一調用notify_one,A就開始行動(不再阻塞)。

std::condition_variable的具體使用代碼實例可以參見文章中“生產者與消費者問題”章節。

wait(locker) :

wait函數需要傳入一個std::mutex(一般會傳入std::unique_lock對象),即上述的locker。wait函數會自動調用 locker.unlock() 釋放鎖(因為需要釋放鎖,所以要傳入mutex)並阻塞當前線程,本線程釋放鎖使得其他的線程得以繼續競爭鎖。一旦當前線程獲得notify(通常是另外某個線程調用 notify_* 喚醒瞭當前線程),wait() 函數此時再自動調用 locker.lock()上鎖。

cond.notify_one(): 隨機喚醒一個等待的線程

cond.notify_all(): 喚醒所有等待的線程

condition_variable的wait

std::condition_variable::wait 有兩個重載:

void wait( std::unique_lock<std::mutex>& lock );                    (1)	(since C++11)

template< class Predicate >
void wait( std::unique_lock<std::mutex>& lock, Predicate pred );    (2)	(since C++11)

void wait( std::unique_lockstd::mutex& lock )

先unlock之前獲得的mutex,然後阻塞當前的執行線程。把當前線程添加到等待線程列表中,該線程會持續 block 直到被 notify_all() 或 notify_one() 喚醒。被喚醒後,該thread會重新獲取mutex,獲取到mutex後執行後面的動作。

線程block時候也可能被意外或者錯誤喚醒。

template< class Predicate > void wait( std::unique_lockstd::mutex& lock, Predicate pred );

該重載設置瞭第二個參數 Predicate, 隻有當pred為false時,wait才會阻塞當前線程。蓋崇仔等同於下面:

該情況下,線程被喚醒後,先重新判斷pred的值。如果pred為false,則會釋放mutex並重新阻塞在wait。因此,該mutex必須有pred的權限。該重載消除瞭意外喚醒的影響。

#include <iostream>
#include <thread>
#include <string>
#include <mutex>
#include <condition_variable>
#include <deque>
#include <chrono>

std::deque<int> q;
std::mutex mu;
std::condition_variable condi;

void function_1()
{
	int count = 10;
	while (count > 0)
	{
		std::unique_lock<std::mutex> locker(mu);
		q.push_back(count);
		locker.unlock();
		condi.notify_one();			//通知一個等待線程激活   condi.notify_all()激活所有線程
		count--;
		std::this_thread::sleep_for(std::chrono::seconds(1));
	}
}

void function_2()
{
	int data = 100;
	while (data > 1)
	{
		std::unique_lock<std::mutex> locker(mu);
		condi.wait(locker,			//解鎖locker,並進入休眠  收到notify時又重新加鎖
			[]() { return !q.empty(); });   //如果q不為空 線程才會被激活

		data = q.front();
		q.pop_front();
		locker.unlock();

		std::cout << data << std::endl;
	}
}
int main()
{
	std::thread t1(function_1);
	std::thread t2(function_2);

	t1.join();
	t2.join();
	
	return 0;
}

std::shared_mutex

std::shared_mutex 是讀寫鎖,提供兩種訪問權限的控制:共享性(shared)和排他性(exclusive)。通過lock/try_lock獲取排他性訪問權限,通過lock_shared/try_lock_shared獲取共享性訪問權限。這樣的設置對於區分不同線程的讀寫操作特別有用。shared_mutex是c++17中引入的,使用時需要註意編譯器版本。

#include <iostream>
#include <mutex>  // For std::unique_lock
#include <shared_mutex>
#include <thread>

class ThreadSafeCounter {
 public:
  ThreadSafeCounter() = default;

  // Multiple threads/readers can read the counter's value at the same time.
  unsigned int get() const {
    std::shared_lock lock(mutex_);
    return value_;
  }

  // Only one thread/writer can increment/write the counter's value.
  void increment() {
    std::unique_lock lock(mutex_);
    value_++;
  }

  // Only one thread/writer can reset/write the counter's value.
  void reset() {
    std::unique_lock lock(mutex_);
    value_ = 0;
  }

 private:
  mutable std::shared_mutex mutex_;
  unsigned int value_ = 0;
};

int main() {
  ThreadSafeCounter counter;

  auto increment_and_print = [&counter]() {
    for (int i = 0; i < 3; i++) {
      counter.increment();
      std::cout << std::this_thread::get_id() << ' ' << counter.get() << '\n';

      // Note: Writing to std::cout actually needs to be synchronized as well
      // by another std::mutex. This has been omitted to keep the example small.
    }
  };

  std::thread thread1(increment_and_print);
  std::thread thread2(increment_and_print);

  thread1.join();
  thread2.join();
}

原子操作

所謂原子操作,就是多線程程序中“最小的且不可並行化的”操作。對於在多個線程間共享的一個資源而言,這意味著同一時刻,多個線程中有且僅有一個線程在對這個資源進行操作,即互斥訪問。提到“互斥”訪問,熟悉多線程開發的同學可能立即想到Windows平臺下使用的臨界區/CRITICAL_SECTION、互斥體/Mutex。實現互斥通常需要平臺相關的特殊指令,在C++11標準之前,這意味著需要在C/C++代碼中嵌入平臺相關的內聯匯編代碼。 平臺相關意味著:1.你必須瞭解平臺相關的編譯器擴展;2.無法跨平臺運行你的多線程程序。

多線程中需要同步的總是資源/數據,而不是代碼。因此C++11對數據進行瞭更為良好的抽象,引入”原子數據類型”/atomic類型,以達到對開發者掩蓋互斥鎖、臨界區的目的。要知道,這些臨界區、互斥鎖才是平臺相關的東西。來看下面的示例代碼。

#include<atomic>
#include<thread>
#include<iostream>

using namespace std;
std::atomic_llong total{ 0 };//原子數據類型

void func(int)
{
    for (long long i = 0; i<100000000LL; ++i)
    {
        total += i;
    }
}

int main()
{
    thread t1(func, 0);
    thread t2(func, 0);
    t1.join();
    t2.join();
    cout<<total<<endl;//9999999900000000
    return 0;
}

原子數據類型/atomic類型

atomic模板類

template <class T> struct atomic

//example
#include<atomic>

void test()
{
    std::atomic_int nThreadData; // std::atomic_int  <----> std::atomic<int>
    nThreadData = 10;
    nThreadData.store(10);
    //TODO: use nThreadData here;
}

對於內置型數據類型,C11和C++11標準中都已經提供瞭實例化原子類型,如下表所示:

在這裡插入圖片描述

atomic類型原子操作接口如下

在這裡插入圖片描述

到此這篇關於C++11各種鎖的具體使用的文章就介紹到這瞭,更多相關C++11各種鎖內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: