1 files changed, 55 insertions, 0 deletions
diff --git a/lib/base/objectlock.cpp b/lib/base/objectlock.cpp
new file mode 100644
index 0000000..fc0c7c6
--- /dev/null
+++ b/lib/base/objectlock.cpp
@@ -0,0 +1,55 @@
+/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
+
+#include "base/objectlock.hpp"
+#include <thread>
+
+using namespace icinga;
+
+#define I2MUTEX_UNLOCKED 0
+#define I2MUTEX_LOCKED 1
+
+ObjectLock::~ObjectLock()
+{
+	Unlock();
+}
+
+ObjectLock::ObjectLock(const Object::Ptr& object)
+	: ObjectLock(object.get())
+{
+}
+
+ObjectLock::ObjectLock(const Object *object)
+	: m_Object(object), m_Locked(false)
+{
+	if (m_Object)
+		Lock();
+}
+
+void ObjectLock::Lock()
+{
+	ASSERT(!m_Locked && m_Object);
+
+	m_Object->m_Mutex.lock();
+
+	m_Locked = true;
+
+#ifdef I2_DEBUG
+	if (++m_Object->m_LockCount == 1u) {
+		m_Object->m_LockOwner.store(std::this_thread::get_id());
+	}
+#endif /* I2_DEBUG */
+}
+
+void ObjectLock::Unlock()
+{
+#ifdef I2_DEBUG
+	if (m_Locked && !--m_Object->m_LockCount) {
+		m_Object->m_LockOwner.store(decltype(m_Object->m_LockOwner.load())());
+	}
+#endif /* I2_DEBUG */
+
+	if (m_Locked) {
+		m_Object->m_Mutex.unlock();
+		m_Locked = false;
+	}
+}