retry.hpp

Go to the documentation of this file.

#pragma once
 
#include "kalshi/error.hpp"
#include "kalshi/http_client.hpp"
 
#include <chrono>
#include <cstdint>
#include <memory>
#include <string_view>
 
namespace kalshi {
 
struct RetryPolicy {
    std::chrono::milliseconds initial_delay{100};
    std::chrono::milliseconds max_delay{std::chrono::seconds{10}};
    double backoff_multiplier{2.0};
    double jitter_factor{0.1};
    std::uint8_t max_attempts{3};
    bool retry_on_network_error{true};
    bool retry_on_rate_limit{true};
    bool retry_on_server_error{true};
    bool retry_writes{false};
};
 
[[nodiscard]] bool should_retry(HttpMethod method, const HttpResponse& response,
                                const RetryPolicy& policy) noexcept;
[[nodiscard]] bool should_retry(HttpMethod method, const Error& error,
                                const RetryPolicy& policy) noexcept;
 
[[nodiscard]] std::chrono::milliseconds retry_delay(std::uint8_t attempt,
                                                    const RetryPolicy& policy);
 
class RetryingTransport final : public HttpTransport {
public:
    explicit RetryingTransport(std::shared_ptr<const HttpTransport> inner, RetryPolicy policy = {});
 
    [[nodiscard]] Result<HttpResponse> request(HttpMethod method, std::string_view path,
                                               std::string_view body = {}) const override;
 
    [[nodiscard]] const RetryPolicy& policy() const noexcept { return policy_; }
 
private:
    std::shared_ptr<const HttpTransport> inner_;
    RetryPolicy policy_;
};
 
} // namespace kalshi