#pragma once

#include "envoy/config/core/v3/http_service.pb.h"

#include "envoy/server/factory_context.h"

#include "envoy/upstream/cluster_manager.h"

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

#include "source/common/http/async_client_impl.h"

#include "source/common/http/async_client_utility.h"

#include "source/common/http/http_service_headers.h"

#include "source/common/http/message_impl.h"

#include "source/common/http/utility.h"

#include "source/extensions/tracers/opentelemetry/trace_exporter.h"

#include "opentelemetry/proto/collector/trace/v1/trace_service.pb.h"

namespace Envoy {

namespace Extensions {

namespace Tracers {

namespace OpenTelemetry {

/**

 * Exporter for OTLP traces over HTTP.

 */

class OpenTelemetryHttpTraceExporter : public OpenTelemetryTraceExporter,

                                       public Http::AsyncClient::Callbacks {

public:

  OpenTelemetryHttpTraceExporter(

      Upstream::ClusterManager& cluster_manager,

      const envoy::config::core::v3::HttpService& http_service,

      std::shared_ptr<const Http::HttpServiceHeadersApplicator> headers_applicator);

  bool log(const ExportTraceServiceRequest& request) override;

  // Http::AsyncClient::Callbacks.

  void onSuccess(const Http::AsyncClient::Request&, Http::ResponseMessagePtr&&) override;

  void onFailure(const Http::AsyncClient::Request&, Http::AsyncClient::FailureReason) override;

private:

  Upstream::ClusterManager& cluster_manager_;

  envoy::config::core::v3::HttpService http_service_;

  // Track active HTTP requests to be able to cancel them on destruction.

  Http::AsyncClientRequestTracker active_requests_;

  std::shared_ptr<const Http::HttpServiceHeadersApplicator> headers_applicator_;

};

} // namespace OpenTelemetry

} // namespace Tracers

} // namespace Extensions

} // namespace Envoy