Add ability to parse HTTP-format moratorium times

(since pretty much every user wants this).

1 files changed, 23 insertions, 1 deletions

diff --git a/common/java/com/android/common/OperationScheduler.java b/common/java/com/android/common/OperationScheduler.java
index 71b22ce..c7b12d3 100644
--- a/common/java/com/android/common/OperationScheduler.java
+++ b/common/java/com/android/common/OperationScheduler.java
@@ -192,7 +192,6 @@ public class OperationScheduler {
 
     /**
      * Forbid any operations until after a certain (absolute) time.
-     * Commonly used when a server returns a "Retry-After:" type directive.
      * Limited by {@link #Options.maxMoratoriumMillis}.
      *
      * @param millis wall clock time ({@link System#currentTimeMillis()}) to
@@ -206,6 +205,29 @@ public class OperationScheduler {
     }
 
     /**
+     * Forbid any operations until after a certain time, as specified in
+     * the format used by the HTTP "Retry-After" header.
+     * Limited by {@link #Options.maxMoratoriumMillis}.
+     *
+     * @param retryAfter moratorium time in HTTP format
+     * @return true if a time was successfully parsed
+     */
+    public boolean setMoratoriumTimeHttp(String retryAfter) {
+        try {
+            long ms = Long.valueOf(retryAfter) * 1000;
+            setMoratoriumTimeMillis(ms + System.currentTimeMillis());
+            return true;
+        } catch (NumberFormatException nfe) {
+            try {
+                setMoratoriumTimeMillis(HttpDateTime.parse(retryAfter));
+                return true;
+            } catch (IllegalArgumentException iae) {
+                return false;
+            }
+        }
+    }
+
+    /**
      * Enable or disable all operations.  When disabled, all calls to
      * {@link #getNextTimeMillis()} return {@link Long#MAX_VALUE}.
      * Commonly used when data network availability goes up and down.