[#2282] fix(spark3): Invoke with object self in DelegationRssShuffleM…

…anager
apache · Dec 10, 2024 · 4afa783 · 4afa783
1 parent 0552d3b
commit 4afa783
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/client-spark/spark3/src/main/java/org/apache/spark/shuffle/DelegationRssShuffleManager.java b/client-spark/spark3/src/main/java/org/apache/spark/shuffle/DelegationRssShuffleManager.java
@@ -247,6 +247,7 @@ public <K, C> ShuffleReader<K, C> getReader(
                       TaskContext.class,
                       ShuffleReadMetricsReporter.class)
                   .invoke(
+                      delegate,
                       handle,
                       startMapIndex,
                       endMapIndex,
@@ -285,6 +286,7 @@ public <K, C> ShuffleReader<K, C> getReaderForRange(
                       TaskContext.class,
                       ShuffleReadMetricsReporter.class)
                   .invoke(
+                      delegate,
                       handle,
                       startMapIndex,
                       endMapIndex,