Updated pull to lease.

GoogleCloudPlatform · Jan 16, 2018 · e4305d3 · e4305d3
1 parent c02c3d3
commit e4305d3
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/cloud-tasks/src/main/java/com.example.cloudtasks/PullQueue.java b/cloud-tasks/src/main/java/com.example.cloudtasks/PullQueue.java
@@ -23,12 +23,9 @@
 import com.google.api.client.json.jackson2.JacksonFactory;
 import com.google.api.services.cloudtasks.v2beta2.CloudTasks;
 import com.google.api.services.cloudtasks.v2beta2.CloudTasksScopes;
-import com.google.api.services.cloudtasks.v2beta2.model.AcknowledgeTaskRequest;
-import com.google.api.services.cloudtasks.v2beta2.model.CreateTaskRequest;
-import com.google.api.services.cloudtasks.v2beta2.model.PullMessage;
-import com.google.api.services.cloudtasks.v2beta2.model.PullTasksRequest;
-import com.google.api.services.cloudtasks.v2beta2.model.PullTasksResponse;
-import com.google.api.services.cloudtasks.v2beta2.model.Task;
+import com.google.api.services.cloudtasks.v2beta2.model.*;
+import com.google.api.services.cloudtasks.v2beta2.model.LeaseTasksRequest;
+import com.google.api.services.cloudtasks.v2beta2.model.LeaseTasksResponse;
 import com.google.common.io.BaseEncoding;
 import java.io.IOException;
 import net.sourceforge.argparse4j.ArgumentParsers;
@@ -110,15 +107,15 @@ private static Task pullTask(
     CloudTasks client = createAuthorizedClient();
 
     // Create the PullTasksRequest
-    PullTasksRequest request = new PullTasksRequest().setMaxTasks(1).setLeaseDuration("600s");
+    LeaseTasksRequest request = new LeaseTasksRequest().setMaxTasks(1).setLeaseDuration("600s");
 
     //Execute the request and return the pulled task
-    PullTasksResponse response = client
+    LeaseTasksResponse response = client
         .projects()
         .locations()
         .queues()
         .tasks()
-        .pull(queueName, request)
+        .lease(queueName, request)
         .execute();
     return response.getTasks().get(0);
   }