Migrate data from old localstorage format to new format.

content/browser/leveldb_wrapper_impl.h

 // Copyright 2016 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 CONTENT_BROWSER_LEVELDB_WRAPPER_IMPL_H_
 #define CONTENT_BROWSER_LEVELDB_WRAPPER_IMPL_H_
 
 #include <map>
 #include <memory>
 #include <string>
@@ skipping 14 matching lines @@
 // pointers can be bound to the same object. The wrapper adds a couple of
 // features not found directly in leveldb.
 // 1) Adds the given prefix, if any, to all keys. This allows the sharing of one
 //    database across many, possibly untrusted, consumers and ensuring that they
 //    can't access each other's values.
 // 2) Enforces a max_size constraint.
 // 3) Informs observers when values scoped by prefix are modified.
 // 4) Throttles requests to avoid overwhelming the disk.
 class CONTENT_EXPORT LevelDBWrapperImpl : public mojom::LevelDBWrapper {
  public:
-  using PrepareToCommitCallback =
-      base::Callback<std::vector<leveldb::mojom::BatchedOperationPtr>(
-          const LevelDBWrapperImpl&)>;
+  using ValueMap = std::map<std::vector<uint8_t>, std::vector<uint8_t>>;
+  using ValueMapCallback = base::OnceCallback<void(std::unique_ptr<ValueMap>)>;
+
+  class CONTENT_EXPORT Delegate {
+   public:
+    virtual void OnNoBindings() = 0;
+    virtual std::vector<leveldb::mojom::BatchedOperationPtr>
+    PrepareToCommit() = 0;
+    virtual void DidCommit(leveldb::mojom::DatabaseError error) = 0;
+    // Called during loading if no data was found. Needs to call |callback|.
+    virtual void MigrateData(ValueMapCallback callback);
+  };
 
   // |no_bindings_callback| will be called when this object has no more
   // bindings and all pending modifications have been processed.
   LevelDBWrapperImpl(leveldb::mojom::LevelDBDatabase* database,
                      const std::string& prefix,
                      size_t max_size,
                      base::TimeDelta default_commit_delay,
                      int max_bytes_per_hour,
                      int max_commits_per_hour,
-                     const base::Closure& no_bindings_callback,
-                     const PrepareToCommitCallback& prepare_to_commit_callback);
+                     Delegate* delegate);
   ~LevelDBWrapperImpl() override;
 
   void Bind(mojom::LevelDBWrapperRequest request);
 
+  bool empty() const { return bytes_used_ == 0; }
   size_t bytes_used() const { return bytes_used_; }
 
   // Commence aggressive flushing. This should be called early during startup,
   // before any localStorage writing. Currently scheduled writes will not be
   // rescheduled and will be flushed at the scheduled time after which
   // aggressive flushing will commence.
   static void EnableAggressiveCommitDelay();
 
   // Commits any uncommitted data to the database as soon as possible. This
   // usually means data will be committed immediately, but if we're currently
@@ skipping 12 matching lines @@
               const DeleteCallback& callback) override;
   void DeleteAll(const std::string& source,
                  const DeleteAllCallback& callback) override;
   void Get(const std::vector<uint8_t>& key,
            const GetCallback& callback) override;
   void GetAll(
       mojom::LevelDBWrapperGetAllCallbackAssociatedPtrInfo complete_callback,
       const GetAllCallback& callback) override;
 
  private:
-  using ValueMap = std::map<std::vector<uint8_t>, std::vector<uint8_t>>;
-
   // Used to rate limit commits.
   class RateLimiter {
    public:
     RateLimiter(size_t desired_rate, base::TimeDelta time_quantum);
 
     void add_samples(size_t samples) { samples_ += samples;  }
 
     // Computes the total time needed to process the total samples seen
     // at the desired rate.
     base::TimeDelta ComputeTimeNeeded() const;
@@ skipping 13 matching lines @@
   struct CommitBatch {
     bool clear_all_first;
     std::set<std::vector<uint8_t>> changed_keys;
 
     CommitBatch();
     ~CommitBatch();
   };
 
   void OnConnectionError();
   void LoadMap(const base::Closure& completion_callback);
-  void OnLoadComplete(leveldb::mojom::DatabaseError status,
-                      std::vector<leveldb::mojom::KeyValuePtr> data);
+  void OnMapLoaded(leveldb::mojom::DatabaseError status,
+                   std::vector<leveldb::mojom::KeyValuePtr> data);
+  void OnGotMigrationData(std::unique_ptr<ValueMap> data);
+  void OnLoadComplete();
   void CreateCommitBatchIfNeeded();
   void StartCommitTimer();
   base::TimeDelta ComputeCommitDelay() const;
   void CommitChanges();
   void OnCommitComplete(leveldb::mojom::DatabaseError error);
 
   std::vector<uint8_t> prefix_;
   mojo::BindingSet<mojom::LevelDBWrapper> bindings_;
   mojo::AssociatedInterfacePtrSet<mojom::LevelDBObserver> observers_;
-  base::Closure no_bindings_callback_;
-  PrepareToCommitCallback prepare_to_commit_callback_;
+  Delegate* delegate_;
   leveldb::mojom::LevelDBDatabase* database_;
   std::unique_ptr<ValueMap> map_;
   std::vector<base::Closure> on_load_complete_tasks_;
   size_t bytes_used_;
   size_t max_size_;
   base::TimeTicks start_time_;
   base::TimeDelta default_commit_delay_;
   RateLimiter data_rate_limiter_;
   RateLimiter commit_rate_limiter_;
   int commit_batches_in_flight_ = 0;
   std::unique_ptr<CommitBatch> commit_batch_;
   base::WeakPtrFactory<LevelDBWrapperImpl> weak_ptr_factory_;
 
   static bool s_aggressive_flushing_enabled_;
 
   DISALLOW_COPY_AND_ASSIGN(LevelDBWrapperImpl);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_LEVELDB_WRAPPER_IMPL_H_