add a download slices table into history download db

components/history/core/browser/download_database.h

 // 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.
 
 #ifndef COMPONENTS_HISTORY_CORE_BROWSER_DOWNLOAD_DATABASE_H_
 #define COMPONENTS_HISTORY_CORE_BROWSER_DOWNLOAD_DATABASE_H_
 
 #include <stddef.h>
 #include <stdint.h>
 
+#include <map>
 #include <string>
 #include <vector>
 
 #include "base/gtest_prod_util.h"
 #include "base/macros.h"
 #include "base/threading/platform_thread.h"
 #include "components/history/core/browser/download_types.h"
 
 namespace sql {
 class Connection;
 }
 
 namespace history {
 
+struct DownloadSliceInfo;
 struct DownloadRow;
 
 // Maintains a table of downloads.
 class DownloadDatabase {
  public:
   // Must call InitDownloadTable before using any other functions.
   DownloadDatabase(DownloadInterruptReason download_interrupt_reason_none,
                    DownloadInterruptReason download_interrupt_reason_crash);
   virtual ~DownloadDatabase();
 
   uint32_t GetNextDownloadId();
 
   // Get all the downloads from the database.
   void QueryDownloads(std::vector<DownloadRow>* results);
 
   // Update the state of one download. Returns true if successful.
   // Does not update |url|, |start_time|; uses |id| only
   // to select the row in the database table to update.
   bool UpdateDownload(const DownloadRow& data);
 
   // Create a new database entry for one download and return true if the
   // creation succeeded, false otherwise.
   bool CreateDownload(const DownloadRow& info);
 
   // Remove |id| from the database.
-  void RemoveDownload(uint32_t id);
+  void RemoveDownload(DownloadId id);
 
   size_t CountDownloads();
 
  protected:
   // Returns the database for the functions in this interface.
   virtual sql::Connection& GetDB() = 0;
 
   // Returns true if able to successfully add mime types to the downloads table.
   bool MigrateMimeType();
 
@@ skipping 43 matching lines @@
 
   // Fixes state of the download entries. Sometimes entries with IN_PROGRESS
   // state are not updated during browser shutdown (particularly when crashing).
   // On the next start such entries are considered interrupted with
   // interrupt reason |DOWNLOAD_INTERRUPT_REASON_CRASH|.  This function
   // fixes such entries.
   void EnsureInProgressEntriesCleanedUp();
 
   bool EnsureColumnExists(const std::string& name, const std::string& type);
 
-  void RemoveDownloadURLs(uint32_t id);
+  void RemoveDownloadURLs(DownloadId id);
+
+  // Creates a new download slice if it doesn't exist, or updates an existing
+  // one. Returns true on success, or false otherwise.
+  bool CreateOrUpdateDownloadSlice(const DownloadSliceInfo& info);
+
+  // Delete all the download slices associated with one DownloadRow.
+  void RemoveDownloadSlices(DownloadId id);
+
+  // Helper method to query the download slices for all the records in
+  // |download_row_map|.
+  using DownloadRowMap = std::map<DownloadId, DownloadRow*>;
+  void QueryDownloadSlices(DownloadRowMap* download_row_map);
 
   bool owning_thread_set_;
   base::PlatformThreadId owning_thread_;
 
   // Initialized to false on construction, and checked in all functional
   // routines post-migration in the database for a possible call to
   // CleanUpInProgressEntries().  This allows us to avoid
   // doing the cleanup until after any DB migration and unless we are
   // actually use the downloads database.
   bool in_progress_entry_cleanup_completed_;
 
   // Those constants are defined in the embedder and injected into the
   // database in the constructor. They represent the interrupt reason
   // to use for respectively an undefined value and in case of a crash.
   DownloadInterruptReason download_interrupt_reason_none_;
   DownloadInterruptReason download_interrupt_reason_crash_;
 
   DISALLOW_COPY_AND_ASSIGN(DownloadDatabase);
 };
 
 }  // namespace history
 
 #endif  // COMPONENTS_HISTORY_CORE_BROWSER_DOWNLOAD_DATABASE_H_