health_check_builder.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include <chrono>
#include <functional>
#include <memory>
#include <string>
 
#include "health_check.h"
 
namespace kcenon::common::interfaces {
 
class health_check_builder {
public:
    using check_fn_type = std::function<health_check_result()>;
 
    health_check_builder() = default;
 
    health_check_builder& name(std::string value) {
        name_ = std::move(value);
        return *this;
    }
 
    health_check_builder& type(health_check_type value) {
        type_ = value;
        return *this;
    }
 
    health_check_builder& with_check(check_fn_type fn) {
        check_fn_ = std::move(fn);
        return *this;
    }
 
    health_check_builder& critical(bool value) {
        critical_ = value;
        return *this;
    }
 
    health_check_builder& timeout(std::chrono::milliseconds value) {
        timeout_ = value;
        return *this;
    }
 
    [[nodiscard]] Result<std::shared_ptr<health_check>> build() const {
        if (name_.empty()) {
            return {error_info{1, "Health check name is required", "health_check_builder"}};
        }
 
        if (!check_fn_) {
            return {error_info{2, "Check function is required", "health_check_builder"}};
        }
 
        std::shared_ptr<health_check> check = std::make_shared<lambda_health_check>(
            name_, type_, check_fn_, critical_, timeout_);
        return ok(std::move(check));
    }
 
    [[nodiscard]] std::shared_ptr<health_check> build_unsafe() const {
        return std::make_shared<lambda_health_check>(
            name_, type_, check_fn_, critical_, timeout_);
    }
 
    health_check_builder& reset() {
        name_.clear();
        type_ = health_check_type::custom;
        check_fn_ = nullptr;
        critical_ = true;
        timeout_ = std::chrono::milliseconds{5000};
        return *this;
    }
 
private:
    std::string name_;
    health_check_type type_ = health_check_type::custom;
    check_fn_type check_fn_;
    bool critical_ = true;
    std::chrono::milliseconds timeout_{5000};
};
 
}  // namespace kcenon::common::interfaces