| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 <string> | 5 #include <string> |
| 6 #include "base/logging.h" | 6 #include "base/logging.h" |
| 7 #include "base/string_util.h" | 7 #include "base/string_util.h" |
| 8 #include "chrome/browser/extensions/activity_database.h" | 8 #include "chrome/browser/extensions/activity_database.h" |
| 9 #include "chrome/browser/history/url_database.h" | 9 #include "chrome/browser/history/url_database.h" |
| 10 #include "content/public/browser/browser_thread.h" | 10 #include "content/public/browser/browser_thread.h" |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 43 if (!committer.Begin()) | 43 if (!committer.Begin()) |
| 44 return LogInitFailure(); | 44 return LogInitFailure(); |
| 45 | 45 |
| 46 #if defined(OS_MACOSX) | 46 #if defined(OS_MACOSX) |
| 47 // Exclude the database from backups. | 47 // Exclude the database from backups. |
| 48 base::mac::SetFileBackupExclusion(db_name); | 48 base::mac::SetFileBackupExclusion(db_name); |
| 49 #endif | 49 #endif |
| 50 | 50 |
| 51 db_.Preload(); | 51 db_.Preload(); |
| 52 | 52 |
| 53 // Create the UrlAction database. | 53 // Create the DOMAction database. |
| 54 if (InitializeTable(UrlAction::kTableName, UrlAction::kTableStructure) != | 54 if (!DOMAction::InitializeTable(&db_)) |
| 55 sql::INIT_OK) | |
| 56 return LogInitFailure(); | 55 return LogInitFailure(); |
| 57 | 56 |
| 58 // Create the APIAction database. | 57 // Create the APIAction database. |
| 59 if (!APIAction::InitializeTable(&db_)) | 58 if (!APIAction::InitializeTable(&db_)) |
| 60 return LogInitFailure(); | 59 return LogInitFailure(); |
| 61 | 60 |
| 62 // Create the BlockedAction database. | 61 // Create the BlockedAction database. |
| 63 if (InitializeTable(BlockedAction::kTableName, BlockedAction::kTableStructure) | 62 if (!BlockedAction::InitializeTable(&db_)) |
| 64 != sql::INIT_OK) | |
| 65 return LogInitFailure(); | 63 return LogInitFailure(); |
| 66 | 64 |
| 67 sql::InitStatus stat = committer.Commit() ? sql::INIT_OK : sql::INIT_FAILURE; | 65 sql::InitStatus stat = committer.Commit() ? sql::INIT_OK : sql::INIT_FAILURE; |
| 68 if (stat == sql::INIT_OK) | 66 if (stat == sql::INIT_OK) |
| 69 initialized_ = true; | 67 initialized_ = true; |
| 70 else | 68 else |
| 71 return LogInitFailure(); | 69 return LogInitFailure(); |
| 72 } | 70 } |
| 73 | 71 |
| 74 void ActivityDatabase::LogInitFailure() { | 72 void ActivityDatabase::LogInitFailure() { |
| 75 LOG(ERROR) << "Couldn't initialize the activity log database."; | 73 LOG(ERROR) << "Couldn't initialize the activity log database."; |
| 76 } | 74 } |
| 77 | 75 |
| 78 sql::InitStatus ActivityDatabase::InitializeTable(const char* table_name, | |
| 79 const char* table_structure) { | |
| 80 if (!db_.DoesTableExist(table_name)) { | |
| 81 char table_creator[1000]; | |
| 82 base::snprintf(table_creator, | |
| 83 arraysize(table_creator), | |
| 84 "CREATE TABLE %s %s", table_name, table_structure); | |
| 85 if (!db_.Execute(table_creator)) | |
| 86 return sql::INIT_FAILURE; | |
| 87 } | |
| 88 return sql::INIT_OK; | |
| 89 } | |
| 90 | |
| 91 void ActivityDatabase::RecordAction(scoped_refptr<Action> action) { | 76 void ActivityDatabase::RecordAction(scoped_refptr<Action> action) { |
| 92 if (initialized_) | 77 if (initialized_) |
| 93 action->Record(&db_); | 78 action->Record(&db_); |
| 94 } | 79 } |
| 95 | 80 |
| 96 void ActivityDatabase::BeginTransaction() { | 81 void ActivityDatabase::BeginTransaction() { |
| 97 db_.BeginTransaction(); | 82 db_.BeginTransaction(); |
| 98 } | 83 } |
| 99 | 84 |
| 100 void ActivityDatabase::CommitTransaction() { | 85 void ActivityDatabase::CommitTransaction() { |
| (...skipping 10 matching lines...) Expand all Loading... |
| 111 | 96 |
| 112 void ActivityDatabase::Close() { | 97 void ActivityDatabase::Close() { |
| 113 db_.Close(); | 98 db_.Close(); |
| 114 } | 99 } |
| 115 | 100 |
| 116 void ActivityDatabase::KillDatabase() { | 101 void ActivityDatabase::KillDatabase() { |
| 117 db_.RazeAndClose(); | 102 db_.RazeAndClose(); |
| 118 } | 103 } |
| 119 | 104 |
| 120 } // namespace extensions | 105 } // namespace extensions |
| OLD | NEW |