1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109
|
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_IMAGE_FETCHER_CORE_IMAGE_FETCHER_IMPL_H_
#define COMPONENTS_IMAGE_FETCHER_CORE_IMAGE_FETCHER_IMPL_H_
#include <map>
#include <memory>
#include <string>
#include <utility>
#include <vector>
#include "base/functional/callback.h"
#include "base/memory/weak_ptr.h"
#include "components/image_fetcher/core/image_data_fetcher.h"
#include "components/image_fetcher/core/image_decoder.h"
#include "components/image_fetcher/core/image_fetcher.h"
#include "net/traffic_annotation/network_traffic_annotation.h"
#include "ui/gfx/geometry/size.h"
#include "url/gurl.h"
namespace gfx {
class Image;
}
namespace network {
class SharedURLLoaderFactory;
}
namespace image_fetcher {
// The standard (non-test) implementation of ImageFetcher.
class ImageFetcherImpl : public ImageFetcher {
public:
ImageFetcherImpl(
std::unique_ptr<ImageDecoder> image_decoder,
scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory);
ImageFetcherImpl(const ImageFetcherImpl&) = delete;
ImageFetcherImpl& operator=(const ImageFetcherImpl&) = delete;
~ImageFetcherImpl() override;
void FetchImageAndData(const GURL& image_url,
ImageDataFetcherCallback image_data_callback,
ImageFetcherCallback image_callback,
ImageFetcherParams params) override;
ImageDecoder* GetImageDecoder() override;
private:
// State related to an image fetch (id, pending callbacks).
struct ImageRequest {
ImageRequest();
ImageRequest(const ImageRequest&) = delete;
ImageRequest& operator=(const ImageRequest&) = delete;
ImageRequest(ImageRequest&& other);
~ImageRequest();
std::string id;
// These have the default value if the image data has not yet been fetched.
RequestMetadata request_metadata;
std::string image_data;
// Queue for pending callbacks, which may accumulate while the request is in
// flight.
std::vector<ImageFetcherCallback> image_callbacks;
std::vector<ImageDataFetcherCallback> image_data_callbacks;
};
using ImageRequestMap = std::map<GURL, ImageRequest>;
// Processes image URL fetched events. This is the continuation method used
// for creating callbacks that are passed to the ImageDataFetcher.
void OnImageURLFetched(const GURL& image_url,
ImageFetcherParams params,
const std::string& image_data,
const RequestMetadata& metadata);
// Processes image decoded events. This is the continuation method used for
// creating callbacks that are passed to the ImageDecoder.
void OnImageDecoded(const GURL& image_url,
const RequestMetadata& metadata,
const gfx::Image& image);
// Used to run `image_data_callback` only if `this` is still valid.
void RunImageDataCallback(ImageDataFetcherCallback image_data_callback,
std::string image_data,
RequestMetadata request_metadata);
scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory_;
std::unique_ptr<ImageDecoder> image_decoder_;
std::unique_ptr<ImageDataFetcher> image_data_fetcher_;
// Map from each image URL to the request information (associated website
// url, fetcher, pending callbacks).
ImageRequestMap pending_net_requests_;
base::WeakPtrFactory<ImageFetcherImpl> weak_ptr_factory_{this};
};
} // namespace image_fetcher
#endif // COMPONENTS_IMAGE_FETCHER_CORE_IMAGE_FETCHER_IMPL_H_
|