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 "chrome/browser/predictors/predictor_table_base.h" | 5 #include "chrome/browser/predictors/predictor_table_base.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "content/public/browser/browser_thread.h" | 8 #include "content/public/browser/browser_thread.h" |
9 #include "sql/connection.h" | 9 #include "sql/connection.h" |
10 | 10 |
11 using content::BrowserThread; | 11 using content::BrowserThread; |
12 | 12 |
13 namespace predictors { | 13 namespace predictors { |
14 | 14 |
15 PredictorTableBase::PredictorTableBase() : db_(NULL) { | 15 PredictorTableBase::PredictorTableBase() : db_(NULL) { |
16 } | 16 } |
17 | 17 |
18 PredictorTableBase::~PredictorTableBase() { | 18 PredictorTableBase::~PredictorTableBase() { |
19 } | 19 } |
20 | 20 |
21 void PredictorTableBase::Initialize(sql::Connection* db) { | 21 void PredictorTableBase::Initialize(sql::Connection* db) { |
22 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); | 22 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); |
23 db_ = db; | 23 db_ = db; |
24 CreateTableIfNonExistent(); | 24 CreateTableIfNonExistent(); |
25 } | 25 } |
26 | 26 |
| 27 void PredictorTableBase::SetCancelled() { |
| 28 cancelled_.Set(); |
| 29 } |
| 30 |
27 sql::Connection* PredictorTableBase::DB() { | 31 sql::Connection* PredictorTableBase::DB() { |
28 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); | 32 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); |
29 return db_; | 33 return db_; |
30 } | 34 } |
31 | 35 |
32 void PredictorTableBase::ResetDB() { | 36 void PredictorTableBase::ResetDB() { |
33 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); | 37 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); |
34 db_ = NULL; | 38 db_ = NULL; |
35 } | 39 } |
36 | 40 |
37 bool PredictorTableBase::CantAccessDatabase() { | 41 bool PredictorTableBase::CantAccessDatabase() { |
38 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); | 42 CHECK(BrowserThread::CurrentlyOn(BrowserThread::DB)); |
39 return cancelled_.IsSet() || !db_; | 43 return cancelled_.IsSet() || !db_; |
40 } | 44 } |
41 | 45 |
42 } // namespace predictors | 46 } // namespace predictors |
OLD | NEW |