mirror of
https://github.com/RGBCube/serenity
synced 2025-05-31 07:38:10 +00:00
bffcb3e92a
Enable the LOCK_DEBUG functionality for these new APIs, as it looks like we want to move the whole system to use this in the not so distant future. :^)
67 lines
1.9 KiB
C++
67 lines
1.9 KiB
C++
/*
|
|
* Copyright (c) 2021, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <Kernel/Locking/ContendedResource.h>
|
|
#include <Kernel/Locking/LockLocation.h>
|
|
|
|
namespace Kernel {
|
|
|
|
template<typename T>
|
|
class ProtectedValue : private T
|
|
, public ContendedResource {
|
|
AK_MAKE_NONCOPYABLE(ProtectedValue);
|
|
AK_MAKE_NONMOVABLE(ProtectedValue);
|
|
|
|
protected:
|
|
using LockedShared = LockedResource<T const, LockMode::Shared>;
|
|
using LockedExclusive = LockedResource<T, LockMode::Exclusive>;
|
|
|
|
LockedShared lock_shared(LockLocation const& location) const { return LockedShared(this, this->ContendedResource::m_mutex, location); }
|
|
LockedExclusive lock_exclusive(LockLocation const& location) { return LockedExclusive(this, this->ContendedResource::m_mutex, location); }
|
|
|
|
public:
|
|
using T::T;
|
|
|
|
ProtectedValue() = default;
|
|
|
|
template<typename Callback>
|
|
decltype(auto) with_shared(Callback callback, LockLocation const& location = LockLocation::current()) const
|
|
{
|
|
auto lock = lock_shared(location);
|
|
return callback(*lock);
|
|
}
|
|
|
|
template<typename Callback>
|
|
decltype(auto) with_exclusive(Callback callback, LockLocation const& location = LockLocation::current())
|
|
{
|
|
auto lock = lock_exclusive(location);
|
|
return callback(*lock);
|
|
}
|
|
|
|
template<typename Callback>
|
|
void for_each_shared(Callback callback, LockLocation const& location = LockLocation::current()) const
|
|
{
|
|
with_shared([&](const auto& value) {
|
|
for (auto& item : value)
|
|
callback(item);
|
|
},
|
|
location);
|
|
}
|
|
|
|
template<typename Callback>
|
|
void for_each_exclusive(Callback callback, LockLocation const& location = LockLocation::current())
|
|
{
|
|
with_exclusive([&](auto& value) {
|
|
for (auto& item : value)
|
|
callback(item);
|
|
},
|
|
location);
|
|
}
|
|
};
|
|
|
|
}
|