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 #ifndef CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_ | 5 #ifndef CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_ |
6 #define CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_ | 6 #define CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
| 9 #include "base/callback.h" |
9 #include "chrome/browser/history/history_types.h" | 10 #include "chrome/browser/history/history_types.h" |
10 | 11 |
11 struct DownloadHistoryInfo; | 12 struct DownloadHistoryInfo; |
12 class FilePath; | 13 class FilePath; |
13 | 14 |
14 namespace sql { | 15 namespace sql { |
15 class Connection; | 16 class Connection; |
16 } | 17 } |
17 | 18 |
18 namespace history { | 19 namespace history { |
19 | 20 |
20 // Maintains a table of downloads. | 21 // Maintains a table of downloads. |
21 class DownloadDatabase { | 22 class DownloadDatabase { |
22 public: | 23 public: |
23 // Must call InitDownloadTable before using any other functions. | 24 // Must call InitDownloadTable before using any other functions. |
24 DownloadDatabase(); | 25 DownloadDatabase(); |
25 virtual ~DownloadDatabase(); | 26 virtual ~DownloadDatabase(); |
26 | 27 |
27 // Get all the downloads from the database. | 28 // Get all the downloads from the database. |
28 void QueryDownloads(std::vector<DownloadHistoryInfo>* results); | 29 void QueryDownloads(DownloadQueryParameters* params); |
29 | 30 |
30 // Update the state of one download. Returns true if successful. | 31 // Update the state of one download. Returns true if successful. |
31 bool UpdateDownload(int64 received_bytes, int32 state, DownloadID db_handle); | 32 bool UpdateDownload(int64 received_bytes, int32 state, DownloadID id); |
32 | 33 |
33 // Update the path of one download. Returns true if successful. | 34 // Update the path of one download. Returns true if successful. |
34 bool UpdateDownloadPath(const FilePath& path, DownloadID db_handle); | 35 bool UpdateDownloadPath(const FilePath& path, DownloadID id); |
35 | 36 |
36 // Fixes state of the download entries. Sometimes entries with IN_PROGRESS | 37 // Fixes state of the download entries. Sometimes entries with IN_PROGRESS |
37 // state are not updated during browser shutdown (particularly when crashing). | 38 // state are not updated during browser shutdown (particularly when crashing). |
38 // On the next start such entries are considered canceled. This functions | 39 // On the next start such entries are considered canceled. This functions |
39 // fixes such entries. | 40 // fixes such entries. |
40 bool CleanUpInProgressEntries(); | 41 bool CleanUpInProgressEntries(); |
41 | 42 |
42 // Create a new database entry for one download and return its primary db id. | 43 // Create a new database entry for one download and return its primary db id. |
43 int64 CreateDownload(const DownloadHistoryInfo& info); | 44 bool CreateDownload(const DownloadHistoryInfo& info); |
44 | 45 |
45 // Remove a download from the database. | 46 // Remove a download from the database. |
46 void RemoveDownload(DownloadID db_handle); | 47 void RemoveDownload(DownloadID id); |
47 | 48 |
48 // Remove all completed downloads that started after |remove_begin| | 49 // Remove all completed downloads that started after |remove_begin| |
49 // (inclusive) and before |remove_end|. You may use null Time values | 50 // (inclusive) and before |remove_end|. You may use null Time values |
50 // to do an unbounded delete in either direction. This function ignores | 51 // to do an unbounded delete in either direction. This function ignores |
51 // all downloads that are in progress or are waiting to be cancelled. | 52 // all downloads that are in progress or are waiting to be cancelled. |
52 void RemoveDownloadsBetween(base::Time remove_begin, base::Time remove_end); | 53 void RemoveDownloadsBetween(base::Time remove_begin, base::Time remove_end); |
53 | 54 |
54 protected: | 55 protected: |
55 // Returns the database for the functions in this interface. | 56 // Returns the database for the functions in this interface. |
56 virtual sql::Connection& GetDB() = 0; | 57 virtual sql::Connection& GetDB() = 0; |
57 | 58 |
58 // Creates the downloads table if needed. | 59 // Creates the downloads table if needed. |
59 bool InitDownloadTable(); | 60 bool InitDownloadTable(); |
60 | 61 |
61 // Used to quickly clear the downloads. First you would drop it, then you | 62 // Used to quickly clear the downloads. First you would drop it, then you |
62 // would re-initialize it. | 63 // would re-initialize it. |
63 bool DropDownloadTable(); | 64 bool DropDownloadTable(); |
64 | 65 |
65 private: | 66 private: |
| 67 bool MaybeUpgradeTable(DownloadQueryParameters::GetNextIdThunk get_next_id); |
| 68 |
66 DISALLOW_COPY_AND_ASSIGN(DownloadDatabase); | 69 DISALLOW_COPY_AND_ASSIGN(DownloadDatabase); |
67 }; | 70 }; |
68 | 71 |
69 } // namespace history | 72 } // namespace history |
70 | 73 |
71 #endif // CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_ | 74 #endif // CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_ |
OLD | NEW |