| Index: chrome/browser/history/in_memory_url_index.h
|
| ===================================================================
|
| --- chrome/browser/history/in_memory_url_index.h (revision 125451)
|
| +++ chrome/browser/history/in_memory_url_index.h (working copy)
|
| @@ -15,8 +15,8 @@
|
| #include "base/basictypes.h"
|
| #include "base/file_path.h"
|
| #include "base/gtest_prod_util.h"
|
| -#include "base/memory/linked_ptr.h"
|
| #include "base/memory/scoped_ptr.h"
|
| +#include "base/memory/weak_ptr.h"
|
| #include "base/string16.h"
|
| #include "chrome/browser/autocomplete/autocomplete_match.h"
|
| #include "chrome/browser/autocomplete/history_provider_util.h"
|
| @@ -24,7 +24,6 @@
|
| #include "chrome/browser/history/history.h"
|
| #include "chrome/browser/history/history_types.h"
|
| #include "chrome/browser/history/in_memory_url_index_types.h"
|
| -#include "chrome/browser/history/in_memory_url_index_cache.pb.h"
|
| #include "content/public/browser/notification_observer.h"
|
| #include "content/public/browser/notification_registrar.h"
|
| #include "sql/connection.h"
|
| @@ -45,6 +44,7 @@
|
| namespace imui = in_memory_url_index;
|
|
|
| class HistoryDatabase;
|
| +class RefCountedURLIndexPrivateDataPtr;
|
| class URLIndexPrivateData;
|
| struct URLVisitedDetails;
|
| struct URLsModifiedDetails;
|
| @@ -69,8 +69,33 @@
|
| // will eliminate such words except in the case where a single character
|
| // is being searched on and which character occurs as the second char16 of a
|
| // multi-char16 instance.
|
| -class InMemoryURLIndex : public content::NotificationObserver {
|
| +class InMemoryURLIndex : public content::NotificationObserver,
|
| + public base::SupportsWeakPtr<InMemoryURLIndex> {
|
| public:
|
| + // Defines an abstract class which is notified upon completion of restoring
|
| + // the index's private data either by reading from the cache file or by
|
| + // rebuilding from the history database.
|
| + class RestoreCacheObserver {
|
| + public:
|
| + virtual ~RestoreCacheObserver();
|
| +
|
| + // Callback that lets the observer know that the restore operation has
|
| + // completed. |succeeded| indicates if the restore was successful. This is
|
| + // called on the UI thread.
|
| + virtual void OnCacheRestoreFinished(bool succeeded) = 0;
|
| + };
|
| +
|
| + // Defines an abstract class which is notified upon completion of saving
|
| + // the index's private data to the cache file.
|
| + class SaveCacheObserver {
|
| + public:
|
| + virtual ~SaveCacheObserver();
|
| +
|
| + // Callback that lets the observer know that the save succeeded.
|
| + // This is called on the UI thread.
|
| + virtual void OnCacheSaveFinished(bool succeeded) = 0;
|
| + };
|
| +
|
| // |profile|, which may be NULL during unit testing, is used to register for
|
| // history changes. |history_dir| is a path to the directory containing the
|
| // history database within the profile wherein the cache and transaction
|
| @@ -96,11 +121,21 @@
|
| // refer to that class.
|
| ScoredHistoryMatches HistoryItemsForTerms(const string16& term_string);
|
|
|
| + // Sets the optional observers for completion of restoral and saving of the
|
| + // index's private data.
|
| + void set_restore_cache_observer(
|
| + RestoreCacheObserver* restore_cache_observer) {
|
| + restore_cache_observer_ = restore_cache_observer;
|
| + }
|
| + void set_save_cache_observer(SaveCacheObserver* save_cache_observer) {
|
| + save_cache_observer_ = save_cache_observer;
|
| + }
|
| +
|
| private:
|
| friend class ::HistoryQuickProviderTest;
|
| friend class InMemoryURLIndexTest;
|
| + friend class InMemoryURLIndexCacheTest;
|
| FRIEND_TEST_ALL_PREFIXES(LimitedInMemoryURLIndexTest, Initialization);
|
| - FRIEND_TEST_ALL_PREFIXES(InMemoryURLIndexCacheTest, CacheFilePath);
|
|
|
| // Creating one of me without a history path is not allowed (tests excepted).
|
| InMemoryURLIndex();
|
| @@ -133,14 +168,14 @@
|
| // provided as a hook for unit testing.)
|
| bool GetCacheFilePath(FilePath* file_path);
|
|
|
| + // Sets the directory wherein the cache file will be maintained.
|
| + // For unit test usage only.
|
| + void set_history_dir(const FilePath& dir_path) { history_dir_ = dir_path; }
|
| +
|
| // Restores the index's private data from the cache file stored in the
|
| // profile directory.
|
| - void RestoreFromCacheFile();
|
| + void PostRestoreFromCacheFileTask();
|
|
|
| - // Restores private_data_ from the given |path|. Runs on the UI thread.
|
| - // Provided for unit testing so that a test cache file can be used.
|
| - void DoRestoreFromCacheFile(const FilePath& path);
|
| -
|
| // Schedules a history task to rebuild our private data from the history
|
| // database.
|
| void ScheduleRebuildFromHistory();
|
| @@ -155,14 +190,42 @@
|
| // Used for unit testing only.
|
| void RebuildFromHistory(HistoryDatabase* history_db);
|
|
|
| - // Caches the index private data and writes the cache file to the profile
|
| - // directory.
|
| - void SaveToCacheFile();
|
| + // Posts a task to cache the index private data and write the cache file to
|
| + // the profile directory.
|
| + void PostSaveToCacheFileTask();
|
|
|
| // Saves private_data_ to the given |path|. Runs on the UI thread.
|
| // Provided for unit testing so that a test cache file can be used.
|
| void DoSaveToCacheFile(const FilePath& path);
|
|
|
| + // Notifies the observer, if any, of the success of the private data caching.
|
| + // |succeeded| is true on a successful save.
|
| + void OnCacheSaveDone(scoped_refptr<RefCountedBool> succeeded);
|
| +
|
| + // Callback used by RebuildPrivateDataFromHistoryDBTask to signal completion
|
| + // or rebuilding our private data from the history database. |succeeded|
|
| + // will be true if the rebuild was successful. |data| will point to a new
|
| + // instanceof the private data just rebuilt.
|
| + void DoneRebuidingPrivateDataFromHistoryDB(bool succeeded,
|
| + URLIndexPrivateData* data);
|
| +
|
| + // Rebuilds the history index from the history database in |history_db|.
|
| + // Used for unit testing only.
|
| + void RebuildFromHistory(URLDatabase* history_db);
|
| +
|
| + // Determines if the private data was successfully reloaded from the cache
|
| + // file or if the private data must be rebuilt from the history database.
|
| + // |private_data_ptr|'s data will be NULL if the cache file load failed.
|
| + void OnCacheLoadDone(
|
| + scoped_refptr<RefCountedURLIndexPrivateDataPtr> private_data_ptr);
|
| +
|
| + // Callback function that sets the private data from the just-restored-from-
|
| + // file |private_data| if |succeeded| otherwise clears the private data.
|
| + // Notifies any |restore_cache_observer_| of success status.
|
| + void OnCacheRestored(URLIndexPrivateData* private_data, bool succeeded);
|
| +
|
| + // Notifications -------------------------------------------------------------
|
| +
|
| // Handles notifications of history changes.
|
| virtual void Observe(int notification_type,
|
| const content::NotificationSource& source,
|
| @@ -187,12 +250,16 @@
|
| // The index's durable private data.
|
| scoped_ptr<URLIndexPrivateData> private_data_;
|
|
|
| - // Set to true once the shutdown process has begun.
|
| - bool shutdown_;
|
| + // Observers to notify upon restoral or save of the private data cache.
|
| + RestoreCacheObserver* restore_cache_observer_;
|
| + SaveCacheObserver* save_cache_observer_;
|
|
|
| CancelableRequestConsumer cache_reader_consumer_;
|
| content::NotificationRegistrar registrar_;
|
|
|
| + // Set to true once the shutdown process has begun.
|
| + bool shutdown_;
|
| +
|
| // Set to true when changes to the index have been made and the index needs
|
| // to be cached. Set to false when the index has been cached. Used as a
|
| // temporary safety check to insure that the cache is saved before the
|
|
|