#pragma once

#include "envoy/common/optref.h"

#include "envoy/common/random_generator.h"

#include "envoy/common/time.h"

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

#include "envoy/runtime/runtime.h"

#include "envoy/upstream/load_balancer.h"

#include "envoy/upstream/upstream.h"

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

#include "source/common/upstream/load_balancer_factory_base.h"

#include "absl/status/statusor.h"

namespace Envoy {

namespace Extensions {

namespace LoadBalancingPolicies {

namespace OverrideHost {

using ::envoy::extensions::load_balancing_policies::override_host::v3::OverrideHost;

using ::Envoy::Random::RandomGenerator;

using ::Envoy::Runtime::Loader;

using ::Envoy::Upstream::ClusterInfo;

using ::Envoy::Upstream::PrioritySet;

class OverrideHostLoadBalancerFactory

    : public Upstream::TypedLoadBalancerFactoryBase<OverrideHost> {

public:

  OverrideHostLoadBalancerFactory()

  absl::StatusOr<Upstream::LoadBalancerConfigPtr>

  loadConfig(Server::Configuration::ServerFactoryContext& context,

             const Protobuf::Message& config) override;

  Upstream::ThreadAwareLoadBalancerPtr

  create(Envoy::OptRef<const Upstream::LoadBalancerConfig> lb_config,

         const ClusterInfo& cluster_info, const PrioritySet& priority_set, Loader& runtime,

         RandomGenerator& random, TimeSource& time_source) override;

};

} // namespace OverrideHost

} // namespace LoadBalancingPolicies

} // namespace Extensions

} // namespace Envoy