|
23 | 23 | import com.google.api.client.json.jackson2.JacksonFactory; |
24 | 24 | import com.google.api.services.cloudtasks.v2beta2.CloudTasks; |
25 | 25 | import com.google.api.services.cloudtasks.v2beta2.CloudTasksScopes; |
26 | | -import com.google.api.services.cloudtasks.v2beta2.model.AcknowledgeTaskRequest; |
27 | | -import com.google.api.services.cloudtasks.v2beta2.model.CreateTaskRequest; |
28 | | -import com.google.api.services.cloudtasks.v2beta2.model.PullMessage; |
29 | | -import com.google.api.services.cloudtasks.v2beta2.model.PullTasksRequest; |
30 | | -import com.google.api.services.cloudtasks.v2beta2.model.PullTasksResponse; |
31 | | -import com.google.api.services.cloudtasks.v2beta2.model.Task; |
| 26 | +import com.google.api.services.cloudtasks.v2beta2.model.*; |
| 27 | +import com.google.api.services.cloudtasks.v2beta2.model.LeaseTasksRequest; |
| 28 | +import com.google.api.services.cloudtasks.v2beta2.model.LeaseTasksResponse; |
32 | 29 | import com.google.common.io.BaseEncoding; |
33 | 30 | import java.io.IOException; |
34 | 31 | import net.sourceforge.argparse4j.ArgumentParsers; |
@@ -110,15 +107,15 @@ private static Task pullTask( |
110 | 107 | CloudTasks client = createAuthorizedClient(); |
111 | 108 |
|
112 | 109 | // Create the PullTasksRequest |
113 | | - PullTasksRequest request = new PullTasksRequest().setMaxTasks(1).setLeaseDuration("600s"); |
| 110 | + LeaseTasksRequest request = new LeaseTasksRequest().setMaxTasks(1).setLeaseDuration("600s"); |
114 | 111 |
|
115 | 112 | //Execute the request and return the pulled task |
116 | | - PullTasksResponse response = client |
| 113 | + LeaseTasksResponse response = client |
117 | 114 | .projects() |
118 | 115 | .locations() |
119 | 116 | .queues() |
120 | 117 | .tasks() |
121 | | - .pull(queueName, request) |
| 118 | + .lease(queueName, request) |
122 | 119 | .execute(); |
123 | 120 | return response.getTasks().get(0); |
124 | 121 | } |
|
0 commit comments