haiku/headers/os/support/Autolock.h

106 lines
1.3 KiB
C++

/*
* Copyright 2001-2009, Haiku, Inc. All Rights Reserved.
* Distributed under the terms of the MIT License.
*/
#ifndef _AUTOLOCK_H
#define _AUTOLOCK_H
#include <Locker.h>
#include <Looper.h>
class BAutolock {
public:
inline BAutolock(BLooper* looper);
inline BAutolock(BLocker* locker);
inline BAutolock(BLocker& locker);
inline ~BAutolock();
inline bool IsLocked();
inline bool Lock();
inline void Unlock();
private:
BLocker* fLocker;
BLooper* fLooper;
bool fIsLocked;
};
inline
BAutolock::BAutolock(BLooper *looper)
:
fLocker(NULL),
fLooper(looper),
fIsLocked(looper->Lock())
{
}
inline
BAutolock::BAutolock(BLocker *locker)
:
fLocker(locker),
fLooper(NULL),
fIsLocked(locker->Lock())
{
}
inline
BAutolock::BAutolock(BLocker &locker)
:
fLocker(&locker),
fLooper(NULL),
fIsLocked(locker.Lock())
{
}
inline
BAutolock::~BAutolock()
{
Unlock();
}
inline bool
BAutolock::IsLocked()
{
return fIsLocked;
}
inline bool
BAutolock::Lock()
{
if (fIsLocked)
return true;
if (fLooper != NULL)
fIsLocked = fLooper->Lock();
else
fIsLocked = fLocker->Lock();
return fIsLocked;
}
inline void
BAutolock::Unlock()
{
if (!fIsLocked)
return;
fIsLocked = false;
if (fLooper != NULL)
fLooper->Unlock();
else
fLocker->Unlock();
}
#endif // _AUTOLOCK_H