Skip to content

Commit 27af9b2

Browse files
committed
Safeguard boundedtrie result
1 parent 8527d43 commit 27af9b2

File tree

1 file changed

+8
-1
lines changed
  • runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow

1 file changed

+8
-1
lines changed

runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/DataflowMetrics.java

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,14 @@ private BoundedTrieResult getBoundedTrieValue(MetricUpdate metricUpdate) {
209209
if (metricUpdate.getTrie() == null) {
210210
return BoundedTrieResult.empty();
211211
}
212-
BoundedTrie bTrie = (BoundedTrie) metricUpdate.getTrie();
212+
BoundedTrie bTrie;
213+
try {
214+
bTrie = (BoundedTrie) metricUpdate.getTrie();
215+
} catch (ClassCastException cce) {
216+
LOG.warn(
217+
"Failed to retrieve BoundedTrie metrics from result. Returning empty result.", cce);
218+
return BoundedTrieResult.empty();
219+
}
213220
BoundedTrieData trieData = BoundedTrieData.fromProto(bTrie);
214221
return BoundedTrieResult.create(trieData.extractResult().getResult());
215222
}

0 commit comments

Comments
 (0)