Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
import org.apache.spark.util.AccumulatorV2;
import scala.Function1;
import scala.collection.JavaConverters;
import scala.collection.Seq;
import scala.collection.mutable.ArrayBuffer;

/**
Expand Down Expand Up @@ -91,7 +92,9 @@ protected List<AccumulatorV2> getExternalAccumulators(TaskMetrics metrics) {
}

// withExternalAccums didn't work, try the legacy method
ArrayBuffer<AccumulatorV2> accumulators = methodLoader.invoke(externalAccums, metrics);
// Use Seq (not ArrayBuffer) since some Spark distributions (e.g. Amazon EMR) return a
// JListWrapper which implements Seq but cannot be cast to ArrayBuffer
Seq<AccumulatorV2> accumulators = methodLoader.invoke(externalAccums, metrics);
if (accumulators != null && !accumulators.isEmpty()) {
return JavaConverters.seqAsJavaList(accumulators);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
import org.apache.spark.sql.execution.metric.SQLMetricInfo;
import org.apache.spark.util.AccumulatorV2;
import scala.Function1;
import scala.collection.Seq;
import scala.collection.mutable.ArrayBuffer;
import scala.jdk.javaapi.CollectionConverters;

Expand Down Expand Up @@ -91,7 +92,9 @@ protected List<AccumulatorV2> getExternalAccumulators(TaskMetrics metrics) {
}

// withExternalAccums didn't work, try the legacy method
ArrayBuffer<AccumulatorV2> accumulators = methodLoader.invoke(externalAccums, metrics);
// Use Seq (not ArrayBuffer) since some Spark distributions (e.g. Amazon EMR) return a
// JListWrapper which implements Seq but cannot be cast to ArrayBuffer
Seq<AccumulatorV2> accumulators = methodLoader.invoke(externalAccums, metrics);
if (accumulators != null && !accumulators.isEmpty()) {
return CollectionConverters.asJava(accumulators);
}
Expand Down
Loading