RetryPolicy.hpp

#pragma once
 
#include "data/cassandra/Error.hpp"
#include "data/cassandra/Types.hpp"
#include "util/Retry.hpp"
#include "util/log/Logger.hpp"
 
#include <boost/asio.hpp>
#include <boost/asio/io_context.hpp>
#include <boost/asio/strand.hpp>
 
#include <chrono>
 
namespace data::cassandra::impl {

class ExponentialBackoffRetryPolicy {
    util::Logger log_{"Backend"};
 
    util::Retry retry_;
 
public:
    ExponentialBackoffRetryPolicy(boost::asio::io_context& ioc)
        : retry_(
              util::makeRetryExponentialBackoff(
                  {.initial = std::chrono::milliseconds(1), .max = std::chrono::seconds(1)},
                  boost::asio::make_strand(ioc)
              )
          )
    {
    }

    [[nodiscard]] bool
    shouldRetry([[maybe_unused]] CassandraError err)
    {
        auto const delayMs =
            std::chrono::duration_cast<std::chrono::milliseconds>(retry_.delayValue()).count();
        LOG(log_.error()) << "Cassandra write error: " << err << ", current retries "
                          << retry_.attemptNumber() << ", retrying in " << delayMs
                          << " milliseconds";
 
        return true;  // keep retrying forever
    }

    template <typename Fn>
    void
    retry(Fn&& fn)
    {
        retry_.retry(std::forward<Fn>(fn));
    }
};
 
}  // namespace data::cassandra::impl