Add pause/resume functionality to precache

components/precache/core/precache_database.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_DATABASE_H_
 #define COMPONENTS_PRECACHE_CORE_PRECACHE_DATABASE_H_
 
 #include <stdint.h>
 
+#include <list>
 #include <memory>
 #include <string>
 #include <vector>
 
 #include "base/callback.h"
 #include "base/containers/hash_tables.h"
 #include "base/macros.h"
+#include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/threading/thread_checker.h"
+#include "components/precache/core/precache_fetcher.h"
+#include "components/precache/core/precache_session_table.h"
 #include "components/precache/core/precache_url_table.h"
 
 class GURL;
 
 namespace base {
 class FilePath;
 class Time;
 }
 
 namespace sql {
 class Connection;
 }
 
 namespace precache {
 
+class PrecacheUnfinishedWork;
+
 // Class that tracks information related to precaching. This class may be
 // constructed on any thread, but all calls to, and destruction of this class
 // must be done on the the DB thread.
 class PrecacheDatabase {
  public:
   // A PrecacheDatabase can be constructed on any thread.
   PrecacheDatabase();
 
   ~PrecacheDatabase();
 
@@ skipping 20 matching lines @@
   // the fetch was not motivated by precaching. |is_connection_cellular|
   // indicates whether the current network connection is a cellular network.
   void RecordURLNonPrefetch(const GURL& url,
                             const base::TimeDelta& latency,
                             const base::Time& fetch_time,
                             int64_t size,
                             bool was_cached,
                             int host_rank,
                             bool is_connection_cellular);
 
+  // Gets the state required to continue a precache session.
+  std::unique_ptr<PrecacheUnfinishedWork> GetUnfinishedWork();
+
+  // Stores the state required to continue a precache session so that the
+  // session can be resumed later.
+  void SaveUnfinishedWork(
+      std::unique_ptr<PrecacheUnfinishedWork> unfinished_work);
+
+  // Deletes unfinished work from the database.
+  void DeleteUnfinishedWork();
+
+  base::WeakPtr<PrecacheDatabase> GetWeakPtr();
+
  private:
   friend class PrecacheDatabaseTest;
 
   bool IsDatabaseAccessible() const;
 
   // Flushes any buffered write operations. |buffered_writes_| will be empty
   // after calling this function. To maximize performance, all the buffered
   // writes are run in a single database transaction.
   void Flush();
 
   // Same as Flush(), but also updates the flag |is_flush_posted_| to indicate
   // that a flush is no longer posted.
   void PostedFlush();
 
   // Post a call to PostedFlush() on the current thread's MessageLoop, if
   // |buffered_writes_| is non-empty and there isn't already a flush call
   // posted.
   void MaybePostFlush();
 
   std::unique_ptr<sql::Connection> db_;
 
   // Table that keeps track of URLs that are in the cache because of precaching,
   // and wouldn't be in the cache otherwise. If |buffered_writes_| is non-empty,
   // then this table will not be up to date until the next call to Flush().
   PrecacheURLTable precache_url_table_;
 
+  // Table that persists state related to a precache session, including
+  // unfinished work to be done.
+  PrecacheSessionTable precache_session_table_;
+
   // A vector of write operations to be run on the database.
   std::vector<base::Closure> buffered_writes_;
 
   // Set of URLs that have been modified in |buffered_writes_|. It's a hash set
   // of strings, and not GURLs, because there is no hash function on GURL.
   base::hash_set<std::string> buffered_urls_;
 
   // Flag indicating whether or not a call to Flush() has been posted to run in
   // the future.
   bool is_flush_posted_;
 
   // ThreadChecker used to ensure that all methods other than the constructor
   // or destructor are called on the same thread.
   base::ThreadChecker thread_checker_;
 
   base::WeakPtrFactory<PrecacheDatabase> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(PrecacheDatabase);
 };
 
 }  // namespace precache
 
 #endif  // COMPONENTS_PRECACHE_CORE_PRECACHE_DATABASE_H_