chrome/browser/search/suggestions/image_fetcher_impl.h

   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 #ifndef CHROME_BROWSER_SEARCH_SUGGESTIONS_IMAGE_FETCHER_IMPL_H_
   6 #define CHROME_BROWSER_SEARCH_SUGGESTIONS_IMAGE_FETCHER_IMPL_H_
   7
   8 #include <map>
   9 #include <utility>
  10 #include <vector>
  11
  12 #include "base/basictypes.h"
  13 #include "base/callback.h"
  14 #include "chrome/browser/bitmap_fetcher/bitmap_fetcher.h"
  15 #include "components/suggestions/image_fetcher.h"
  16 #include "ui/gfx/image/image_skia.h"
  17 #include "url/gurl.h"
  18
  19 namespace net {
  20 class URLRequestContextGetter;
  21 }
  22
  23 namespace suggestions {
  24
  25 // A class used to fetch server images. It can be called from any thread and the
  26 // callback will be called on the thread which initiated the fetch.
  27 class ImageFetcherImpl : public ImageFetcher,
  28                          public chrome::BitmapFetcherDelegate {
  29  public:
  30   explicit ImageFetcherImpl(net::URLRequestContextGetter* url_request_context);
  31   ~ImageFetcherImpl() override;
  32
  33   void SetImageFetcherDelegate(ImageFetcherDelegate* delegate) override;
  34
  35   void StartOrQueueNetworkRequest(
  36       const GURL& url,
  37       const GURL& image_url,
  38       base::Callback<void(const GURL&, const SkBitmap*)> callback) override;
  39
  40  private:
  41   // Inherited from BitmapFetcherDelegate.
  42   void OnFetchComplete(const GURL& image_url, const SkBitmap* bitmap) override;
  43
  44   typedef std::vector<base::Callback<void(const GURL&, const SkBitmap*)> >
  45       CallbackVector;
  46
  47   // State related to an image fetch (associated website url, image_url,
  48   // fetcher, pending callbacks).
  49   struct ImageRequest {
  50     ImageRequest();
  51     // Struct takes ownership of |f|.
  52     explicit ImageRequest(chrome::BitmapFetcher* f);
  53     ~ImageRequest();
  54
  55     void swap(ImageRequest* other) {
  56       std::swap(url, other->url);
  57       std::swap(image_url, other->image_url);
  58       std::swap(callbacks, other->callbacks);
  59       std::swap(fetcher, other->fetcher);
  60     }
  61
  62     GURL url;
  63     GURL image_url;
  64     chrome::BitmapFetcher* fetcher;
  65     // Queue for pending callbacks, which may accumulate while the request is in
  66     // flight.
  67     CallbackVector callbacks;
  68   };
  69
  70   typedef std::map<const GURL, ImageRequest> ImageRequestMap;
  71
  72   // Map from each image URL to the request information (associated website
  73   // url, fetcher, pending callbacks).
  74   ImageRequestMap pending_net_requests_;
  75
  76   ImageFetcherDelegate* delegate_;
  77
  78   net::URLRequestContextGetter* url_request_context_;
  79
  80   DISALLOW_COPY_AND_ASSIGN(ImageFetcherImpl);
  81 };
  82
  83 }  // namespace suggestions
  84
  85 #endif  // CHROME_BROWSER_SEARCH_SUGGESTIONS_IMAGE_FETCHER_IMPL_H_