cancellation_token.hpp

Go to the documentation of this file.

#pragma once
 
#include <atomic>
#include <memory>
#include <stdexcept>
 
namespace bb::avm2::simulation {
 
class CancelledException : public std::runtime_error {
  public:
    CancelledException()
        : std::runtime_error("Simulation cancelled")
    {}
};
 
class CancellationToken {
  public:
    CancellationToken()
        : cancelled_(false)
    {}
 
    // Non-copyable, non-movable
    CancellationToken(const CancellationToken&) = delete;
    CancellationToken& operator=(const CancellationToken&) = delete;
    CancellationToken(CancellationToken&&) = delete;
    CancellationToken& operator=(CancellationToken&&) = delete;
 
    void cancel() { cancelled_.store(true, std::memory_order_release); }
 
    bool is_cancelled() const { return cancelled_.load(std::memory_order_acquire); }
 
    void check_and_throw() const
    {
        if (is_cancelled()) {
            throw CancelledException();
        }
    }
 
  private:
    std::atomic<bool> cancelled_;
};
 
using CancellationTokenPtr = std::shared_ptr<CancellationToken>;
 
} // namespace bb::avm2::simulation