mirror of
https://github.com/matrix-construct/construct
synced 2024-08-22 09:34:45 +02:00
202 lines
3.4 KiB
C++
202 lines
3.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_PARALLEL_H
|
|
|
|
namespace ircd::ctx
|
|
{
|
|
template<class arg> class parallel;
|
|
}
|
|
|
|
template<class arg>
|
|
struct ircd::ctx::parallel
|
|
{
|
|
using closure = std::function<void (arg &)>;
|
|
|
|
pool *p {nullptr};
|
|
vector_view<arg> a;
|
|
closure c;
|
|
dock d;
|
|
std::exception_ptr eptr;
|
|
uint64_t snd {0}; // sends to pool
|
|
uint64_t rcv {0}; // receives by worker
|
|
uint64_t fin {0}; // finished by worker
|
|
|
|
bool done() const;
|
|
bool avail() const;
|
|
void wait_done();
|
|
void wait_avail();
|
|
void rethrow_any_exception();
|
|
void receiver() noexcept;
|
|
void sender() noexcept;
|
|
|
|
public:
|
|
size_t nextpos() const;
|
|
|
|
void operator()();
|
|
void operator()(const arg &a);
|
|
|
|
parallel(pool &, const vector_view<arg> &, closure);
|
|
parallel(parallel &&) = delete;
|
|
parallel(const parallel &) = delete;
|
|
~parallel() noexcept;
|
|
};
|
|
|
|
template<class arg>
|
|
ircd::ctx::parallel<arg>::parallel(pool &p,
|
|
const vector_view<arg> &a,
|
|
closure c)
|
|
:p{&p}
|
|
,a{a}
|
|
,c{std::move(c)}
|
|
{
|
|
p.min(this->a.size());
|
|
}
|
|
|
|
template<class arg>
|
|
ircd::ctx::parallel<arg>::~parallel()
|
|
noexcept
|
|
{
|
|
const uninterruptible::nothrow ui;
|
|
wait_done();
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::operator()(const arg &a)
|
|
{
|
|
const uninterruptible ui;
|
|
rethrow_any_exception();
|
|
assert(avail());
|
|
this->a.at(nextpos()) = a;
|
|
sender();
|
|
wait_avail();
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::operator()()
|
|
{
|
|
const uninterruptible ui;
|
|
rethrow_any_exception();
|
|
assert(avail());
|
|
sender();
|
|
wait_avail();
|
|
}
|
|
|
|
template<class arg>
|
|
size_t
|
|
ircd::ctx::parallel<arg>::nextpos()
|
|
const
|
|
{
|
|
return snd % a.size();
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::sender()
|
|
noexcept
|
|
{
|
|
auto &p(*this->p);
|
|
auto func
|
|
{
|
|
std::bind(¶llel::receiver, this)
|
|
};
|
|
|
|
++snd;
|
|
assert(snd > rcv);
|
|
if(likely(p.size()))
|
|
p(std::move(func));
|
|
else
|
|
func();
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::receiver()
|
|
noexcept
|
|
{
|
|
assert(snd > rcv);
|
|
const auto pos
|
|
{
|
|
rcv++ % this->a.size()
|
|
};
|
|
|
|
if(!this->eptr) try
|
|
{
|
|
c(this->a.at(pos));
|
|
}
|
|
catch(...)
|
|
{
|
|
this->eptr = std::current_exception();
|
|
}
|
|
|
|
assert(rcv > fin);
|
|
fin++;
|
|
d.notify_one();
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::rethrow_any_exception()
|
|
{
|
|
if(likely(!this->eptr))
|
|
return;
|
|
|
|
wait_done();
|
|
const auto eptr(this->eptr);
|
|
this->eptr = {};
|
|
std::rethrow_exception(eptr);
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::wait_avail()
|
|
{
|
|
d.wait([this]
|
|
{
|
|
return avail();
|
|
});
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::wait_done()
|
|
{
|
|
d.wait([this]
|
|
{
|
|
return done();
|
|
});
|
|
}
|
|
|
|
template<class arg>
|
|
bool
|
|
ircd::ctx::parallel<arg>::avail()
|
|
const
|
|
{
|
|
assert(snd >= rcv);
|
|
assert(rcv >= fin);
|
|
assert(snd - rcv <= a.size());
|
|
assert(snd - fin <= a.size());
|
|
return snd - fin < a.size();
|
|
}
|
|
|
|
template<class arg>
|
|
bool
|
|
ircd::ctx::parallel<arg>::done()
|
|
const
|
|
{
|
|
assert(snd >= rcv);
|
|
assert(rcv >= fin);
|
|
assert(snd - rcv <= a.size());
|
|
return snd - fin == 0;
|
|
}
|