0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-12-29 17:04:03 +01:00
construct/include/ircd/ctx/latch.h
2018-12-05 15:15:36 -08:00

90 lines
1.4 KiB
C++

// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
#pragma once
#define HAVE_IRCD_CTX_LATCH_H
namespace ircd::ctx
{
class latch;
}
class ircd::ctx::latch
{
mutable dock d;
size_t count {0};
public:
bool is_ready() const noexcept;
void count_down(const size_t &n = 1);
void count_down_and_wait();
void wait() const;
latch(const size_t &count);
latch() = default;
latch(latch &&);
latch(const latch &) = delete;
~latch() noexcept;
};
inline
ircd::ctx::latch::latch(const size_t &count)
:count{count}
{}
inline
ircd::ctx::latch::latch(latch &&o)
:d{std::move(o.d)}
,count{std::move(o.count)}
{
o.count = 0;
}
inline
ircd::ctx::latch::~latch()
noexcept
{
assert(d.empty());
}
inline void
ircd::ctx::latch::wait()
const
{
d.wait([this]
{
return is_ready();
});
}
inline void
ircd::ctx::latch::count_down_and_wait()
{
if(--count == 0)
d.notify_all();
else
wait();
}
inline void
ircd::ctx::latch::count_down(const size_t &n)
{
count -= n;
if(is_ready())
d.notify_all();
}
inline bool
ircd::ctx::latch::is_ready()
const noexcept
{
return count == 0;
}