/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
 * Copyright 2022-Present Couchbase, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of the License at
 *
 *     https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under
 * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
 * ANY KIND, either express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

#include "retry_orchestrator.hxx"

#include "core/logger/logger.hxx"
#include "couchbase/best_effort_retry_strategy.hxx"
#include "mcbp/queue_request.hxx"

#include <couchbase/fmt/retry_reason.hxx>

#include <fmt/chrono.h>

namespace couchbase::core
{
auto
retry_orchestrator::should_retry(std::shared_ptr<mcbp::queue_request> request, retry_reason reason) -> retry_action
{
    if (always_retry(reason)) {
        auto duration = controlled_backoff(request->retry_attempts());
        CB_LOG_DEBUG("will retry request. backoff={}, operation_id={}, reason={}", duration, request->identifier(), reason);
        request->record_retry_attempt(reason);
        return retry_action{ duration };
    }

    auto strategy = request->retry_strategy();
    if (strategy == nullptr) {
        return retry_action::do_not_retry();
    }

    auto action = strategy->retry_after(*request, reason);
    if (!action.need_to_retry()) {
        CB_LOG_DEBUG("will not retry request. operation_id={}, reason={}", request->identifier(), reason);
        return retry_action::do_not_retry();
    }
    CB_LOG_DEBUG("will retry request. backoff={}, operation_id={}, reason={}", action.duration(), request->identifier(), reason);
    request->record_retry_attempt(reason);
    return action;
}
} // namespace couchbase::core
