icinga2/lib/base/objectlock.hpp

99 lines
2.7 KiB
C++

/******************************************************************************
* Icinga 2 *
* Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org) *
* *
* This program is free software; you can redistribute it and/or *
* modify it under the terms of the GNU General Public License *
* as published by the Free Software Foundation; either version 2 *
* of the License, or (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the Free Software Foundation *
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
******************************************************************************/
#ifndef OBJECTLOCK_H
#define OBJECTLOCK_H
#include "base/object.hpp"
namespace icinga
{
/**
* A scoped lock for Objects.
*/
struct I2_BASE_API ObjectLock {
public:
inline ObjectLock(void)
: m_Object(NULL), m_Locked(false)
{ }
inline ~ObjectLock(void)
{
Unlock();
}
inline ObjectLock(const Object::Ptr& object)
: m_Object(object.get()), m_Locked(false)
{
if (m_Object)
Lock();
}
inline ObjectLock(const Object *object)
: m_Object(object), m_Locked(false)
{
if (m_Object)
Lock();
}
inline void Lock(void)
{
ASSERT(!m_Locked && m_Object != NULL);
ASSERT(!m_Object->OwnsLock());
m_Object->m_Mutex.Lock();
m_Locked = true;
#ifdef I2_DEBUG
# ifdef _WIN32
InterlockedExchange(&m_Object->m_LockOwner, GetCurrentThreadId());
# else /* _WIN32 */
__sync_lock_test_and_set(&m_Object->m_LockOwner, pthread_self());
# endif /* _WIN32 */
#endif /* I2_DEBUG */
}
inline void Unlock(void)
{
#ifdef I2_DEBUG
if (m_Locked) {
# ifdef _WIN32
InterlockedExchange(&m_Object->m_LockOwner, 0);
# else /* _WIN32 */
__sync_lock_release(&m_Object->m_LockOwner);
# endif /* _WIN32 */
}
#endif /* I2_DEBUG */
if (m_Locked) {
m_Object->m_Mutex.Unlock();
m_Locked = false;
}
}
private:
const Object *m_Object;
bool m_Locked;
};
}
#endif /* OBJECTLOCK_H */