874 lines
24 KiB
C++
874 lines
24 KiB
C++
/*
|
|
* flow.h
|
|
*
|
|
* This source file is part of the FoundationDB open source project
|
|
*
|
|
* Copyright 2013-2018 Apple Inc. and the FoundationDB project authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef FLOW_FLOW_H
|
|
#define FLOW_FLOW_H
|
|
#pragma once
|
|
|
|
#pragma warning( disable: 4244 4267 ) // SOMEDAY: Carefully check for integer overflow issues (e.g. size_t to int conversions like this suppresses)
|
|
#pragma warning( disable: 4345 )
|
|
#pragma warning( error: 4239 )
|
|
|
|
#include <vector>
|
|
#include <queue>
|
|
#include <map>
|
|
#include <unordered_map>
|
|
#include <set>
|
|
#include <functional>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <algorithm>
|
|
|
|
#include "flow/Platform.h"
|
|
#include "flow/FastAlloc.h"
|
|
#include "flow/IRandom.h"
|
|
#include "flow/serialize.h"
|
|
#include "flow/Deque.h"
|
|
#include "flow/ThreadPrimitives.h"
|
|
#include "flow/network.h"
|
|
|
|
#include <boost/version.hpp>
|
|
|
|
using namespace std::rel_ops;
|
|
|
|
#define TEST( condition ) if (!(condition)); else { static TraceEvent* __test = &(TraceEvent("CodeCoverage").detail("File", __FILE__).detail("Line",__LINE__).detail("Condition", #condition)); }
|
|
|
|
/*
|
|
usage:
|
|
if (BUGGIFY) (
|
|
// code here is executed on some runs (with probability P_BUGGIFIED_SECTION_ACTIVATED),
|
|
// sometimes --
|
|
)
|
|
*/
|
|
|
|
extern double P_BUGGIFIED_SECTION_ACTIVATED, P_BUGGIFIED_SECTION_FIRES, P_EXPENSIVE_VALIDATION;
|
|
int getSBVar(std::string file, int line);
|
|
void enableBuggify(bool enabled); // Currently controls buggification and (randomized) expensive validation
|
|
bool validationIsEnabled();
|
|
|
|
#define BUGGIFY_WITH_PROB(x) (getSBVar(__FILE__, __LINE__) && g_random->random01() < (x))
|
|
#define BUGGIFY BUGGIFY_WITH_PROB(P_BUGGIFIED_SECTION_FIRES)
|
|
#define EXPENSIVE_VALIDATION (validationIsEnabled() && g_random->random01() < P_EXPENSIVE_VALIDATION)
|
|
|
|
extern Optional<uint64_t> parse_with_suffix(std::string toparse, std::string default_unit = "");
|
|
extern std::string format(const char* form, ...);
|
|
|
|
// On success, returns the number of characters written. On failure, returns a negative number.
|
|
extern int vsformat(std::string &outputString, const char* form, va_list args);
|
|
|
|
extern Standalone<StringRef> strinc(StringRef const& str);
|
|
extern StringRef strinc(StringRef const& str, Arena& arena);
|
|
extern Standalone<StringRef> addVersionStampAtEnd(StringRef const& str);
|
|
extern StringRef addVersionStampAtEnd(StringRef const& str, Arena& arena);
|
|
|
|
template <typename Iter>
|
|
StringRef concatenate( Iter b, Iter const& e, Arena& arena ) {
|
|
int rsize = 0;
|
|
Iter i = b;
|
|
while(i != e) {
|
|
rsize += i->size();
|
|
++i;
|
|
}
|
|
uint8_t* s = new (arena) uint8_t[ rsize ];
|
|
uint8_t* p = s;
|
|
while(b != e) {
|
|
memcpy(p, b->begin(),b->size());
|
|
p += b->size();
|
|
++b;
|
|
}
|
|
return StringRef(s, rsize);
|
|
}
|
|
|
|
template <typename Iter>
|
|
Standalone<StringRef> concatenate( Iter b, Iter const& e ) {
|
|
Standalone<StringRef> r;
|
|
((StringRef &)r) = concatenate(b, e, r.arena());
|
|
return r;
|
|
}
|
|
|
|
class Void {
|
|
public:
|
|
template <class Ar>
|
|
void serialize(Ar&) {}
|
|
};
|
|
|
|
class Never {};
|
|
|
|
template <class T>
|
|
class ErrorOr {
|
|
public:
|
|
ErrorOr() : error(default_error_or()) {}
|
|
ErrorOr(Error const& error) : error(error) {}
|
|
ErrorOr(const ErrorOr<T>& o) : error(o.error) {
|
|
if (present()) new (&value) T(o.get());
|
|
}
|
|
|
|
template <class U>
|
|
ErrorOr(const U& t) : error() { new (&value) T(t); }
|
|
|
|
ErrorOr(Arena& a, const ErrorOr<T>& o) : error(o.error) {
|
|
if (present()) new (&value) T(a, o.get());
|
|
}
|
|
int expectedSize() const { return present() ? get().expectedSize() : 0; }
|
|
|
|
template <class R> ErrorOr<R> castTo() const {
|
|
return map<R>([](const T& v){ return (R)v; });
|
|
}
|
|
|
|
template <class R> ErrorOr<R> map(std::function<R(T)> f) const {
|
|
if (present()) {
|
|
return ErrorOr<R>(f(get()));
|
|
}
|
|
else {
|
|
return ErrorOr<R>(error);
|
|
}
|
|
}
|
|
|
|
~ErrorOr() {
|
|
if (present()) ((T*)&value)->~T();
|
|
}
|
|
|
|
ErrorOr & operator=(ErrorOr const& o) {
|
|
if (present()) {
|
|
((T*)&value)->~T();
|
|
}
|
|
if (o.present()) {
|
|
new (&value) T(o.get());
|
|
}
|
|
error = o.error;
|
|
return *this;
|
|
}
|
|
|
|
bool present() const { return error.code() == invalid_error_code; }
|
|
T& get() {
|
|
UNSTOPPABLE_ASSERT(present());
|
|
return *(T*)&value;
|
|
}
|
|
T const& get() const {
|
|
UNSTOPPABLE_ASSERT(present());
|
|
return *(T const*)&value;
|
|
}
|
|
T orDefault(T const& default_value) const { if (present()) return get(); else return default_value; }
|
|
|
|
template <class Ar>
|
|
void serialize(Ar& ar) {
|
|
// SOMEDAY: specialize for space efficiency?
|
|
serializer(ar, error);
|
|
if (present()) {
|
|
if (Ar::isDeserializing) new (&value) T();
|
|
serializer(ar, *(T*)&value);
|
|
}
|
|
}
|
|
|
|
bool isError() const { return error.code() != invalid_error_code; }
|
|
bool isError(int code) const { return error.code() == code; }
|
|
Error getError() const { ASSERT(isError()); return error; }
|
|
|
|
private:
|
|
typename std::aligned_storage< sizeof(T), __alignof(T) >::type value;
|
|
Error error;
|
|
};
|
|
|
|
template <class T>
|
|
struct Callback {
|
|
Callback<T> *prev, *next;
|
|
|
|
virtual void fire(T const&) {}
|
|
virtual void error(Error) {}
|
|
virtual void unwait() {}
|
|
|
|
void insert(Callback<T>* into) {
|
|
// Add this (uninitialized) callback just after `into`
|
|
this->prev = into;
|
|
this->next = into->next;
|
|
into->next->prev = this;
|
|
into->next = this;
|
|
}
|
|
|
|
void insertBack(Callback<T>* into) {
|
|
// Add this (uninitialized) callback just before `into`
|
|
this->next = into;
|
|
this->prev = into->prev;
|
|
into->prev->next = this;
|
|
into->prev = this;
|
|
}
|
|
|
|
void insertChain(Callback<T>* into) {
|
|
// Combine this callback's (initialized) chain and `into`'s such that this callback is just after `into`
|
|
auto p = this->prev;
|
|
auto n = into->next;
|
|
this->prev = into;
|
|
into->next = this;
|
|
p->next = n;
|
|
n->prev = p;
|
|
}
|
|
|
|
void remove() {
|
|
// Remove this callback from the list it is in, and call unwait() on the head of that list if this was the last callback
|
|
next->prev = prev;
|
|
prev->next = next;
|
|
if (prev == next)
|
|
next->unwait();
|
|
}
|
|
|
|
int countCallbacks() {
|
|
int count = 0;
|
|
for (Callback* c = next; c != this; c = c->next)
|
|
count++;
|
|
return count;
|
|
}
|
|
};
|
|
|
|
template <class T>
|
|
struct SingleCallback {
|
|
// Used for waiting on FutureStreams, which don't support multiple callbacks
|
|
SingleCallback<T> *next;
|
|
|
|
virtual void fire(T const&) {}
|
|
virtual void error(Error) {}
|
|
virtual void unwait() {}
|
|
|
|
void insert(SingleCallback<T>* into) {
|
|
this->next = into->next;
|
|
into->next = this;
|
|
}
|
|
|
|
void remove() {
|
|
ASSERT(next->next == this);
|
|
next->next = next;
|
|
next->unwait();
|
|
}
|
|
};
|
|
|
|
template <class T>
|
|
struct SAV : private Callback<T>, FastAllocated<SAV<T>> {
|
|
int promises; // one for each promise (and one for an active actor if this is an actor)
|
|
int futures; // one for each future and one more if there are any callbacks
|
|
|
|
private:
|
|
typename std::aligned_storage< sizeof(T), __alignof(T) >::type value_storage;
|
|
public:
|
|
Error error_state;
|
|
|
|
enum { UNSET_ERROR_CODE = -3, NEVER_ERROR_CODE, SET_ERROR_CODE };
|
|
|
|
T& value() { return *(T*)&value_storage; }
|
|
|
|
SAV(int futures, int promises) : futures(futures), promises(promises), error_state(Error::fromCode(UNSET_ERROR_CODE)) {
|
|
Callback<T>::prev = Callback<T>::next = this;
|
|
}
|
|
~SAV() {
|
|
if (int16_t(error_state.code()) == SET_ERROR_CODE)
|
|
value().~T();
|
|
}
|
|
|
|
bool isSet() const { return int16_t(error_state.code()) > NEVER_ERROR_CODE; }
|
|
bool canBeSet() const { return int16_t(error_state.code()) == UNSET_ERROR_CODE; }
|
|
bool isError() const { return int16_t(error_state.code()) > SET_ERROR_CODE; }
|
|
|
|
T const& get() {
|
|
ASSERT(isSet());
|
|
if (isError()) throw error_state;
|
|
return value();
|
|
}
|
|
|
|
template <class U>
|
|
void send(U && value) {
|
|
ASSERT(canBeSet());
|
|
new (&value_storage) T(std::forward<U>(value));
|
|
this->error_state = Error::fromCode(SET_ERROR_CODE);
|
|
while (Callback<T>::next != this)
|
|
Callback<T>::next->fire(this->value());
|
|
}
|
|
|
|
void send(Never) {
|
|
ASSERT(canBeSet());
|
|
this->error_state = Error::fromCode(NEVER_ERROR_CODE);
|
|
}
|
|
|
|
void sendError(Error err) {
|
|
ASSERT(canBeSet() && int16_t(err.code()) > 0);
|
|
this->error_state = err;
|
|
while (Callback<T>::next != this)
|
|
Callback<T>::next->error(err);
|
|
}
|
|
|
|
template <class U>
|
|
void sendAndDelPromiseRef(U && value) {
|
|
ASSERT(canBeSet());
|
|
if (promises == 1 && !futures) {
|
|
// No one is left to receive the value, so we can just die
|
|
destroy();
|
|
return;
|
|
}
|
|
|
|
new (&value_storage) T(std::forward<U>(value));
|
|
finishSendAndDelPromiseRef();
|
|
}
|
|
|
|
void finishSendAndDelPromiseRef() {
|
|
// Call only after value_storage has already been initialized!
|
|
this->error_state = Error::fromCode(SET_ERROR_CODE);
|
|
while (Callback<T>::next != this)
|
|
Callback<T>::next->fire(this->value());
|
|
|
|
if (!--promises && !futures)
|
|
destroy();
|
|
}
|
|
|
|
void sendAndDelPromiseRef(Never) {
|
|
ASSERT(canBeSet());
|
|
this->error_state = Error::fromCode(NEVER_ERROR_CODE);
|
|
if (!--promises && !futures)
|
|
destroy();
|
|
}
|
|
|
|
void sendErrorAndDelPromiseRef(Error err) {
|
|
ASSERT(canBeSet() && int16_t(err.code()) > 0);
|
|
if (promises == 1 && !futures) {
|
|
// No one is left to receive the value, so we can just die
|
|
destroy();
|
|
return;
|
|
}
|
|
|
|
this->error_state = err;
|
|
while (Callback<T>::next != this)
|
|
Callback<T>::next->error(err);
|
|
|
|
if (!--promises && !futures)
|
|
destroy();
|
|
}
|
|
|
|
void addPromiseRef() { promises++; }
|
|
void addFutureRef() { futures++; }
|
|
|
|
void delPromiseRef() {
|
|
if (promises == 1) {
|
|
if (futures && canBeSet()) {
|
|
sendError(broken_promise());
|
|
ASSERT(promises == 1); // Once there is only one promise, there is no one else with the right to change the promise reference count
|
|
}
|
|
promises = 0;
|
|
if (!futures)
|
|
destroy();
|
|
}
|
|
else
|
|
--promises;
|
|
}
|
|
void delFutureRef() {
|
|
if (!--futures) {
|
|
if (promises)
|
|
cancel();
|
|
else
|
|
destroy();
|
|
}
|
|
}
|
|
|
|
int getFutureReferenceCount() const { return futures; }
|
|
int getPromiseReferenceCount() const { return promises; }
|
|
|
|
virtual void destroy() { delete this; }
|
|
virtual void cancel() {}
|
|
|
|
void addCallbackAndDelFutureRef(Callback<T>* cb) {
|
|
// We are always *logically* dropping one future reference from this, but if we are adding a first callback
|
|
// we also need to add one (since futures is defined as being +1 if there are any callbacks), so net nothing
|
|
if (Callback<T>::next != this)
|
|
delFutureRef();
|
|
cb->insert(this);
|
|
}
|
|
|
|
void addYieldedCallbackAndDelFutureRef(Callback<T>* cb) {
|
|
// Same contract as addCallbackAndDelFutureRef, except that the callback is placed at the end of the callback chain rather than at the beginning
|
|
if (Callback<T>::next != this)
|
|
delFutureRef();
|
|
cb->insertBack(this);
|
|
}
|
|
|
|
void addCallbackChainAndDelFutureRef(Callback<T>* cb) {
|
|
if (Callback<T>::next != this)
|
|
delFutureRef();
|
|
cb->insertChain(this);
|
|
}
|
|
|
|
virtual void unwait() {
|
|
delFutureRef();
|
|
}
|
|
virtual void fire() { ASSERT(false); }
|
|
};
|
|
|
|
template <class T>
|
|
struct NotifiedQueue : private SingleCallback<T>, FastAllocated<NotifiedQueue<T>> {
|
|
int promises; // one for each promise (and one for an active actor if this is an actor)
|
|
int futures; // one for each future and one more if there are any callbacks
|
|
|
|
// Invariant: SingleCallback<T>::next==this || (queue.empty() && !error.isValid())
|
|
std::queue<T, Deque<T>> queue;
|
|
Error error;
|
|
|
|
NotifiedQueue(int futures, int promises) : futures(futures), promises(promises) {
|
|
SingleCallback<T>::next = this;
|
|
}
|
|
|
|
bool isReady() const { return !queue.empty() || error.isValid(); }
|
|
bool isError() const { return queue.empty() && error.isValid(); } // the *next* thing queued is an error
|
|
|
|
T pop() {
|
|
if (queue.empty()) {
|
|
if (error.isValid()) throw error;
|
|
throw internal_error();
|
|
}
|
|
auto copy = queue.front();
|
|
queue.pop();
|
|
return copy;
|
|
}
|
|
|
|
template <class U>
|
|
void send(U && value) {
|
|
if (error.isValid()) return;
|
|
|
|
if (SingleCallback<T>::next != this) {
|
|
SingleCallback<T>::next->fire(std::forward<U>(value));
|
|
}
|
|
else {
|
|
queue.emplace(std::forward<U>(value));
|
|
}
|
|
}
|
|
|
|
void sendError(Error err) {
|
|
if (error.isValid()) return;
|
|
|
|
this->error = err;
|
|
if (SingleCallback<T>::next != this)
|
|
SingleCallback<T>::next->error(err);
|
|
}
|
|
|
|
void addPromiseRef() { promises++; }
|
|
void addFutureRef() { futures++; }
|
|
|
|
void delPromiseRef() {
|
|
if (!--promises) {
|
|
if (futures) {
|
|
sendError(broken_promise());
|
|
}
|
|
else
|
|
destroy();
|
|
}
|
|
}
|
|
void delFutureRef() {
|
|
if (!--futures) {
|
|
if (promises)
|
|
cancel();
|
|
else
|
|
destroy();
|
|
}
|
|
}
|
|
|
|
int getFutureReferenceCount() const { return futures; }
|
|
int getPromiseReferenceCount() const { return promises; }
|
|
|
|
virtual void destroy() { delete this; }
|
|
virtual void cancel() {}
|
|
|
|
void addCallbackAndDelFutureRef(SingleCallback<T>* cb) {
|
|
ASSERT(SingleCallback<T>::next == this);
|
|
cb->insert(this);
|
|
}
|
|
virtual void unwait() {
|
|
delFutureRef();
|
|
}
|
|
virtual void fire() { ASSERT(false); }
|
|
};
|
|
|
|
|
|
template <class T>
|
|
class Promise;
|
|
|
|
template <class T>
|
|
class Future
|
|
{
|
|
public:
|
|
T const& get() const { return sav->get(); }
|
|
T getValue() const { return get(); }
|
|
|
|
bool isValid() const {
|
|
return sav != 0;
|
|
}
|
|
bool isReady() const {
|
|
return sav->isSet();
|
|
}
|
|
bool isError() const {
|
|
return sav->isError();
|
|
}
|
|
Error& getError() const {
|
|
ASSERT(isError());
|
|
return sav->error_state;
|
|
}
|
|
|
|
Future() : sav(0) {}
|
|
Future(const Future<T>& rhs) : sav(rhs.sav) {
|
|
if (sav) sav->addFutureRef();
|
|
//if (sav->endpoint.isValid()) cout << "Future copied for " << sav->endpoint.key << endl;
|
|
}
|
|
Future(Future<T>&& rhs) BOOST_NOEXCEPT : sav(rhs.sav) {
|
|
rhs.sav = 0;
|
|
//if (sav->endpoint.isValid()) cout << "Future moved for " << sav->endpoint.key << endl;
|
|
}
|
|
Future(const T& presentValue)
|
|
: sav(new SAV<T>(1, 0))
|
|
{
|
|
sav->send(presentValue);
|
|
}
|
|
Future(Never)
|
|
: sav(new SAV<T>(1, 0))
|
|
{
|
|
sav->send(Never());
|
|
}
|
|
Future(const Error& error)
|
|
: sav(new SAV<T>(1, 0))
|
|
{
|
|
sav->sendError(error);
|
|
}
|
|
|
|
#ifndef NO_INTELLISENSE
|
|
template<class U>
|
|
Future(const U&, typename std::enable_if<std::is_assignable<T, U>::value, int*>::type = 0) {}
|
|
#endif
|
|
|
|
~Future() {
|
|
//if (sav && sav->endpoint.isValid()) cout << "Future destroyed for " << sav->endpoint.key << endl;
|
|
if (sav) sav->delFutureRef();
|
|
}
|
|
void operator=(const Future<T>& rhs) {
|
|
if (rhs.sav) rhs.sav->addFutureRef();
|
|
if (sav) sav->delFutureRef();
|
|
sav = rhs.sav;
|
|
}
|
|
void operator=(Future<T>&& rhs) BOOST_NOEXCEPT {
|
|
if (sav != rhs.sav) {
|
|
if (sav) sav->delFutureRef();
|
|
sav = rhs.sav;
|
|
rhs.sav = 0;
|
|
}
|
|
}
|
|
bool operator == (const Future& rhs) { return rhs.sav == sav; }
|
|
bool operator != (const Future& rhs) { return rhs.sav != sav; }
|
|
|
|
void cancel() {
|
|
if (sav) sav->cancel();
|
|
}
|
|
|
|
void addCallbackAndClear(Callback<T>* cb) {
|
|
sav->addCallbackAndDelFutureRef(cb);
|
|
sav = 0;
|
|
}
|
|
|
|
void addYieldedCallbackAndClear(Callback<T>* cb) {
|
|
sav->addYieldedCallbackAndDelFutureRef(cb);
|
|
sav = 0;
|
|
}
|
|
|
|
void addCallbackChainAndClear(Callback<T>* cb) {
|
|
sav->addCallbackChainAndDelFutureRef(cb);
|
|
sav = 0;
|
|
}
|
|
|
|
int getFutureReferenceCount() const { return sav->getFutureReferenceCount(); }
|
|
int getPromiseReferenceCount() const { return sav->getPromiseReferenceCount(); }
|
|
|
|
explicit Future(SAV<T> * sav) : sav(sav) {
|
|
//if (sav->endpoint.isValid()) cout << "Future created for " << sav->endpoint.key << endl;
|
|
}
|
|
|
|
private:
|
|
SAV<T>* sav;
|
|
friend class Promise<T>;
|
|
};
|
|
|
|
// This class is used by the flow compiler when generating code around wait statements to avoid confusing situations
|
|
// regarding Futures.
|
|
//
|
|
// For example, the following is legal with Future but not with StrictFuture:
|
|
//
|
|
// Future<T> x = ...
|
|
// T result = wait(x); // This is the correct code
|
|
// Future<T> result = wait(x); // This is legal if wait() generates Futures, but it's probably wrong. It's a compilation error if wait() generates StrictFutures.
|
|
template <class T>
|
|
class StrictFuture : public Future<T> {
|
|
public:
|
|
inline StrictFuture(Future<T> const& f) : Future<T>(f) {}
|
|
inline StrictFuture(Never n) : Future<T>(n) {}
|
|
private:
|
|
StrictFuture(T t) {}
|
|
StrictFuture(Error e) {}
|
|
};
|
|
|
|
template <class T>
|
|
class Promise sealed
|
|
{
|
|
public:
|
|
template <class U>
|
|
void send(U && value) const {
|
|
sav->send(std::forward<U>(value));
|
|
}
|
|
template <class E>
|
|
void sendError(const E& exc) const { sav->sendError(exc); }
|
|
|
|
Future<T> getFuture() const { sav->addFutureRef(); return Future<T>(sav); }
|
|
bool isSet() { return sav->isSet(); }
|
|
bool canBeSet() { return sav->canBeSet(); }
|
|
bool isValid() const { return sav != NULL; }
|
|
Promise() : sav(new SAV<T>(0, 1)) {}
|
|
Promise(const Promise& rhs) : sav(rhs.sav) { sav->addPromiseRef(); }
|
|
Promise(Promise&& rhs) BOOST_NOEXCEPT : sav(rhs.sav) { rhs.sav = 0; }
|
|
~Promise() { if (sav) sav->delPromiseRef(); }
|
|
|
|
void operator=(const Promise& rhs) {
|
|
if (rhs.sav) rhs.sav->addPromiseRef();
|
|
if (sav) sav->delPromiseRef();
|
|
sav = rhs.sav;
|
|
}
|
|
void operator=(Promise && rhs) BOOST_NOEXCEPT {
|
|
if (sav != rhs.sav) {
|
|
if (sav) sav->delPromiseRef();
|
|
sav = rhs.sav;
|
|
rhs.sav = 0;
|
|
}
|
|
}
|
|
void reset() {
|
|
*this = Promise<T>();
|
|
}
|
|
void swap(Promise& other) {
|
|
std::swap(sav, other.sav);
|
|
}
|
|
|
|
// Beware, these operations are very unsafe
|
|
SAV<T>* extractRawPointer() { auto ptr = sav; sav = NULL; return ptr; }
|
|
explicit Promise<T>(SAV<T>* ptr) : sav(ptr) {}
|
|
|
|
int getFutureReferenceCount() const { return sav->getFutureReferenceCount(); }
|
|
int getPromiseReferenceCount() const { return sav->getPromiseReferenceCount(); }
|
|
|
|
private:
|
|
SAV<T> *sav;
|
|
};
|
|
|
|
|
|
template <class T>
|
|
class FutureStream {
|
|
public:
|
|
bool isValid() const {
|
|
return queue != 0;
|
|
}
|
|
bool isReady() const {
|
|
return queue->isReady();
|
|
}
|
|
bool isError() const {
|
|
// This means that the next thing to be popped is an error - it will be false if there is an error in the stream but some actual data first
|
|
return queue->isError();
|
|
}
|
|
void addCallbackAndClear(SingleCallback<T>* cb) {
|
|
queue->addCallbackAndDelFutureRef(cb);
|
|
queue = 0;
|
|
}
|
|
FutureStream() : queue(NULL) {}
|
|
FutureStream(const FutureStream& rhs) : queue(rhs.queue) { queue->addFutureRef(); }
|
|
FutureStream(FutureStream&& rhs) BOOST_NOEXCEPT : queue(rhs.queue) { rhs.queue = 0; }
|
|
~FutureStream() { if (queue) queue->delFutureRef(); }
|
|
void operator=(const FutureStream& rhs) {
|
|
rhs.queue->addFutureRef();
|
|
if (queue) queue->delFutureRef();
|
|
queue = rhs.queue;
|
|
}
|
|
void operator=(FutureStream&& rhs) BOOST_NOEXCEPT {
|
|
if (rhs.queue != queue) {
|
|
if (queue) queue->delFutureRef();
|
|
queue = rhs.queue;
|
|
rhs.queue = 0;
|
|
}
|
|
}
|
|
bool operator == (const FutureStream& rhs) { return rhs.queue == queue; }
|
|
bool operator != (const FutureStream& rhs) { return rhs.queue != queue; }
|
|
|
|
T pop() {
|
|
return queue->pop();
|
|
}
|
|
Error getError() {
|
|
ASSERT(queue->isError());
|
|
return queue->error;
|
|
}
|
|
|
|
explicit FutureStream(NotifiedQueue<T>* queue) : queue(queue) {}
|
|
|
|
private:
|
|
NotifiedQueue<T>* queue;
|
|
};
|
|
|
|
template <class Request>
|
|
decltype(fake<Request>().reply) const& getReplyPromise(Request const& r) { return r.reply; }
|
|
|
|
|
|
|
|
// Neither of these implementations of REPLY_TYPE() works on both MSVC and g++, so...
|
|
#ifdef __GNUG__
|
|
#define REPLY_TYPE(RequestType) decltype( getReplyPromise( fake<RequestType>() ).getFuture().getValue() )
|
|
//#define REPLY_TYPE(RequestType) decltype( getReplyFuture( fake<RequestType>() ).getValue() )
|
|
#else
|
|
template <class T>
|
|
struct ReplyType {
|
|
// Doing this calculation directly in the return value declaration for PromiseStream<T>::getReply()
|
|
// breaks IntelliSense in VS2010; this is a workaround.
|
|
typedef decltype(fake<T>().reply.getFuture().getValue()) Type;
|
|
};
|
|
template <class T> class ReplyPromise;
|
|
template <class T>
|
|
struct ReplyType<ReplyPromise<T>> {
|
|
typedef T Type;
|
|
};
|
|
#define REPLY_TYPE(RequestType) typename ReplyType<RequestType>::Type
|
|
#endif
|
|
|
|
|
|
|
|
|
|
template <class T>
|
|
class PromiseStream {
|
|
public:
|
|
// stream.send( request )
|
|
// Unreliable at most once delivery: Delivers request unless there is a connection failure (zero or one times)
|
|
|
|
void send(const T& value) const {
|
|
queue->send(value);
|
|
}
|
|
void sendError(const Error& error) const {
|
|
queue->sendError(error);
|
|
}
|
|
|
|
// stream.getReply( request )
|
|
// Reliable at least once delivery: Eventually delivers request at least once and returns one of the replies if communication is possible. Might deliver request
|
|
// more than once.
|
|
// If a reply is returned, request was or will be delivered one or more times.
|
|
// If cancelled, request was or will be delivered zero or more times.
|
|
template <class X>
|
|
Future<REPLY_TYPE(X)> getReply(const X& value) const {
|
|
send(value);
|
|
return getReplyPromise(value).getFuture();
|
|
}
|
|
template <class X>
|
|
Future<REPLY_TYPE(X)> getReply(const X& value, int taskID) const {
|
|
setReplyPriority(value, taskID);
|
|
return getReplyPromise(value).getFuture();
|
|
}
|
|
|
|
template <class X>
|
|
Future<X> getReply() const {
|
|
return getReply(Promise<X>());
|
|
}
|
|
template <class X>
|
|
Future<X> getReplyWithTaskID(int taskID) const {
|
|
Promise<X> reply;
|
|
reply.getEndpoint(taskID);
|
|
return getReply(reply);
|
|
}
|
|
|
|
FutureStream<T> getFuture() const { queue->addFutureRef(); return FutureStream<T>(queue); }
|
|
PromiseStream() : queue(new NotifiedQueue<T>(0, 1)) {}
|
|
PromiseStream(const PromiseStream& rhs) : queue(rhs.queue) { queue->addPromiseRef(); }
|
|
PromiseStream(PromiseStream&& rhs) BOOST_NOEXCEPT : queue(rhs.queue) { rhs.queue = 0; }
|
|
void operator=(const PromiseStream& rhs) {
|
|
rhs.queue->addPromiseRef();
|
|
if (queue) queue->delPromiseRef();
|
|
queue = rhs.queue;
|
|
}
|
|
void operator=(PromiseStream&& rhs) BOOST_NOEXCEPT {
|
|
if (queue != rhs.queue) {
|
|
if (queue) queue->delPromiseRef();
|
|
queue = rhs.queue;
|
|
rhs.queue = 0;
|
|
}
|
|
}
|
|
~PromiseStream() {
|
|
if (queue)
|
|
queue->delPromiseRef();
|
|
//queue = (NotifiedQueue<T>*)0xdeadbeef;
|
|
}
|
|
|
|
bool operator == (const PromiseStream<T>& rhs) const { return queue == rhs.queue; }
|
|
bool isEmpty() const { return !queue->isReady(); }
|
|
|
|
private:
|
|
NotifiedQueue<T>* queue;
|
|
};
|
|
|
|
|
|
//extern int actorCount;
|
|
|
|
template <class T>
|
|
static inline void destruct(T& t) {
|
|
t.~T();
|
|
}
|
|
|
|
template <class ReturnValue>
|
|
struct Actor : SAV<ReturnValue> {
|
|
int8_t actor_wait_state; // -1 means actor is cancelled; 0 means actor is not waiting; 1-N mean waiting in callback group #
|
|
|
|
Actor() : SAV<ReturnValue>(1, 1), actor_wait_state(0) { /*++actorCount;*/ }
|
|
//~Actor() { --actorCount; }
|
|
};
|
|
|
|
template <>
|
|
struct Actor<void> {
|
|
// This specialization is for a void actor (one not returning a future, hence also uncancellable)
|
|
|
|
int8_t actor_wait_state; // 0 means actor is not waiting; 1-N mean waiting in callback group #
|
|
|
|
Actor() : actor_wait_state(0) { /*++actorCount;*/ }
|
|
//~Actor() { --actorCount; }
|
|
};
|
|
|
|
template <class ActorType, int CallbackNumber, class ValueType>
|
|
struct ActorCallback : Callback<ValueType> {
|
|
virtual void fire(ValueType const& value) {
|
|
static_cast<ActorType*>(this)->a_callback_fire(this, value);
|
|
}
|
|
virtual void error(Error e) {
|
|
static_cast<ActorType*>(this)->a_callback_error(this, e);
|
|
}
|
|
};
|
|
|
|
template <class ActorType, int CallbackNumber, class ValueType>
|
|
struct ActorSingleCallback : SingleCallback<ValueType> {
|
|
virtual void fire(ValueType const& value) {
|
|
static_cast<ActorType*>(this)->a_callback_fire(this, value);
|
|
}
|
|
virtual void error(Error e) {
|
|
static_cast<ActorType*>(this)->a_callback_error(this, e);
|
|
}
|
|
};
|
|
inline double now() { return g_network->now(); }
|
|
inline Future<Void> delay(double seconds, int taskID = TaskDefaultDelay) { return g_network->delay(seconds, taskID); }
|
|
inline Future<Void> delayUntil(double time, int taskID = TaskDefaultDelay) { return g_network->delay(std::max(0.0, time - g_network->now()), taskID); }
|
|
inline Future<Void> delayJittered(double seconds, int taskID = TaskDefaultDelay) { return g_network->delay(seconds*(FLOW_KNOBS->DELAY_JITTER_OFFSET + FLOW_KNOBS->DELAY_JITTER_RANGE*g_random->random01()), taskID); }
|
|
inline Future<Void> yield(int taskID = TaskDefaultYield) { return g_network->yield(taskID); }
|
|
inline bool check_yield(int taskID = TaskDefaultYield) { return g_network->check_yield(taskID); }
|
|
#include "flow/genericactors.actor.h"
|
|
#endif
|