chrome/browser/search/suggestions/image_fetcher_impl.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/search/suggestions/image_fetcher_impl.h"
   6
   7 #include <string>
   8
   9 #include "content/public/browser/browser_thread.h"
  10 #include "net/base/load_flags.h"
  11 #include "net/url_request/url_request_context_getter.h"
  12
  13 namespace suggestions {
  14
  15 ImageFetcherImpl::ImageFetcherImpl(
  16     net::URLRequestContextGetter* url_request_context)
  17     : url_request_context_(url_request_context) {}
  18
  19 ImageFetcherImpl::~ImageFetcherImpl() {}
  20
  21 ImageFetcherImpl::ImageRequest::ImageRequest() : fetcher(NULL) {}
  22
  23 ImageFetcherImpl::ImageRequest::ImageRequest(chrome::BitmapFetcher* f)
  24     : fetcher(f) {}
  25
  26 ImageFetcherImpl::ImageRequest::~ImageRequest() { delete fetcher; }
  27
  28 void ImageFetcherImpl::SetImageFetcherDelegate(ImageFetcherDelegate* delegate) {
  29   DCHECK(delegate);
  30   delegate_ = delegate;
  31 }
  32
  33 void ImageFetcherImpl::StartOrQueueNetworkRequest(
  34     const GURL& url, const GURL& image_url,
  35     base::Callback<void(const GURL&, const SkBitmap*)> callback) {
  36   // Before starting to fetch the image. Look for a request in progress for
  37   // |image_url|, and queue if appropriate.
  38   ImageRequestMap::iterator it = pending_net_requests_.find(image_url);
  39   if (it == pending_net_requests_.end()) {
  40     // |image_url| is not being fetched, so create a request and initiate
  41     // the fetch.
  42     ImageRequest request(new chrome::BitmapFetcher(image_url, this));
  43     request.url = url;
  44     request.callbacks.push_back(callback);
  45     request.fetcher->Start(
  46         url_request_context_, std::string(),
  47         net::URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE,
  48         net::LOAD_NORMAL);
  49     pending_net_requests_[image_url].swap(&request);
  50   } else {
  51     // Request in progress. Register as an interested callback.
  52     it->second.callbacks.push_back(callback);
  53   }
  54 }
  55
  56 void ImageFetcherImpl::OnFetchComplete(const GURL& image_url,
  57                                        const SkBitmap* bitmap) {
  58   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
  59
  60   ImageRequestMap::iterator image_iter = pending_net_requests_.find(image_url);
  61   DCHECK(image_iter != pending_net_requests_.end());
  62
  63   ImageRequest* request = &image_iter->second;
  64
  65   // Here |bitmap| could be NULL or a pointer to a bitmap which is owned by the
  66   // BitmapFetcher and which ceases to exist after this function. Pass the
  67   // un-owned pointer to the registered callbacks.
  68   for (CallbackVector::iterator callback_iter = request->callbacks.begin();
  69        callback_iter != request->callbacks.end(); ++callback_iter) {
  70     callback_iter->Run(request->url, bitmap);
  71   }
  72
  73   // Inform the ImageFetcherDelegate.
  74   if (delegate_) {
  75     delegate_->OnImageFetched(request->url, bitmap);
  76   }
  77
  78   // Erase the completed ImageRequest.
  79   pending_net_requests_.erase(image_iter);
  80 }
  81
  82 }  // namespace suggestions