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 <set> | 9 #include <set> |
10 | 10 |
(...skipping 16 matching lines...) Expand all Loading... |
27 // Must call InitDownloadTable before using any other functions. | 27 // Must call InitDownloadTable before using any other functions. |
28 DownloadDatabase(); | 28 DownloadDatabase(); |
29 virtual ~DownloadDatabase(); | 29 virtual ~DownloadDatabase(); |
30 | 30 |
31 int next_download_id() const { return next_id_; } | 31 int next_download_id() const { return next_id_; } |
32 | 32 |
33 // Get all the downloads from the database. | 33 // Get all the downloads from the database. |
34 void QueryDownloads(std::vector<DownloadPersistentStoreInfo>* results); | 34 void QueryDownloads(std::vector<DownloadPersistentStoreInfo>* results); |
35 | 35 |
36 // Update the state of one download. Returns true if successful. | 36 // Update the state of one download. Returns true if successful. |
37 bool UpdateDownload(int64 received_bytes, int32 state, DownloadID db_handle); | 37 bool UpdateDownload(int64 received_bytes, int32 state, |
| 38 const base::Time& end_time, DownloadID db_handle); |
38 | 39 |
39 // Update the path of one download. Returns true if successful. | 40 // Update the path of one download. Returns true if successful. |
40 bool UpdateDownloadPath(const FilePath& path, DownloadID db_handle); | 41 bool UpdateDownloadPath(const FilePath& path, DownloadID db_handle); |
41 | 42 |
42 // Fixes state of the download entries. Sometimes entries with IN_PROGRESS | 43 // Fixes state of the download entries. Sometimes entries with IN_PROGRESS |
43 // state are not updated during browser shutdown (particularly when crashing). | 44 // state are not updated during browser shutdown (particularly when crashing). |
44 // On the next start such entries are considered canceled. This functions | 45 // On the next start such entries are considered canceled. This functions |
45 // fixes such entries. | 46 // fixes such entries. |
46 bool CleanUpInProgressEntries(); | 47 bool CleanUpInProgressEntries(); |
47 | 48 |
48 // Create a new database entry for one download and return its primary db id. | 49 // Create a new database entry for one download and return its primary db id. |
49 int64 CreateDownload(const DownloadPersistentStoreInfo& info); | 50 int64 CreateDownload(const DownloadPersistentStoreInfo& info); |
50 | 51 |
51 // Remove a download from the database. | 52 // Remove a download from the database. |
52 void RemoveDownload(DownloadID db_handle); | 53 void RemoveDownload(DownloadID db_handle); |
53 | 54 |
54 // Remove all completed downloads that started after |remove_begin| | 55 // Remove all completed downloads that started after |remove_begin| |
55 // (inclusive) and before |remove_end|. You may use null Time values | 56 // (inclusive) and before |remove_end|. You may use null Time values |
56 // to do an unbounded delete in either direction. This function ignores | 57 // to do an unbounded delete in either direction. This function ignores |
57 // all downloads that are in progress or are waiting to be cancelled. | 58 // all downloads that are in progress or are waiting to be cancelled. |
58 void RemoveDownloadsBetween(base::Time remove_begin, base::Time remove_end); | 59 void RemoveDownloadsBetween(base::Time remove_begin, base::Time remove_end); |
59 | 60 |
| 61 // Set the |opened| flag to true. |
| 62 void MarkDownloadOpened(DownloadID db_handle); |
| 63 |
60 protected: | 64 protected: |
61 // Returns the database for the functions in this interface. | 65 // Returns the database for the functions in this interface. |
62 virtual sql::Connection& GetDB() = 0; | 66 virtual sql::Connection& GetDB() = 0; |
63 | 67 |
64 // Creates the downloads table if needed. | 68 // Creates the downloads table if needed. |
65 bool InitDownloadTable(); | 69 bool InitDownloadTable(); |
66 | 70 |
67 // Used to quickly clear the downloads. First you would drop it, then you | 71 // Used to quickly clear the downloads. First you would drop it, then you |
68 // would re-initialize it. | 72 // would re-initialize it. |
69 bool DropDownloadTable(); | 73 bool DropDownloadTable(); |
70 | 74 |
| 75 bool MaybeUpgradeDownloadsSchema(); |
| 76 |
| 77 bool CreateDownloadsTable(); |
| 78 |
| 79 enum DownloadsSchemaVersion { |
| 80 DOWNLOADS_SCHEMA_VERSION_UNINITIALIZED = -1, |
| 81 DOWNLOADS_SCHEMA_VERSION_BASE = 0, |
| 82 DOWNLOADS_SCHEMA_VERSION_OPENED = 1, |
| 83 // Add new schema versions here with explicit numbers, update kCurrent, and |
| 84 // support your new version in CreateDownloadsTable, |
| 85 // MaybeUpgradeDownloadsSchema, and QueryDownloads. |
| 86 }; |
| 87 static const DownloadsSchemaVersion kCurrentDownloadsSchemaVersion = |
| 88 DOWNLOADS_SCHEMA_VERSION_OPENED; |
| 89 |
| 90 DownloadsSchemaVersion downloads_schema_version() const { |
| 91 return schema_version_; |
| 92 } |
| 93 |
71 private: | 94 private: |
72 // TODO(rdsmith): Remove after http://crbug.com/96627 has been resolved. | 95 // TODO(rdsmith): Remove after http://crbug.com/96627 has been resolved. |
73 std::set<int64> returned_ids_; | 96 std::set<int64> returned_ids_; |
74 bool owning_thread_set_; | 97 bool owning_thread_set_; |
75 base::PlatformThreadId owning_thread_; | 98 base::PlatformThreadId owning_thread_; |
76 | 99 |
77 int next_id_; | 100 int next_id_; |
78 sql::MetaTable meta_table_; | 101 sql::MetaTable meta_table_; |
| 102 DownloadsSchemaVersion schema_version_; |
79 | 103 |
80 DISALLOW_COPY_AND_ASSIGN(DownloadDatabase); | 104 DISALLOW_COPY_AND_ASSIGN(DownloadDatabase); |
81 }; | 105 }; |
82 | 106 |
83 } // namespace history | 107 } // namespace history |
84 | 108 |
85 #endif // CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_ | 109 #endif // CHROME_BROWSER_HISTORY_DOWNLOAD_DATABASE_H_ |
OLD | NEW |