discard.hpp

#ifndef HALCHECK_GEN_DISCARD_HPP
#define HALCHECK_GEN_DISCARD_HPP
 
#include <halcheck/gen/label.hpp>
#include <halcheck/lib/atom.hpp>
#include <halcheck/lib/effect.hpp>
#include <halcheck/lib/functional.hpp>
#include <halcheck/lib/type_traits.hpp>
 
#include <cstdint>
#include <exception>
 
namespace halcheck { namespace gen {
 
void guard(bool cond);
 
struct result_exception : std::exception {};
 
struct discard_exception : gen::result_exception {
  const char *what() const noexcept override { return "halcheck::gen::discard_exception"; }
};
 
struct succeed_effect {
  void fallback() const {}
};
 
inline void succeed() { lib::effect::invoke<succeed_effect>(); }
 
template<typename F, HALCHECK_REQUIRE(lib::is_invocable<F, lib::atom>())>
lib::invoke_result_t<F, lib::atom> retry(lib::atom id, std::uintmax_t max, F func) {
  auto _ = gen::label(id);
  for (std::uintmax_t i = 0;; i++) {
    try {
      return lib::invoke(func, i);
    } catch (const gen::discard_exception &) {
      if (max > 0 && i >= max)
        throw;
    }
  }
}
 
template<typename F, HALCHECK_REQUIRE(lib::is_invocable<F, lib::atom>())>
lib::invoke_result_t<F, lib::atom> retry(lib::atom id, F func) {
  return gen::retry(id, 100, std::move(func));
}
 
}} // namespace halcheck::gen
 
#endif