mirror of
https://github.com/matrix-construct/construct
synced 2024-10-01 05:08:59 +02:00
ircd::ctx: Add ctx::parallel device.
This commit is contained in:
parent
9f70ce1b28
commit
3d79c94bb2
2 changed files with 115 additions and 0 deletions
|
@ -96,6 +96,7 @@ namespace ircd::ctx
|
|||
#include "pool.h"
|
||||
#include "ole.h"
|
||||
#include "fault.h"
|
||||
#include "parallel.h"
|
||||
|
||||
// Exports to ircd::
|
||||
namespace ircd
|
||||
|
|
114
include/ircd/ctx/parallel.h
Normal file
114
include/ircd/ctx/parallel.h
Normal file
|
@ -0,0 +1,114 @@
|
|||
// 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;
|
||||
size_t snd {0};
|
||||
size_t rcv {0};
|
||||
size_t out {0};
|
||||
|
||||
public:
|
||||
void wait_avail();
|
||||
void wait_done();
|
||||
|
||||
void operator()(const arg &a);
|
||||
|
||||
parallel(pool &, vector_view<arg>, closure);
|
||||
~parallel() noexcept;
|
||||
};
|
||||
|
||||
template<class arg>
|
||||
ircd::ctx::parallel<arg>::parallel(pool &p,
|
||||
vector_view<arg> a,
|
||||
closure c)
|
||||
:p{&p}
|
||||
,a{std::move(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)
|
||||
{
|
||||
wait_avail();
|
||||
if(this->eptr)
|
||||
std::rethrow_exception(this->eptr);
|
||||
|
||||
auto &p(*this->p);
|
||||
this->a.at(snd++ % this->a.size()) = a;
|
||||
out++;
|
||||
p([this]()
|
||||
mutable
|
||||
{
|
||||
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>::wait_avail()
|
||||
{
|
||||
d.wait([this]
|
||||
{
|
||||
assert(snd >= rcv);
|
||||
return out < a.size();
|
||||
});
|
||||
}
|
||||
|
||||
template<class arg>
|
||||
void
|
||||
ircd::ctx::parallel<arg>::wait_done()
|
||||
{
|
||||
d.wait([this]
|
||||
{
|
||||
return !out;
|
||||
});
|
||||
}
|
Loading…
Reference in a new issue