#pragma once

#include <memory>

#include <string>

#include "envoy/common/pure.h"

#include "envoy/network/address.h"

#include "source/common/network/io_socket_handle_impl.h"

namespace Envoy {

namespace Extensions {

namespace Tracers {

namespace XRay {

/**

 * The broker is a way to isolate the network dependency required to communicate with the X-Ray

 * daemon.

 */

class DaemonBroker {

public:

  /**

   * Sends the input string as data to the X-Ray daemon.

   * The input string is typically a JSON serialized Span.

   * This method prefixes the data with a header necessary for the daemon to accept the input.

   */

  virtual void send(const std::string& data) const PURE;

};

using DaemonBrokerPtr = std::unique_ptr<DaemonBroker>;

class DaemonBrokerImpl : public DaemonBroker {

public:

  /**

   * Creates a new Broker instance.

   *

   * @param daemon_endpoint The ip and port on which the X-Ray daemon is listening.

   */

  explicit DaemonBrokerImpl(const std::string& daemon_endpoint);

  void send(const std::string& data) const final;

private:

  const Network::Address::InstanceConstSharedPtr address_;

  const Network::IoHandlePtr io_handle_;

};

} // namespace XRay

} // namespace Tracers

} // namespace Extensions

} // namespace Envoy