#pragma once

#include "envoy/extensions/load_balancing_policies/random/v3/random.pb.h"

#include "envoy/extensions/load_balancing_policies/random/v3/random.pb.validate.h"

#include "envoy/upstream/load_balancer.h"

#include "source/common/common/logger.h"

#include "source/extensions/load_balancing_policies/common/factory_base.h"

namespace Envoy {

namespace Extensions {

namespace LoadBalancingPolicies {

namespace Random {

using RandomLbProto = envoy::extensions::load_balancing_policies::random::v3::Random;

using ClusterProto = envoy::config::cluster::v3::Cluster;

using CommonLbConfigProto = envoy::config::cluster::v3::Cluster::CommonLbConfig;

/**

 * Load balancer config that used to wrap the random config.

 */

class TypedRandomLbConfig : public Upstream::LoadBalancerConfig {

public:

  TypedRandomLbConfig(const RandomLbProto& lb_config);

  TypedRandomLbConfig(const CommonLbConfigProto& common_lb_config);

  RandomLbProto lb_config_;

};

struct RandomCreator : public Logger::Loggable<Logger::Id::upstream> {

  Upstream::LoadBalancerPtr operator()(

      Upstream::LoadBalancerParams params, OptRef<const Upstream::LoadBalancerConfig> lb_config,

      const Upstream::ClusterInfo& cluster_info, const Upstream::PrioritySet& priority_set,

      Runtime::Loader& runtime, Envoy::Random::RandomGenerator& random, TimeSource& time_source);

};

class Factory : public Common::FactoryBase<RandomLbProto, RandomCreator> {

public:

  absl::StatusOr<Upstream::LoadBalancerConfigPtr>

  loadConfig(Server::Configuration::ServerFactoryContext&,

  absl::StatusOr<Upstream::LoadBalancerConfigPtr>

};

DECLARE_FACTORY(Factory);

} // namespace Random

} // namespace LoadBalancingPolicies

} // namespace Extensions

} // namespace Envoy