dev

分支 (7)

标签 (101)

管理

管理

dev

2.5.x

2.4.x

2.3.x

2.2.x

2.1.x

next

2.5.20

2.4.11client

2.4.8client

2.4.9

2.3.7

2.3.6

2.2.4client

2.2.7

2.2.6

2.1.20

2.2.3client

2.2.5

2.2.4

2.2.1client

2.2.2

2.1.15client

2.1.14client

2.1.18

2.1.13client

2.1.17

urbackup_backend
/
SharedMutex_std.h

#pragma once

#include <shared_mutex>

#include "Interface/SharedMutex.h"
#include <memory>

#if defined(_DEBUG) || (!defined(_WIN32) && !defined(NDEBUG))
#define SHARED_MUTEX_CHECK
#define SHARED_MUTEX_CHECK_P(x) ,x
#include <mutex>
#include <set>
#include <thread>
#else
#define SHARED_MUTEX_CHECK_P(x)
#endif

class SharedMutex : public ISharedMutex
{
public:
	virtual ILock* readLock();

	virtual ILock* writeLock();

#ifdef SHARED_MUTEX_CHECK
	void rmLockCheck();
#endif

private:
	std::shared_timed_mutex mutex;

#ifdef SHARED_MUTEX_CHECK
	std::mutex check_mutex;
	std::set<std::thread::id> check_threads;
#endif
};

class ReadLock : public ILock
{
public:
	ReadLock(std::shared_lock<std::shared_timed_mutex>* read_lock
		SHARED_MUTEX_CHECK_P(SharedMutex* shared_mutex));

#ifdef SHARED_MUTEX_CHECK
	~ReadLock() {
		shared_mutex->rmLockCheck();
	}
#endif

private:
	std::unique_ptr<std::shared_lock<std::shared_timed_mutex> > read_lock;

#ifdef SHARED_MUTEX_CHECK
	SharedMutex* shared_mutex;
#endif
};

class WriteLock : public ILock
{
public:
	WriteLock(std::unique_lock<std::shared_timed_mutex>* write_lock);

private:
	std::unique_ptr<std::unique_lock<std::shared_timed_mutex> > write_lock;
};