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

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

namespace Envoy {

namespace Http {

AsyncClientRequestTracker::~AsyncClientRequestTracker() {

  for (auto* active_request : active_requests_) {

    active_request->cancel();

  }

}

void AsyncClientRequestTracker::add(AsyncClient::Request& request) {

  ASSERT(active_requests_.find(&request) == active_requests_.end(), "request is already tracked.");

  active_requests_.insert(&request);

}

void AsyncClientRequestTracker::remove(const AsyncClient::Request& request) {

  // Notice that use of "const_cast" here is motivated by keeping API convenient for client code.

  // In the context where remove() will be typically called, request.cancel() is no longer

  // desirable and therefore get prevented by means of "const" modifier.

  auto it = active_requests_.find(const_cast<AsyncClient::Request*>(&request));

  // Support a use case where request callbacks might get called prior to a request handle

  // is returned from AsyncClient::send().

  if (it != active_requests_.end()) {

    active_requests_.erase(it);

  }

}

} // namespace Http

} // namespace Envoy