Author: gunther
Date: Thu Dec 12 04:38:31 2013
New Revision: 1550347
URL: http://svn.apache.org/r1550347
Log:
HIVE-6019: Tez: Analyze command fails with dbclass=counter (Gunther Hagleitner)
Modified:
hive/branches/tez/ql/src/java/org/apache/hadoop/hive/ql/stats/CounterStatsAggregatorTez.java
Modified:
hive/branches/tez/ql/src/java/org/apache/hadoop/hive/ql/stats/CounterStatsAggregatorTez.java
URL:
http://svn.apache.org/viewvc/hive/branches/tez/ql/src/java/org/apache/hadoop/hive/ql/stats/CounterStatsAggregatorTez.java?rev=1550347&r1=1550346&r2=1550347&view=diff
==============================================================================
---
hive/branches/tez/ql/src/java/org/apache/hadoop/hive/ql/stats/CounterStatsAggregatorTez.java
(original)
+++
hive/branches/tez/ql/src/java/org/apache/hadoop/hive/ql/stats/CounterStatsAggregatorTez.java
Thu Dec 12 04:38:31 2013
@@ -32,15 +32,29 @@ public class CounterStatsAggregatorTez i
private static final Log LOG =
LogFactory.getLog(CounterStatsAggregatorTez.class.getName());
private TezCounters counters;
+ private CounterStatsAggregator mrAggregator;
+ private boolean delegate;
+
+ public CounterStatsAggregatorTez() {
+ mrAggregator = new CounterStatsAggregator();
+ }
@Override
public boolean connect(Configuration hconf, Task<?> sourceTask) {
+ if (!(sourceTask instanceof TezTask)) {
+ delegate = true;
+ return mrAggregator.connect(hconf, sourceTask);
+ }
counters = ((TezTask) sourceTask).getTezCounters();
return counters != null;
}
@Override
public String aggregateStats(String keyPrefix, String statType) {
+ if (delegate) {
+ return mrAggregator.aggregateStats(keyPrefix, statType);
+ }
+
long value = 0;
for (String groupName : counters.getGroupNames()) {
if (groupName.startsWith(keyPrefix)) {
