samples: Add task create & lease/acknowledge sample. (#1153)

* Add task create & lease/acknowledge sample.

* Add Appengine Queue Tasks sample.

tasks/snippets/src/main/java/com/example/Quickstart.java

@@ -0,0 +1,175 @@
+/*
+ * Copyright 2018 Google LLC
+ *
+ * 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.example;
+
+// [START tasks_quickstart]
+
+import com.google.cloud.tasks.v2beta2.AcknowledgeTaskRequest;
+import com.google.cloud.tasks.v2beta2.CloudTasksClient;
+import com.google.cloud.tasks.v2beta2.CreateTaskRequest;
+import com.google.cloud.tasks.v2beta2.LeaseTasksRequest;
+import com.google.cloud.tasks.v2beta2.LeaseTasksResponse;
+import com.google.cloud.tasks.v2beta2.PullMessage;
+import com.google.cloud.tasks.v2beta2.QueueName;
+import com.google.cloud.tasks.v2beta2.Task;
+import com.google.common.base.Strings;
+import com.google.protobuf.ByteString;
+import com.google.protobuf.Duration;
+
+import java.io.IOException;
+import java.nio.charset.Charset;
+
+import org.apache.commons.cli.CommandLine;
+import org.apache.commons.cli.CommandLineParser;
+import org.apache.commons.cli.DefaultParser;
+import org.apache.commons.cli.HelpFormatter;
+import org.apache.commons.cli.Option;
+import org.apache.commons.cli.Options;
+import org.apache.commons.cli.ParseException;
+
+public class Quickstart {
+  private static String GGOGLE_CLOUD_PROJECT_KEY = "GOOGLE_CLOUD_PROJECT";
+
+  private static Option PROJECT_ID_OPTION = Option.builder("project")
+      .longOpt("project-id")
+      .desc("The Google Cloud Project, if not set as GOOGLE_CLOUD_PROJECT env var.")
+      .hasArg()
+      .argName("project-id")
+      .type(String.class)
+      .build();
+
+  private static Option QUEUE_OPTION = Option.builder("queue")
+      .required()
+      .longOpt("queue")
+      .desc("The Cloud Tasks queue.")
+      .hasArg()
+      .argName("queue")
+      .type(String.class)
+      .build();
+
+  private static Option LOCATION_OPTION = Option.builder("location")
+      .required()
+      .longOpt("location")
+      .desc("The region in which your queue is running.")
+      .hasArg()
+      .argName("location")
+      .type(String.class)
+      .build();
+
+  public static void main(String... args) throws Exception {
+    Options options = new Options();
+    options.addOption(PROJECT_ID_OPTION);
+    options.addOption(QUEUE_OPTION);
+    options.addOption(LOCATION_OPTION);
+
+    if (args.length == 0) {
+      printUsage(options);
+      return;
+    }
+
+    CommandLineParser parser = new DefaultParser();
+    CommandLine params;
+    try {
+      params = parser.parse(options, args);
+    } catch (ParseException e) {
+      System.err.println("Invalid command line: " + e.getMessage());
+      printUsage(options);
+      return;
+    }
+
+    String projectId;
+    if (params.hasOption("project-id")) {
+      projectId = params.getOptionValue("project-id");
+    } else {
+      projectId = System.getenv(GGOGLE_CLOUD_PROJECT_KEY);
+    }
+    if (Strings.isNullOrEmpty(projectId)) {
+      printUsage(options);
+      return;
+    }
+
+    String queue = params.getOptionValue(QUEUE_OPTION.getOpt());
+    String location = params.getOptionValue(LOCATION_OPTION.getOpt());
+
+    switch (args[0]) {
+      default:
+        printUsage(options);
+        break;
+      case "create-task":
+        createTask(projectId, queue, location);
+        break;
+      case "lease-and-ack-task":
+        pullAndAckTask(projectId, queue, location);
+        break;
+    }
+  }
+
+  // [START cloud_tasks_create_task]
+  private static void createTask(String projectId, String queueName, String location)
+      throws IOException {
+    try (CloudTasksClient client = CloudTasksClient.create()) {
+      Task.Builder taskBuilder = Task
+          .newBuilder()
+          .setPullMessage(PullMessage.newBuilder().setPayload(
+              ByteString.copyFrom("a message for recipient", Charset.defaultCharset())));
+
+      Task newTask = client.createTask(CreateTaskRequest
+          .newBuilder()
+          .setParent(QueueName.of(projectId, location, queueName).toString())
+          .setTask(taskBuilder)
+          .build());
+      System.out.println("Task created: " + newTask.getName());
+    }
+  }
+  // [END cloud_tasks_create_task]
+
+  // [START cloud_tasks_lease_and_acknowledge_task]
+  private static void pullAndAckTask(String projectId, String queueName, String location) {
+    try (CloudTasksClient client = CloudTasksClient.create()) {
+      LeaseTasksRequest leaseReq = LeaseTasksRequest.newBuilder()
+          .setParent(QueueName.of(projectId, location, queueName).toString())
+          .setLeaseDuration(Duration.newBuilder().setSeconds(600))
+          .setMaxTasks(1)
+          .setResponseView(Task.View.FULL)
+          .build();
+      LeaseTasksResponse response = client.leaseTasks(leaseReq);
+      if (response.getTasksCount() == 0) {
+        System.out.println("No tasks found in queue.");
+        return;
+      }
+      Task task = response.getTasksList().get(0);
+      System.out.println("Leased task: " + task.getName());
+      AcknowledgeTaskRequest ackRequest = AcknowledgeTaskRequest
+          .newBuilder()
+          .setName(task.getName())
+          .setScheduleTime(task.getScheduleTime())
+          .build();
+      client.acknowledgeTask(ackRequest);
+      System.out.println("Acknowledged task: " + task.getName());
+    } catch (Exception e) {
+      System.out.println("Exception during PullAndAckTask: " + e.getMessage());
+    }
+  }
+  // [END cloud_tasks_lease_and_acknowledge_task]
+
+  private static void printUsage(Options options) {
+    HelpFormatter formatter = new HelpFormatter();
+    formatter.printHelp("client",
+        "A simple Cloud Tasks command line client.", options, "", true);
+  }
+}
+// [END tasks_quickstart]

tasks/snippets/src/test/java/com/example/QuickstartIT.java

@@ -0,0 +1,82 @@
+/*
+ * Copyright 2017 Google Inc.
+ *
+ * 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.example;
+
+import static com.google.common.truth.Truth.assertThat;
+import static org.junit.Assert.assertTrue;
+
+import com.google.cloud.tasks.v2beta2.CloudTasksClient;
+import com.google.cloud.tasks.v2beta2.QueueName;
+
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.PrintStream;
+
+import org.junit.AfterClass;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+/**
+ * Integration (system) tests for {@link Quickstart}.
+ */
+@RunWith(JUnit4.class)
+@SuppressWarnings("checkstyle:abbreviationaswordinname")
+public class QuickstartIT {
+  private static String queue_name = "my-pull-queue";
+  private static String location = "us-east1";
+  private ByteArrayOutputStream bout;
+  private PrintStream out;
+
+  @BeforeClass
+  public static void setUpClass() throws Exception {
+    ByteArrayOutputStream bout = new ByteArrayOutputStream();
+    PrintStream out = new PrintStream(bout);
+    System.setOut(out);
+  }
+
+  // Purge the task queue when tests done.
+  @AfterClass
+  public static void tearDownClass() throws IOException {
+    try (CloudTasksClient client = CloudTasksClient.create()) {
+      client.purgeQueue(QueueName.of(System.getenv("GOOGLE_CLOUD_PROJECT"), location, queue_name));
+    }
+  }
+
+  @Before
+  public void setUp() throws Exception {
+    bout = new ByteArrayOutputStream();
+    out = new PrintStream(bout);
+    System.setOut(out);
+  }
+
+  @Test
+  public void createTaskTest() throws Exception {
+    Quickstart.main("create-task", "--queue", queue_name, "--location", location);
+    assertThat(bout.toString()).contains("Task created: ");
+  }
+
+  @Test
+  public void leaseAndAcknowledge() throws Exception {
+    Quickstart.main("lease-and-ack-task", "--queue", queue_name, "--location", location);
+    assertThat(bout.toString()).contains("Leased task: ");
+    assertThat(bout.toString()).contains("Acknowledged task: ");
+  }
+}