Add daily quota for precache

components/precache/core/precache_fetcher.h

 // Copyright 2013 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.
 
 #ifndef COMPONENTS_PRECACHE_CORE_PRECACHE_FETCHER_H_
 #define COMPONENTS_PRECACHE_CORE_PRECACHE_FETCHER_H_
 
 #include <stdint.h>
 
 #include <deque>
 #include <memory>
 #include <string>
 #include <vector>
 
 #include "base/callback.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/single_thread_task_runner.h"
 #include "base/time/time.h"
 #include "components/precache/core/fetcher_pool.h"
+#include "components/precache/core/proto/quota.pb.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "url/gurl.h"
 
 namespace base {
 class SingleThreadTaskRunner;
 }
 
 namespace net {
 class URLRequestContextGetter;
@@ skipping 113 matching lines @@
   // PrecacheFetcher instance.
   void Start();
 
   // Stops all precaching work. The PreacheFetcher should not be used after
   // calling this method.
   std::unique_ptr<PrecacheUnfinishedWork> CancelPrecaching();
 
  private:
   friend class PrecacheFetcherTest;
   FRIEND_TEST_ALL_PREFIXES(PrecacheFetcherTest, FetcherPoolMaxLimitReached);
+  FRIEND_TEST_ALL_PREFIXES(PrecacheFetcherTest, DailyQuota);
 
   // Notifies the precache delete that precaching is done, and report
   // completion statistics.
   void NotifyDone(size_t remaining_manifest_urls_to_fetch,
                   size_t remaining_resource_urls_to_fetch);
 
   // Fetches the next resource or manifest URL, if any remain. Fetching is done
   // sequentially and depth-first: all resources are fetched for a manifest
   // before the next manifest is fetched. This is done to limit the length of
   // the |resource_urls_to_fetch_| list, reducing the memory usage.
@@ skipping 18 matching lines @@
 
   // Called when a resource has been fetched.
   void OnResourceFetchComplete(const Fetcher& source);
 
   // Adds up the response sizes.
   void UpdateStats(int64_t response_bytes, int64_t network_response_bytes);
 
   // Callback invoked when the manifest info for all the top hosts is retrieved.
   void OnManifestInfoRetrieved(std::deque<ManifestHostInfo> manifests_info);
 
+  // Callback invoked when the quota is retrieved.
+  void OnQuotaInfoRetrieved(const PrecacheQuota& quota);
+
   // The request context used when fetching URLs.
   const scoped_refptr<net::URLRequestContextGetter> request_context_;
 
   // The custom URL to use when fetching the config. If not provided, the
   // default flag-specified URL will be used.
   const GURL config_url_;
 
   // The custom URL prefix to use when fetching manifests. If not provided, the
   // default flag-specified prefix will be used.
   const std::string manifest_url_prefix_;
 
   // PrecacheDatabase should be accessed on the DB thread.
   base::WeakPtr<PrecacheDatabase> precache_database_;
   scoped_refptr<base::SingleThreadTaskRunner> db_task_runner_;
 
   // Non-owning pointer. Should not be NULL.
   PrecacheDelegate* precache_delegate_;
 
   std::deque<ManifestHostInfo> top_hosts_to_fetch_;
   std::deque<std::pair<GURL, std::string>> resources_to_fetch_;
 
   FetcherPool<Fetcher> pool_;
 
   std::unique_ptr<PrecacheUnfinishedWork> unfinished_work_;
 
+  // Daily quota.
+  PrecacheQuota quota_;
+
   // The fieldtrial experiment ID.
   uint32_t experiment_id_;
 
   DISALLOW_COPY_AND_ASSIGN(PrecacheFetcher);
 };
 
 // Class that fetches a URL, and runs the specified callback when the fetch is
 // complete. This class exists so that a different method can be run in
 // response to different kinds of fetches, e.g. OnConfigFetchComplete when
 // configuration settings are fetched, OnManifestFetchComplete when a manifest
@@ skipping 58 matching lines @@
   int64_t response_bytes_;
   int64_t network_response_bytes_;
   bool was_cached_;
 
   DISALLOW_COPY_AND_ASSIGN(Fetcher);
 };
 
 }  // namespace precache
 
 #endif  // COMPONENTS_PRECACHE_CORE_PRECACHE_FETCHER_H_