OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/history/in_memory_history_backend.h" | 5 #include "chrome/browser/history/in_memory_history_backend.h" |
6 | 6 |
7 #include <set> | 7 #include <set> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/command_line.h" | 10 #include "base/command_line.h" |
11 #include "base/time.h" | 11 #include "base/time.h" |
12 #include "base/utf_string_conversions.h" | 12 #include "base/utf_string_conversions.h" |
13 #include "chrome/browser/browser_process.h" | 13 #include "chrome/browser/browser_process.h" |
14 #include "chrome/browser/history/history_notifications.h" | 14 #include "chrome/browser/history/history_notifications.h" |
15 #include "chrome/browser/history/in_memory_database.h" | 15 #include "chrome/browser/history/in_memory_database.h" |
| 16 #include "chrome/browser/history/in_memory_url_index.h" |
16 #include "chrome/browser/history/url_database.h" | 17 #include "chrome/browser/history/url_database.h" |
17 #include "chrome/browser/profiles/profile.h" | 18 #include "chrome/browser/profiles/profile.h" |
18 #include "chrome/common/chrome_notification_types.h" | 19 #include "chrome/common/chrome_notification_types.h" |
19 #include "content/public/browser/notification_service.h" | 20 #include "chrome/common/chrome_switches.h" |
| 21 #include "content/public/browser/notification_details.h" |
| 22 #include "content/public/browser/notification_source.h" |
20 | 23 |
21 namespace history { | 24 namespace history { |
22 | 25 |
23 InMemoryHistoryBackend::InMemoryHistoryBackend() | 26 InMemoryHistoryBackend::InMemoryHistoryBackend() |
24 : profile_(NULL) { | 27 : profile_(NULL) { |
25 } | 28 } |
26 | 29 |
27 InMemoryHistoryBackend::~InMemoryHistoryBackend() {} | 30 InMemoryHistoryBackend::~InMemoryHistoryBackend() { |
| 31 if (index_.get()) |
| 32 index_->ShutDown(); |
| 33 } |
28 | 34 |
29 bool InMemoryHistoryBackend::Init(const FilePath& history_filename, | 35 bool InMemoryHistoryBackend::Init(const FilePath& history_filename, |
30 const FilePath& history_dir, | 36 const FilePath& history_dir, |
31 URLDatabase* db, | 37 URLDatabase* db, |
32 const std::string& languages) { | 38 const std::string& languages) { |
33 db_.reset(new InMemoryDatabase); | 39 db_.reset(new InMemoryDatabase); |
34 return db_->InitFromDisk(history_filename); | 40 bool success = db_->InitFromDisk(history_filename); |
| 41 if (!CommandLine::ForCurrentProcess()->HasSwitch( |
| 42 switches::kDisableHistoryQuickProvider)) { |
| 43 index_.reset(new InMemoryURLIndex(history_dir)); |
| 44 index_->Init(db, languages); |
| 45 } |
| 46 return success; |
35 } | 47 } |
36 | 48 |
37 void InMemoryHistoryBackend::AttachToHistoryService(Profile* profile) { | 49 void InMemoryHistoryBackend::AttachToHistoryService(Profile* profile) { |
38 if (!db_.get()) { | 50 if (!db_.get()) { |
39 NOTREACHED(); | 51 NOTREACHED(); |
40 return; | 52 return; |
41 } | 53 } |
42 | 54 |
43 profile_ = profile; | 55 profile_ = profile; |
44 | 56 |
(...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
112 // IDs in the main database. This sucks. Instead of Add and Remove, we should | 124 // IDs in the main database. This sucks. Instead of Add and Remove, we should |
113 // have Sync(), which would take the ID if it's given and add it. | 125 // have Sync(), which would take the ID if it's given and add it. |
114 std::vector<history::URLRow>::const_iterator i; | 126 std::vector<history::URLRow>::const_iterator i; |
115 for (i = details.changed_urls.begin(); | 127 for (i = details.changed_urls.begin(); |
116 i != details.changed_urls.end(); i++) { | 128 i != details.changed_urls.end(); i++) { |
117 URLID id = db_->GetRowForURL(i->url(), NULL); | 129 URLID id = db_->GetRowForURL(i->url(), NULL); |
118 if (id) | 130 if (id) |
119 db_->UpdateURLRow(id, *i); | 131 db_->UpdateURLRow(id, *i); |
120 else | 132 else |
121 id = db_->AddURL(*i); | 133 id = db_->AddURL(*i); |
| 134 if (index_.get()) |
| 135 index_->UpdateURL(id, *i); |
122 } | 136 } |
123 } | 137 } |
124 | 138 |
125 void InMemoryHistoryBackend::OnURLsDeleted(const URLsDeletedDetails& details) { | 139 void InMemoryHistoryBackend::OnURLsDeleted(const URLsDeletedDetails& details) { |
126 DCHECK(db_.get()); | 140 DCHECK(db_.get()); |
| 141 |
127 if (details.all_history) { | 142 if (details.all_history) { |
128 // When all history is deleted, the individual URLs won't be listed. Just | 143 // When all history is deleted, the individual URLs won't be listed. Just |
129 // create a new database to quickly clear everything out. | 144 // create a new database to quickly clear everything out. |
130 db_.reset(new InMemoryDatabase); | 145 db_.reset(new InMemoryDatabase); |
131 if (!db_->InitFromScratch()) | 146 if (!db_->InitFromScratch()) |
132 db_.reset(); | 147 db_.reset(); |
| 148 if (index_.get()) |
| 149 index_->ReloadFromHistory(db_.get(), true); |
133 return; | 150 return; |
134 } | 151 } |
135 | 152 |
136 // Delete all matching URLs in our database. | 153 // Delete all matching URLs in our database. |
137 for (std::vector<URLRow>::const_iterator row = details.rows.begin(); | 154 for (std::set<GURL>::const_iterator i = details.urls.begin(); |
138 row != details.rows.end(); ++row) { | 155 i != details.urls.end(); ++i) { |
139 // We typically won't have most of them since we only have a subset of | 156 URLID id = db_->GetRowForURL(*i, NULL); |
140 // history, so ignore errors. | 157 if (id) { |
141 db_->DeleteURLRow(row->id()); | 158 // We typically won't have most of them since we only have a subset of |
| 159 // history, so ignore errors. |
| 160 db_->DeleteURLRow(id); |
| 161 if (index_.get()) |
| 162 index_->DeleteURL(id); |
| 163 } |
142 } | 164 } |
143 } | 165 } |
144 | 166 |
145 void InMemoryHistoryBackend::OnKeywordSearchTermUpdated( | 167 void InMemoryHistoryBackend::OnKeywordSearchTermUpdated( |
146 const KeywordSearchTermDetails& details) { | 168 const KeywordSearchTermDetails& details) { |
147 // The url won't exist for new search terms (as the user hasn't typed it), so | 169 // The url won't exist for new search terms (as the user hasn't typed it), so |
148 // we force it to be added. If we end up adding a URL it won't be | 170 // we force it to be added. If we end up adding a URL it won't be |
149 // autocompleted as the typed count is 0. | 171 // autocompleted as the typed count is 0. |
150 URLRow url_row; | 172 URLRow url_row; |
151 URLID url_id; | 173 URLID url_id; |
(...skipping 15 matching lines...) Expand all Loading... |
167 | 189 |
168 bool InMemoryHistoryBackend::HasKeyword(const GURL& url) { | 190 bool InMemoryHistoryBackend::HasKeyword(const GURL& url) { |
169 URLID id = db_->GetRowForURL(url, NULL); | 191 URLID id = db_->GetRowForURL(url, NULL); |
170 if (!id) | 192 if (!id) |
171 return false; | 193 return false; |
172 | 194 |
173 return db_->GetKeywordSearchTermRow(id, NULL); | 195 return db_->GetKeywordSearchTermRow(id, NULL); |
174 } | 196 } |
175 | 197 |
176 } // namespace history | 198 } // namespace history |
OLD | NEW |