Skip to content

Commit 75395a2

Browse files
committed
feat (TS): TransformationAgent does not bulk calls to dm.getReplicas
1 parent 5f55a2d commit 75395a2

File tree

1 file changed

+12
-13
lines changed

1 file changed

+12
-13
lines changed

src/DIRAC/TransformationSystem/Agent/TransformationAgent.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -500,19 +500,18 @@ def __getDataReplicas(self, transDict, lfns, clients, forJobs=True):
500500
startTime = time.time()
501501
self._logInfo(f"Getting replicas for {len(newLFNs)} files from catalog", method=method, transID=transID)
502502
newReplicas = {}
503-
for chunk in breakListIntoChunks(newLFNs, 10000):
504-
res = self._getDataReplicasDM(transID, chunk, clients, forJobs=forJobs)
505-
if res["OK"]:
506-
reps = {lfn: ses for lfn, ses in res["Value"].items() if ses}
507-
newReplicas.update(reps)
508-
self.__updateCache(transID, reps)
509-
else:
510-
self._logWarn(
511-
f"Failed to get replicas for {len(chunk)} files",
512-
res["Message"],
513-
method=method,
514-
transID=transID,
515-
)
503+
res = self._getDataReplicasDM(transID, newLFNs, clients, forJobs=forJobs)
504+
if res["OK"]:
505+
newReplicas = {lfn: ses for lfn, ses in res["Value"].items() if ses}
506+
507+
self.__updateCache(transID, newReplicas)
508+
else:
509+
self._logWarn(
510+
f"Failed to get replicas for {len(newLFNs)} files",
511+
res["Message"],
512+
method=method,
513+
transID=transID,
514+
)
516515

517516
self._logInfo(
518517
f"Obtained {len(newReplicas)} replicas from catalog in {time.time() - startTime:.1f} seconds",

0 commit comments

Comments
 (0)