From ef095b61da0e095c2066b6bf42d0b27194bcafd5 Mon Sep 17 00:00:00 2001 From: Arun Pandian Date: Wed, 24 Jun 2026 08:02:51 +0000 Subject: [PATCH] [ReduceFnRunner] Fix Prefetches 1. Add a prefetch for CombinedMetadata 2. Prefetch ReduceFn using renamedContext instead of directContext. This improves prefetching on merging windows --- .../main/java/org/apache/beam/runners/core/ReduceFnRunner.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/runners/core-java/src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java b/runners/core-java/src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java index 78505f3c65f6..7fe3b711aa0a 100644 --- a/runners/core-java/src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java +++ b/runners/core-java/src/main/java/org/apache/beam/runners/core/ReduceFnRunner.java @@ -1056,7 +1056,8 @@ private void prefetchOnTrigger( paneInfoTracker.prefetchPaneInfo(directContext); watermarkHold.prefetchExtract(renamedContext); nonEmptyPanes.isEmpty(renamedContext.state()).readLater(); - reduceFn.prefetchOnTrigger(directContext.state()); + directContext.state().access(METADATA_TAG).readLater(); + reduceFn.prefetchOnTrigger(renamedContext.state()); } /**