Docserver: Persist stat cache for versioned file systems

chrome/common/extensions/docs/server2/refresh_tracker.py

+# Copyright 2014 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.
+
+from future import Future
+
+
+class _RefreshWorkOrder(object):
+  '''A set of tasks that must be completed in a refresh cycle.'''
+  def __init__(self, tasks):
+    self.tasks = set(tasks)
+
+
+def _RefreshTaskCompletion(object):
+  '''Marks the completion of a single named task for some refresh cycle.'''
+  def __init__(self, task):
+    self.task = task
+
+
+def _GetTaskCompletionKey(refresh_id, task):
+  return '%s@%s' % (refresh_id, task)
+
+
+class RefreshTracker(object):

[not at google - send to devlin, 2014/10/22 18:08:48]

Tests for this class?

[Ken Rockot(use gerrit already), 2014/10/23 22:36:16]

Done.

+  '''Manages and tracks the progress of data refresh cycles.'''
+  def __init__(self, object_store_creator):
+    self._work_orders = object_store_creator.Create(_RefreshWorkOrder,
+        category='refresh_work_orders')
+    self._task_completions = object_store_creator.Create(_RefreshTaskCompletion,
+        category='refresh_task_completions')
+
+  def _GetWorkOrder(self, refresh_id):
+    '''Retrieves the work order for a refresh cycle identified by |id|.'''
+    return self._work_orders.Get(refresh_id)
+
+  def _GetTaskCompletion(self, refresh_id, task):
+    '''Retrieves the set of task completions for a refresh cycle identified by
+    |id|.'''
+    return self._task_completions.Get(_GetTaskCompletionKey(refresh_id, task))
+
+  def RegisterRefresh(self, refresh_id, tasks):
+    work_order = _RefreshWorkOrder(tasks)
+    return self._work_orders.Set(refresh_id, work_order)
+
+  def GetRefreshComplete(self, refresh_id):
+    '''Determines if a refresh cycle identified by |id| is complete. Returns
+    a |Future| which resolves to either |True| or |False|.'''
+    def is_work_order_complete(work_order):
+      # If the identified refresh cycle is not known, always return False.
+      if work_order is None:
+        return False
+
+      futures = [self._GetTaskCompletion(refresh_id, task)
+                 for task in work_order.tasks]
+      return All(futures).Then(callback=lambda results: None not in results)
+
+    return self._GetWorkOrder(refresh_id).Then(is_work_order_complete)
+
+  def MarkTaskComplete(self, refresh_id, task):
+    return self._task_completions.Set(_GetTaskCompletionKey(refresh_id, task),
+        _RefreshTaskCompletion(task))