利用boost的shared_ptr可以實現“讀不阻塞寫”。過程中每次讀取不需要複製資料，而只需要在要寫的時候複製一份資料：

點選(此處)摺疊或開啟

#include <iostream>
#include <cstdlib>
#include <map>
#include <vector>
#include <boost/smart_ptr.hpp>
#include <boost/make_shared.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/bind.hpp>
#include <boost/function.hpp>
#include <boost/noncopyable.hpp>
#include <pthread.h>
#include <unistd.h>
using namespace std;
class mutex_lock: public boost::noncopyable
{
public:
mutex_lock(){pthread_mutex_init(&_mutex, NULL);}
~mutex_lock(){pthread_mutex_destroy(&_mutex);}
void lock() {pthread_mutex_lock(&_mutex);}
void unlock() {pthread_mutex_unlock(&_mutex);}
private:
pthread_mutex_t _mutex;
};
class mutex_locker: public boost::noncopyable
{
public:
explicit mutex_locker(mutex_lock& lk): _lk(lk){_lk.lock();}
~mutex_locker(){_lk.unlock();}
private:
mutex_lock& _lk;
};
// 負責都和寫的物件
class rwobj
{
public:
rwobj(){sp_data = boost::make_shared<vector<int> >();}
void read_data()
{
boost::shared_ptr<vector<int> > sp;
{
cout << "--------------------- read wait" << endl;
mutex_locker lk(l); // 臨界區足夠小，適合用mutex。
sp = sp_data; // 利用區域性的shared_ptr來增加資料的引用計數，告訴寫執行緒不要修改這份資料
cout << "read sleep 1" << endl;
sleep(1); // 為了阻塞其它執行緒
}
cout << "read size " << sp->size() << endl;
cout << "read sleep 2" << endl;
sleep(2);
}
void write_data(int i)
{
{
cout << "------------------------------ write wait" << endl;
mutex_locker lk(l); // 寫執行緒的臨界區比較大
if(!sp_data.unique())
{
// 如果sp_data引用計數大於1，說明有其他執行緒在讀（透過read_data()提升了引用計數）。
// 此時將資料複製一份，再利用reset或者swap讓sp_data指向新資料，老資料讓讀執行緒繼續讀。
// 這個寫執行緒現在獨佔了sp_data，而之前那份資料在所有讀執行緒讀完之後，引用計數會變成0，被最後一個讀執行緒自動析構。
sp_data.reset(new vector<int>(*sp_data));
cout << "-------------------------------- copy on write" << endl;
}
sp_data->push_back(i);
}
cout << "sleep write" << endl;
sleep(1);
}
private:
boost::shared_ptr<vector<int> > sp_data;
mutex_lock l;
};
rwobj obj;
// 讀執行緒
void* read_thread(void* arg)
{
while(1)
{
obj.read_data();
}
}
寫執行緒
void* write_thread(void* arg)
{
int i = 1;
while(1)
{
obj.write_data(i++);
}
}
int main()
{
pthread_t thread1,thread2, thread3;
pthread_create(&thread1, NULL, &read_thread, NULL );
pthread_create(&thread2, NULL, &read_thread, NULL );
pthread_create(&thread3, NULL, &write_thread, NULL );
pthread_join(thread1,NULL);
pthread_join(thread2,NULL);
pthread_join(thread3,NULL);
return 0;
}

shared_ptr實現多執行緒讀寫copy-on-write

相關文章