client_impl.hpp

#pragma once
 
#include "data_components/big_segments/big_segment_store_status_provider.hpp"
#include "data_components/big_segments/big_segment_store_wrapper.hpp"
#include "data_components/status_notifications/data_source_status_manager.hpp"
#include "data_interfaces/system/idata_system.hpp"
#include "evaluation/evaluator.hpp"
#include "events/event_scope.hpp"
#include "hooks/hook_executor.hpp"
 
#include <launchdarkly/config/client.hpp>
#include <launchdarkly/context.hpp>
#include <launchdarkly/data/evaluation_detail.hpp>
#include <launchdarkly/error.hpp>
#include <launchdarkly/events/event_processor_interface.hpp>
#include <launchdarkly/logging/logger.hpp>
#include <launchdarkly/server_side/client.hpp>
#include <launchdarkly/value.hpp>
 
#include <boost/asio/executor_work_guard.hpp>
#include <boost/asio/io_context.hpp>
 
#include <tl/expected.hpp>
 
#include <condition_variable>
#include <cstdint>
#include <memory>
#include <optional>
#include <shared_mutex>
#include <thread>
#include <tuple>
 
namespace launchdarkly::server_side {
 
class ClientImpl : public IClient {
   public:
    ClientImpl(Config config, std::string const& version);
 
    ClientImpl(ClientImpl&&) = delete;
    ClientImpl(ClientImpl const&) = delete;
    ClientImpl& operator=(ClientImpl) = delete;
    ClientImpl& operator=(ClientImpl&& other) = delete;
 
    bool Initialized() const override;
 
    using FlagKey = std::string;
    [[nodiscard]] class AllFlagsState AllFlagsState(
        Context const& context,
        AllFlagsState::Options options =
            AllFlagsState::Options::Default) override;
 
    void Track(Context const& ctx,
               std::string event_name,
               Value data,
               double metric_value) override;
 
    void Track(Context const& ctx,
               std::string event_name,
               Value data,
               double metric_value,
               hooks::HookContext const& hook_context) override;
 
    void Track(Context const& ctx, std::string event_name, Value data) override;
 
    void Track(Context const& ctx,
               std::string event_name,
               Value data,
               hooks::HookContext const& hook_context) override;
 
    void Track(Context const& ctx, std::string event_name) override;
 
    void Track(Context const& ctx,
               std::string event_name,
               hooks::HookContext const& hook_context) override;
 
    void FlushAsync() override;
 
    void Identify(Context context) override;
 
    bool BoolVariation(Context const& ctx,
                       FlagKey const& key,
                       bool default_value) override;
 
    bool BoolVariation(Context const& ctx,
                       FlagKey const& key,
                       bool default_value,
                       hooks::HookContext const& hook_context) override;
 
    EvaluationDetail<bool> BoolVariationDetail(Context const& ctx,
                                               FlagKey const& key,
                                               bool default_value) override;
 
    EvaluationDetail<bool> BoolVariationDetail(
        Context const& ctx,
        FlagKey const& key,
        bool default_value,
        hooks::HookContext const& hook_context) override;
 
    std::string StringVariation(Context const& ctx,
                                FlagKey const& key,
                                std::string default_value) override;
 
    std::string StringVariation(
        Context const& ctx,
        FlagKey const& key,
        std::string default_value,
        hooks::HookContext const& hook_context) override;
 
    EvaluationDetail<std::string> StringVariationDetail(
        Context const& ctx,
        FlagKey const& key,
        std::string default_value) override;
 
    EvaluationDetail<std::string> StringVariationDetail(
        Context const& ctx,
        FlagKey const& key,
        std::string default_value,
        hooks::HookContext const& hook_context) override;
 
    double DoubleVariation(Context const& ctx,
                           FlagKey const& key,
                           double default_value) override;
 
    double DoubleVariation(Context const& ctx,
                           FlagKey const& key,
                           double default_value,
                           hooks::HookContext const& hook_context) override;
 
    EvaluationDetail<double> DoubleVariationDetail(
        Context const& ctx,
        FlagKey const& key,
        double default_value) override;
 
    EvaluationDetail<double> DoubleVariationDetail(
        Context const& ctx,
        FlagKey const& key,
        double default_value,
        hooks::HookContext const& hook_context) override;
 
    int IntVariation(Context const& ctx,
                     FlagKey const& key,
                     int default_value) override;
 
    int IntVariation(Context const& ctx,
                     FlagKey const& key,
                     int default_value,
                     hooks::HookContext const& hook_context) override;
 
    EvaluationDetail<int> IntVariationDetail(Context const& ctx,
                                             FlagKey const& key,
                                             int default_value) override;
 
    EvaluationDetail<int> IntVariationDetail(
        Context const& ctx,
        FlagKey const& key,
        int default_value,
        hooks::HookContext const& hook_context) override;
 
    Value JsonVariation(Context const& ctx,
                        FlagKey const& key,
                        Value default_value) override;
 
    Value JsonVariation(Context const& ctx,
                        FlagKey const& key,
                        Value default_value,
                        hooks::HookContext const& hook_context) override;
 
    EvaluationDetail<Value> JsonVariationDetail(Context const& ctx,
                                                FlagKey const& key,
                                                Value default_value) override;
 
    EvaluationDetail<Value> JsonVariationDetail(
        Context const& ctx,
        FlagKey const& key,
        Value default_value,
        hooks::HookContext const& hook_context) override;
 
    IDataSourceStatusProvider& DataSourceStatus() override;
 
    IBigSegmentStoreStatusProvider& BigSegmentStoreStatus() override;
 
    ~ClientImpl();
 
    std::future<bool> StartAsync() override;
 
   private:
    [[nodiscard]] EvaluationDetail<Value> VariationInternal(
        Context const& ctx,
        FlagKey const& key,
        Value const& default_value,
        EventScope const& scope,
        hooks::HookContext const& hook_context,
        std::string const& method_name);
 
    template <typename T>
    [[nodiscard]] EvaluationDetail<T> VariationDetail(
        Context const& ctx,
        enum Value::Type value_type,
        IClient::FlagKey const& key,
        Value const& default_value,
        hooks::HookContext const& hook_context,
        std::string const& method_name) {
        auto result =
            VariationInternal(ctx, key, default_value, events_with_reasons_,
                              hook_context, method_name);
        if (result.Value().Type() == value_type) {
            return EvaluationDetail<T>{result.Value(), result.VariationIndex(),
                                       result.Reason()};
        }
        return EvaluationDetail<T>{EvaluationReason::ErrorKind::kWrongType,
                                   default_value};
    }
 
    [[nodiscard]] Value Variation(Context const& ctx,
                                  enum Value::Type value_type,
                                  std::string const& key,
                                  Value const& default_value,
                                  hooks::HookContext const& hook_context,
                                  std::string const& method_name);
 
    [[nodiscard]] EvaluationDetail<Value> PostEvaluation(
        std::string const& key,
        Context const& context,
        Value const& default_value,
        std::variant<enum EvaluationReason::ErrorKind, EvaluationDetail<Value>>
            result,
        EventScope const& event_scope,
        std::optional<data_model::Flag> const& flag);
 
    [[nodiscard]] std::optional<enum EvaluationReason::ErrorKind>
    PreEvaluationChecks(Context const& context) const;
 
    void TrackInternal(Context const& ctx,
                       std::string event_name,
                       std::optional<Value> data,
                       std::optional<double> metric_value,
                       hooks::HookContext const& hook_context);
 
    std::future<bool> StartAsyncInternal(
        std::function<bool(DataSourceStatus::DataSourceState)> predicate);
 
    void LogVariationCall(std::string const& key, bool flag_present) const;
 
    Config config_;
    Logger logger_;
 
    launchdarkly::config::shared::built::HttpProperties http_properties_;
 
    boost::asio::io_context ioc_;
    boost::asio::executor_work_guard<boost::asio::io_context::executor_type>
        work_;
 
    data_components::DataSourceStatusManager status_manager_;
 
    // This is the main polymorphic component that constitutes the
    // guts of how data is retrieved (polling, streaming, persistent stores,
    // etc.)
    std::unique_ptr<data_interfaces::IDataSystem> data_system_;
 
    std::unique_ptr<events::IEventProcessor> event_processor_;
 
    // Null when Big Segments are not configured. Declared before evaluator_ so
    // its pointer can be handed to the evaluator, and before
    // big_segment_status_provider_ which shares ownership.
    std::shared_ptr<data_components::BigSegmentStoreWrapper> big_segment_store_;
    data_components::BigSegmentStoreStatusProvider big_segment_status_provider_;
 
    mutable std::mutex init_mutex_;
    std::condition_variable init_waiter_;
 
    evaluation::Evaluator evaluator_;
 
    EventScope const events_default_;
    EventScope const events_with_reasons_;
 
    std::thread run_thread_;
};
}  // namespace launchdarkly::server_side