net/http/http_cache_lookup_manager.cc

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/http/http_cache_lookup_manager.h"

#include "net/base/load_flags.h"

namespace net {

HttpCacheLookupManager::LookupTransaction::LookupTransaction(
    std::unique_ptr<ServerPushHelper> server_push_helper)
    : push_helper_(std::move(server_push_helper)),
      request_(new HttpRequestInfo()),
      transaction_(nullptr) {}

HttpCacheLookupManager::LookupTransaction::~LookupTransaction() {}

int HttpCacheLookupManager::LookupTransaction::StartLookup(
    HttpCache* cache,
    const CompletionCallback& callback,
    const NetLogWithSource& net_log) {
  request_->url = push_helper_->GetURL();
  request_->method = "GET";
  request_->load_flags = LOAD_ONLY_FROM_CACHE | LOAD_SKIP_CACHE_VALIDATION;
  cache->CreateTransaction(DEFAULT_PRIORITY, &transaction_);
  return transaction_->Start(request_.get(), callback, net_log);
}

void HttpCacheLookupManager::LookupTransaction::CancelPush() {
  DCHECK(push_helper_.get());
  push_helper_->Cancel();
}

HttpCacheLookupManager::HttpCacheLookupManager(HttpCache* http_cache,
                                               const NetLogWithSource& net_log)
    : net_log_(net_log), http_cache_(http_cache), weak_factory_(this) {}

HttpCacheLookupManager::~HttpCacheLookupManager() {}

void HttpCacheLookupManager::OnPush(
    std::unique_ptr<ServerPushHelper> push_helper) {
  GURL pushed_url = push_helper->GetURL();

  // There's a pending lookup transaction sent over already.
  if (base::ContainsKey(lookup_transactions_, pushed_url))
    return;

  auto lookup = base::MakeUnique<LookupTransaction>(std::move(push_helper));

  int rv = lookup->StartLookup(
      http_cache_, base::Bind(&HttpCacheLookupManager::OnLookupComplete,
                              weak_factory_.GetWeakPtr(), pushed_url),
      net_log_);

  if (rv == ERR_IO_PENDING)
    lookup_transactions_[pushed_url] = std::move(lookup);
}

void HttpCacheLookupManager::OnLookupComplete(const GURL& url, int rv) {
  auto it = lookup_transactions_.find(url);
  DCHECK(it != lookup_transactions_.end());

  if (rv == OK)
    it->second->CancelPush();

  lookup_transactions_.erase(it);
}

}  // namespace net