mirror of
https://github.com/matrix-construct/construct
synced 2024-07-03 01:09:05 +02:00
147 lines
2.5 KiB
C++
147 lines
2.5 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;
|
|
ushort snd {0};
|
|
ushort rcv {0};
|
|
ushort out {0};
|
|
|
|
void rethrow_any_exception();
|
|
void receiver() noexcept;
|
|
|
|
public:
|
|
void wait_avail();
|
|
void wait_done();
|
|
|
|
void operator()();
|
|
void operator()(const arg &a);
|
|
|
|
parallel(pool &, const vector_view<arg> &, closure);
|
|
~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)
|
|
{
|
|
auto &p(*this->p);
|
|
rethrow_any_exception();
|
|
p(std::bind(¶llel::receiver, this));
|
|
this->a.at(snd++) = a;
|
|
snd %= this->a.size();
|
|
out++;
|
|
wait_avail();
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::operator()()
|
|
{
|
|
rethrow_any_exception();
|
|
auto &p(*this->p);
|
|
p(std::bind(¶llel::receiver, this));
|
|
snd++;
|
|
snd %= this->a.size();
|
|
out++;
|
|
wait_avail();
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::receiver()
|
|
noexcept
|
|
{
|
|
auto &a
|
|
{
|
|
this->a.at(rcv++ % this->a.size())
|
|
};
|
|
|
|
if(!this->eptr) try
|
|
{
|
|
c(a);
|
|
}
|
|
catch(...)
|
|
{
|
|
this->eptr = std::current_exception();
|
|
}
|
|
|
|
out--;
|
|
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 out < a.size();
|
|
});
|
|
}
|
|
|
|
template<class arg>
|
|
void
|
|
ircd::ctx::parallel<arg>::wait_done()
|
|
{
|
|
d.wait([this]
|
|
{
|
|
return !out;
|
|
});
|
|
}
|