Merge "parallelize add_partition_calls"

This commit is contained in:
Treehugger Robot 2023-05-12 04:54:11 +00:00 committed by Gerrit Code Review
commit 96b51b54b6

View file

@ -65,7 +65,7 @@ import verity_utils
import ota_metadata_pb2
import rangelib
import sparse_img
from concurrent.futures import ThreadPoolExecutor
from apex_utils import GetApexInfoFromTargetFiles
from common import ZipDelete, PARTITIONS_WITH_CARE_MAP, ExternalError, RunAndCheckOutput, IsSparseImage, MakeTempFile, ZipWrite
@ -1083,8 +1083,10 @@ def AddImagesToTargetFiles(filename):
("system_dlkm", has_system_dlkm, AddSystemDlkm, []),
("system_other", has_system_other, AddSystemOther, []),
)
for call in add_partition_calls:
add_partition(*call)
with ThreadPoolExecutor(max_workers=len(add_partition_calls)) as executor:
for future in [executor.submit(add_partition, *call) for call in add_partition_calls]:
future.result()
AddApexInfo(output_zip)