Import code from internal repository to git

java/com/google/domain/registry/flows/async/AsyncFlowUtils.java

@@ -0,0 +1,60 @@
+// Copyright 2016 Google Inc. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.google.domain.registry.flows.async;
+
+import static com.google.domain.registry.request.Actions.getPathForAction;
+
+import com.google.appengine.api.taskqueue.Queue;
+import com.google.appengine.api.taskqueue.QueueFactory;
+import com.google.appengine.api.taskqueue.TaskHandle;
+import com.google.appengine.api.taskqueue.TaskOptions;
+import com.google.appengine.api.taskqueue.TaskOptions.Method;
+import com.google.common.annotations.VisibleForTesting;
+import com.google.common.collect.ImmutableMap;
+import com.google.domain.registry.mapreduce.MapreduceAction;
+import com.google.domain.registry.util.FormattingLogger;
+
+import org.joda.time.Duration;
+
+import java.util.Map.Entry;
+
+/** Utility methods specific to async flows. */
+public final class AsyncFlowUtils {
+
+  private static final FormattingLogger logger = FormattingLogger.getLoggerForCallerClass();
+
+  @VisibleForTesting
+  public static final String ASYNC_FLOW_QUEUE_NAME = "flows-async";  // See queue.xml.
+
+  private AsyncFlowUtils() {}
+
+  /** Enqueues a mapreduce action to perform an async flow operation. */
+  public static TaskHandle enqueueMapreduceAction(
+      Class<? extends MapreduceAction> action,
+      ImmutableMap<String, String> params,
+      Duration executionDelay) {
+    Queue queue = QueueFactory.getQueue(ASYNC_FLOW_QUEUE_NAME);
+    String path = getPathForAction(action);
+    logger.infofmt("Enqueueing async mapreduce action with path %s and params %s", path, params);
+    TaskOptions options = TaskOptions.Builder
+        .withUrl(path)
+        .countdownMillis(executionDelay.getMillis())
+        .method(Method.GET);
+    for (Entry<String, String> entry : params.entrySet()) {
+      options.param(entry.getKey(), entry.getValue());
+    }
+    return queue.add(options);
+  }
+}