state_machine.h

Go to the documentation of this file.

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the Apache 2.0 License.
#pragma once
 
#include "ccf/ds/logger.h"
 
#include <atomic>
#include <string>
 
namespace ds
{
  template <typename T>
  class StateMachine
  {
    const std::string label;
    std::atomic<T> state;
 
  public:
    StateMachine(const std::string& label_, T state_) :
      label(label_),
      state(state_)
    {}
 
    void expect(T state_) const
    {
      auto state_snapshot = state.load();
      if (state_ != state_snapshot)
      {
        throw std::logic_error(fmt::format(
          "[{}] State is {}, but expected {}", label, state_snapshot, state_));
      }
    }
 
    bool check(T state_) const
    {
      return state_ == state.load();
    }
 
    T value() const
    {
      return state.load();
    }
 
    void advance(T state_)
    {
      LOG_DEBUG_FMT("[{}] Advancing to state {}", label, state_);
      state.store(state_);
    }
  };
}