Skip to content

Commit f164cd0

Browse files
feat: implement renew_lease for sqs
1 parent 1d13d60 commit f164cd0

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

taskqueue/aws_queue_api.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -143,11 +143,16 @@ def add_insert_count(self, ct):
143143
def rezero(self):
144144
pass
145145

146-
def renew_lease(self, seconds):
147-
raise NotImplementedError()
146+
@retry
147+
def renew_lease(self, task, seconds):
148+
self.sqs.change_message_visibility(
149+
QueueUrl=self.qurl,
150+
ReceiptHandle=task.id,
151+
VisibilityTimeout=seconds,
152+
)
148153

149-
def cancel_lease(self, rhandle):
150-
raise NotImplementedError()
154+
def cancel_lease(self, task):
155+
self.renew_lease(task, 0)
151156

152157
def release_all(self):
153158
raise NotImplementedError()

0 commit comments

Comments
 (0)