Skip to content

Commit 5586054

Browse files
committed
Updated pull to lease.
1 parent c02c3d3 commit 5586054

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

cloud-tasks/src/main/java/com.example.cloudtasks/PullQueue.java

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,12 +23,9 @@
2323
import com.google.api.client.json.jackson2.JacksonFactory;
2424
import com.google.api.services.cloudtasks.v2beta2.CloudTasks;
2525
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;
3229
import com.google.common.io.BaseEncoding;
3330
import java.io.IOException;
3431
import net.sourceforge.argparse4j.ArgumentParsers;
@@ -110,15 +107,15 @@ private static Task pullTask(
110107
CloudTasks client = createAuthorizedClient();
111108

112109
// Create the PullTasksRequest
113-
PullTasksRequest request = new PullTasksRequest().setMaxTasks(1).setLeaseDuration("600s");
110+
LeaseTasksRequest request = new LeaseTasksRequest().setMaxTasks(1).setLeaseDuration("600s");
114111

115112
//Execute the request and return the pulled task
116-
PullTasksResponse response = client
113+
LeaseTasksResponse response = client
117114
.projects()
118115
.locations()
119116
.queues()
120117
.tasks()
121-
.pull(queueName, request)
118+
.lease(queueName, request)
122119
.execute();
123120
return response.getTasks().get(0);
124121
}

0 commit comments

Comments
 (0)