blob: 6dec314affd2c1a466aed595d45ba4cf379d59d5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
// Copyright (c) 2012 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.
#include <string>
#include "base/logging.h"
#include "base/string_util.h"
#include "chrome/browser/extensions/activity_database.h"
#include "chrome/browser/history/url_database.h"
#include "content/public/browser/browser_thread.h"
#include "sql/transaction.h"
#if defined(OS_MACOSX)
#include "base/mac/mac_util.h"
#endif
using content::BrowserThread;
namespace extensions {
ActivityDatabase::ActivityDatabase() : initialized_(false) {}
ActivityDatabase::~ActivityDatabase() {
Close(); // Safe to call Close() even if Open() never happened.
}
void ActivityDatabase::SetErrorDelegate(sql::ErrorDelegate* error_delegate) {
db_.set_error_delegate(error_delegate);
}
void ActivityDatabase::Init(const FilePath& db_name) {
db_.set_page_size(4096);
db_.set_cache_size(32);
if (!db_.Open(db_name)) {
LOG(ERROR) << db_.GetErrorMessage();
return LogInitFailure();
}
// Wrap the initialization in a transaction so that the db doesn't
// get corrupted if init fails/crashes.
sql::Transaction committer(&db_);
if (!committer.Begin())
return LogInitFailure();
#if defined(OS_MACOSX)
// Exclude the database from backups.
base::mac::SetFileBackupExclusion(db_name);
#endif
db_.Preload();
// Create the UrlAction database.
if (InitializeTable(UrlAction::kTableName, UrlAction::kTableStructure) !=
sql::INIT_OK)
return LogInitFailure();
// Create the APIAction database.
if (InitializeTable(APIAction::kTableName, APIAction::kTableStructure)
!= sql::INIT_OK)
return LogInitFailure();
// Create the BlockedAction database.
if (InitializeTable(BlockedAction::kTableName, BlockedAction::kTableStructure)
!= sql::INIT_OK)
return LogInitFailure();
sql::InitStatus stat = committer.Commit() ? sql::INIT_OK : sql::INIT_FAILURE;
if (stat == sql::INIT_OK)
initialized_ = true;
else
return LogInitFailure();
}
void ActivityDatabase::LogInitFailure() {
LOG(ERROR) << "Couldn't initialize the activity log database.";
}
sql::InitStatus ActivityDatabase::InitializeTable(const char* table_name,
const char* table_structure) {
if (!db_.DoesTableExist(table_name)) {
char table_creator[1000];
base::snprintf(table_creator,
arraysize(table_creator),
"CREATE TABLE %s %s", table_name, table_structure);
if (!db_.Execute(table_creator))
return sql::INIT_FAILURE;
}
return sql::INIT_OK;
}
void ActivityDatabase::RecordAction(scoped_refptr<Action> action) {
if (initialized_)
action->Record(&db_);
}
void ActivityDatabase::BeginTransaction() {
db_.BeginTransaction();
}
void ActivityDatabase::CommitTransaction() {
db_.CommitTransaction();
}
void ActivityDatabase::RollbackTransaction() {
db_.RollbackTransaction();
}
bool ActivityDatabase::Raze() {
return db_.Raze();
}
void ActivityDatabase::Close() {
db_.Close();
}
void ActivityDatabase::KillDatabase() {
db_.RollbackTransaction();
db_.Raze();
db_.Close();
}
} // namespace extensions
|
