互斥量可以保护某些代码只能有一个线程运行这些代码。如果有个线程使用互斥量执行某些代码,其他线程访问是会被阻塞,直到这个线程执行完这些代码,其他线程才可以执行。
一个线程在访问共享数据前,给互斥量上锁,这时其他线程再给互斥量上锁会阻塞直到这个线程给互斥量解锁。
互斥量是C++中最常用的数据保护机制,但是它也不万能的。在编写代码时,合理的组织代码来避免资源竞争非常重要。使用互斥量可能会带来其他问题,比如死锁。
创建互斥量使用mutex,给互斥量上锁使用函数lock(),给互斥量解锁使用unlock()函数。在实际应用中,不建议直接使用上锁、解锁函数,因为你必须记得每次上锁后要解锁,否则会造成死锁。在标准库中提供了lock_guard类模板,它使用了RAII(资源申请即初始化),在构造函数中上锁,在析构函数中解锁。
下面一个例子,给链表添加结点和查找链表中是否包含某个元素,在操作链表前都需要包含链表。
#include<mutex>//包含互斥量头文件 #include<list> #include<algorithm> #include<thread> #include<iostream> #include<Windows.h> //两个全局变量,用于线程间共享 std::list<int> some_list; std::mutex some_mutex; void add_to_list(int new_value) { std::lock_guard<std::mutex> guard(some_mutex);//guard为局部变量,分配在栈上,超出作用域即调用析构函数 some_list.push_back(new_value); Sleep(100);//100ms } bool list_cotains(int value_to_find) { std::lock_guard<std::mutex> guard(some_mutex); return std::find(some_list.begin(), some_list.end(), value_to_find) != some_list.end(); } void fun1() { for (int i = 0; i <50; ++i) { add_to_list(i); } } void fun2() { for (int i = 50; i < 100; ++i) { add_to_list(i); } } int main() { std::thread t1(fun1); std::thread t2(fun2); t1.join(); t2.join(); for (std::list<int>::iterator it = some_list.begin(); it != some_list.end(); ++it) std::cout << *it << std::endl; system("pause"); }
保护共享数据,不是简单的在每个函数里加上lock_guard()对象。指针或引用使用不当回破坏保护的数据。查找迷途指针或引用比较容易,只要在成员函数里不返回共享数据的指针或引用就可以。如果在进一步看,没那么简单。也可能会向你不能控制的函数传递共享数据的指针或引用,这些函数可能会把指针或引用存储起来,后面在用。这样的话,互斥量就没法保护共享数据了。
#include<mutex>//包含互斥量头文件 #include<thread> #include<iostream> class some_data { int a; public: some_data():a(10){} void do_something() { std::cout << a << std::endl; } }; class data_wrapper { private: some_data data; std::mutex m; public: template<typename Function> void process_data(Function func) { std::lock_guard<std::mutex> guard(m);//使用互斥量保护data func(data);//这个函数是可能是引用传递参数,把共享数据传到外面,危险 } }; some_data *unprotected; //把模板函数定义为引用传递 void maliciout_function(some_data& protected_data) { unprotected = &protected_data;//共享数据被传出来了,不再受互斥量保护 } data_wrapper x; int main() { x.process_data(maliciout_function); unprotected->do_something();//这里在没有互斥量保护下使用共享数据了 system("pause"); }
在保护共享数据的作用域内,不要返回共享数据的指针、引用,也不要把共享数据作为参数传递给其他人提供的函数。
一些接口,对于单线程来说是没有资源竞争的,但是多个线程使用时就未必了。例如:
std::stack<int> S; void Func() { int len = S.size(); for (int i = 0; i < len; ++i) { S.pop(); } }
在多线程环境下,先判断栈S大小,再处理。如果再处理过程中,其他线程给栈添加或删除元素,上面代码就会有问题。
单个接口是安全的,但是接口组合使用就未必了。假设栈的接口函数收保护,在某一时刻只能有一个运行。那么下面代码:
if (!S.empty())// 线程1 if (!S.empty())// 线程2 int const value = S.top(); int const value = S.top(); s.pop(); s.pop(); do_something(value); do_something(value);
这是线程1和线程2将去到相同的数据,但是两个线程都pop(),将会造成一个数据未被使用。解决这个问题的一个办法是把top()和pop()用互斥量保护起来。但是Tom Cargill指出,如果对象的复制构造函数在栈上且抛出异常,上面做法就会有问题。
假设把top和pop做成了原子操作,如果在top时,开辟空间失败,这是对象不能返回,而后面又执行了pop,这时元素在没有使用情况下被清除了。所以把top和pop做成两个接口还是有道理的。
有下面集中解决办法:
1、在pop时传递一个引用参数
//void pop(int &value);在删除时,把值赋给value int k; S.pop(k)
2、使用无异常的复制构造函数
在C++11中,右值引用使得move construtor不抛出异常(即使复制构造函数抛出异常)。一个有效的办法是:限制栈中元素的类型,仅使用不抛出异常的数据类型。
这样做是安全的,但是不是理想的。即使在编译阶段就可以借助std::is_nothrow_copy_constructible和std::is_nothrow_move_constructible类型来检测复制构造函数或move constructor是否会抛出异常,但是许多用户自定义类型有复制构造函数,却没有move constructor。这样的话,这种类型就不能存储到线程安全的栈。
3、返回pop对象的指针
返回指针而不是返回值,因为指针可以安全的复制,不会有异常。缺点是返回指针要开辟内存、保证内存不会泄露。不过可以借助shared_ptr来解决。
4、使用1和2或者1和3的组合。
下面是一个线程安全的栈
#include<exception> #include<memory> #include<mutex> #include<stack> struct empty_stack :std::exception { const char* what() const throw(); }; template<typename T> class threadsafe_stack { private: std::stack<T> data; mutable std::mutex m; public: threadsafe_stack(){} threadsafe_stack(const threadsafe_stack& other) { std::lock_guard<std::mutex> lock(other.m); data = other.data; } //不允许使用= threadsafe_stack& operator=(const threadsafe_stack& ) = delete; void push(T new_value) { std::lock_guard<std::mutex> lock(m); data.push(new_value); } std::shared_ptr<T> pop() { std::lock_guard<std::mutex> lock(m); //先检查是否为空 if (data.empty()) throw empty(); std::shared_ptr<T> const res(std::make_shared<T>(data.top())); data.top(); return res; } void pop(T& value) { std::lock_guard<std::mutex> lock(m); //先检查是否为空 if (data.empty()) throw empty(); value = data.top(); data.top(); } bool empty()const { std::lock_guard<std::mutex> lock(m); return data.emplace(); } };
原文:http://blog.csdn.net/kangroger/article/details/40023589