diff options
author | Google Team <no-reply@google.com> | 2023-07-05 17:30:25 -0700 |
---|---|---|
committer | Copybara-Service <copybara-worker@google.com> | 2023-07-05 17:30:55 -0700 |
commit | eeee83cdf826e4339a0902a0e0dea97f577393d8 (patch) | |
tree | da5634c745040a2c1fd2567afb8fa3636bd518f3 | |
parent | 59bae0046cc215ff8bb6a82bad988b53cd672086 (diff) | |
download | tradefed_cluster-eeee83cdf826e4339a0902a0e0dea97f577393d8.tar.gz |
Internal change
PiperOrigin-RevId: 545829435
Change-Id: I4ad048cdd60d17f64865e883af6b118530c3e6bb
-rw-r--r-- | tradefed_cluster/command_task_store.py | 8 |
1 files changed, 1 insertions, 7 deletions
diff --git a/tradefed_cluster/command_task_store.py b/tradefed_cluster/command_task_store.py index 682965c..63d3da2 100644 --- a/tradefed_cluster/command_task_store.py +++ b/tradefed_cluster/command_task_store.py @@ -31,9 +31,6 @@ from tradefed_cluster import datastore_entities DEFAULT_COMMAND_ATTEMPT_HEARTBEAT = datetime.timedelta(hours=24) -# Limit the maximum size of the query -_TASK_LEASE_LIMIT = 1000 - @dataclasses.dataclass class CommandTaskArgs(object): @@ -164,10 +161,7 @@ def GetLeasableTasks(cluster, run_targets): datastore_entities.CommandTask.cluster == cluster, datastore_entities.CommandTask.leasable == True, namespace=common.NAMESPACE) - .order(-datastore_entities.CommandTask.priority) - .fetch(limit=_TASK_LEASE_LIMIT)) - - logging.info('Found %d leasable tasks on cluster %s', len(tasks), cluster) + .order(-datastore_entities.CommandTask.priority)) # Because Datastore will always return the tasks from the above query in the # same order, this means that multiple hosts trying to lease tasks for the |