Keep sync tasks alive as long as it's not finished

chrome/browser/chromeos/drive/sync_client.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 CHROME_BROWSER_CHROMEOS_DRIVE_SYNC_CLIENT_H_
 #define CHROME_BROWSER_CHROMEOS_DRIVE_SYNC_CLIENT_H_
 
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/callback.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "base/time/time.h"
 #include "chrome/browser/chromeos/drive/file_errors.h"
+#include "chrome/browser/chromeos/drive/job_scheduler.h"
 #include "chrome/browser/chromeos/drive/resource_metadata.h"
 
 namespace base {
 class SequencedTaskRunner;
 }
 
 namespace drive {
 
 class FileCacheEntry;
 class JobScheduler;
@@ skipping 59 matching lines @@
 
  private:
   // Types of sync tasks.
   enum SyncType {
     FETCH,  // Fetch a file from the Drive server.
     UPDATE,  // Updates an entry's metadata or content on the Drive server.
   };
 
   // States of sync tasks.
   enum SyncState {
-    PENDING,
-    RUNNING,
+    SUSPENDED,  // Task is currently inactive.
+    PENDING,  // Task is going to run.
+    RUNNING,  // Task is running.
   };
 
+  typedef std::pair<SyncType, std::string> SyncTaskKey;
+
   struct SyncTask {
     SyncTask();
     ~SyncTask();
     SyncState state;
-    base::Callback<base::Closure()> task;
+    ClientContext context;
+    base::Callback<base::Closure(const ClientContext& context)> task;
     bool should_run_again;
     base::Closure cancel_closure;
+    std::vector<SyncTaskKey> dependent_tasks;
   };
 
-  typedef std::map<std::pair<SyncType, std::string>, SyncTask> SyncTasks;
+  typedef std::map<SyncTaskKey, SyncTask> SyncTasks;
+
+  // Performs a FETCH task.
+  base::Closure PerformFetchTask(const std::string& local_id,
+                                 const ClientContext& context);
 
   // Adds a FETCH task.
   void AddFetchTaskInternal(const std::string& local_id,
                             const base::TimeDelta& delay);
 
+  // Performs a UPDATE task.
+  base::Closure PerformUpdateTask(const std::string& local_id,
+                                  const ClientContext& context);
+
   // Adds a UPDATE task.
   void AddUpdateTaskInternal(const ClientContext& context,
                              const std::string& local_id,
                              const base::TimeDelta& delay);
 
   // Adds the given task. If the same task is found, does nothing.
   void AddTask(const SyncTasks::key_type& key,
                const SyncTask& task,
                const base::TimeDelta& delay);
 
   // Called when a task is ready to start.
   void StartTask(const SyncTasks::key_type& key);
 
   // Called when the local IDs of files in the backlog are obtained.
   void OnGetLocalIdsOfBacklog(const std::vector<std::string>* to_fetch,
                               const std::vector<std::string>* to_update);
 
   // Adds fetch tasks.
   void AddFetchTasks(const std::vector<std::string>* local_ids);
 
-  // Erases the task and returns true if task is completed.
-  bool OnTaskComplete(SyncType type, const std::string& local_id);
+  // Called when a task is completed.
+  void OnTaskComplete(SyncType type,
+                      const std::string& local_id,
+                      FileError error);
 
   // Called when the file for |local_id| is fetched.
   void OnFetchFileComplete(const std::string& local_id,
                            FileError error,
                            const base::FilePath& local_path,
                            scoped_ptr<ResourceEntry> entry);
 
-  // Called when the entry is updated.
-  void OnUpdateComplete(const std::string& local_id, FileError error);
-
-  // Adds update tasks for |entries|.
-  void AddChildUpdateTasks(const ResourceEntryVector* entries,
-                           FileError error);
+  // Adds the task as a dependent task of the parent entry's task.
+  void AddDependentTask(SyncType type,
+                        const std::string& local_id,
+                        const ResourceEntry* entry,
+                        FileError error);
 
   scoped_refptr<base::SequencedTaskRunner> blocking_task_runner_;
   file_system::OperationObserver* operation_observer_;
   ResourceMetadata* metadata_;
   FileCache* cache_;
 
   // Used to fetch pinned files.
   scoped_ptr<file_system::DownloadOperation> download_operation_;
 
   // Used to update entry metadata.
@@ skipping 12 matching lines @@
   // invalidate its weak pointers before any other members are destroyed.
   base::WeakPtrFactory<SyncClient> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(SyncClient);
 };
 
 }  // namespace internal
 }  // namespace drive
 
 #endif  // CHROME_BROWSER_CHROMEOS_DRIVE_SYNC_CLIENT_H_