data_source_builder.hpp

#pragma once
 
#include <launchdarkly/config/shared/built/data_source_config.hpp>
#include <launchdarkly/config/shared/defaults.hpp>
#include <launchdarkly/config/shared/sdks.hpp>
 
#include <chrono>
#include <type_traits>
#include <variant>
 
namespace launchdarkly::config::shared::builders {
template <typename SDK>
class DataSourceBuilder;
 
template <typename T>
struct is_server_sdk : std::false_type {};
 
template <>
struct is_server_sdk<ServerSDK> : std::true_type {};
 
template <typename SDK>
class StreamingBuilder {
   public:
    StreamingBuilder();
 
    StreamingBuilder& InitialReconnectDelay(
        std::chrono::milliseconds initial_reconnect_delay);
 
    template <typename T = SDK>
    std::enable_if_t<is_server_sdk<T>::value, StreamingBuilder&> Filter(
        std::string filter_key) {
        config_.filter_key = std::move(filter_key);
        return *this;
    }
 
    [[nodiscard]] built::StreamingConfig<SDK> Build() const;
 
   private:
    built::StreamingConfig<SDK> config_;
};
 
template <typename SDK>
class PollingBuilder {
   public:
    PollingBuilder();
 
    PollingBuilder& PollInterval(std::chrono::seconds poll_interval);
 
    template <typename T = SDK>
    std::enable_if_t<is_server_sdk<T>::value, PollingBuilder&> Filter(
        std::string filter_key) {
        config_.filter_key = std::move(filter_key);
        return *this;
    }
 
    [[nodiscard]] built::PollingConfig<SDK> Build() const;
 
   private:
    built::PollingConfig<SDK> config_;
};
 
template <>
class DataSourceBuilder<ClientSDK> {
   public:
    using Streaming = StreamingBuilder<ClientSDK>;
    using Polling = PollingBuilder<ClientSDK>;
 
    DataSourceBuilder();
 
    DataSourceBuilder& WithReasons(bool value);
 
    DataSourceBuilder& UseReport(bool value);
 
    DataSourceBuilder& Method(Streaming stream_builder);
 
    DataSourceBuilder& Method(Polling polling_builder);
 
    [[nodiscard]] built::DataSourceConfig<ClientSDK> Build() const;
 
   private:
    std::variant<Streaming, Polling> method_;
    bool with_reasons_;
    bool use_report_;
};
 
}  // namespace launchdarkly::config::shared::builders