You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@pig.apache.org by ch...@apache.org on 2014/05/12 05:39:34 UTC
svn commit: r1593885 -
/pig/branches/tez/src/org/apache/pig/backend/hadoop/executionengine/tez/TezTaskContext.java
Author: cheolsoo
Date: Mon May 12 03:39:34 2014
New Revision: 1593885
URL: http://svn.apache.org/r1593885
Log:
PIG-3933: PigRecordWriter throws exception in Tez mode (cheolsoo)
Modified:
pig/branches/tez/src/org/apache/pig/backend/hadoop/executionengine/tez/TezTaskContext.java
Modified: pig/branches/tez/src/org/apache/pig/backend/hadoop/executionengine/tez/TezTaskContext.java
URL: http://svn.apache.org/viewvc/pig/branches/tez/src/org/apache/pig/backend/hadoop/executionengine/tez/TezTaskContext.java?rev=1593885&r1=1593884&r2=1593885&view=diff
==============================================================================
--- pig/branches/tez/src/org/apache/pig/backend/hadoop/executionengine/tez/TezTaskContext.java (original)
+++ pig/branches/tez/src/org/apache/pig/backend/hadoop/executionengine/tez/TezTaskContext.java Mon May 12 03:39:34 2014
@@ -20,6 +20,7 @@ package org.apache.pig.backend.hadoop.ex
import org.apache.hadoop.mapreduce.Counter;
import org.apache.pig.backend.hadoop.executionengine.TaskContext;
import org.apache.tez.common.counters.TezCounter;
+import org.apache.tez.mapreduce.hadoop.mapred.MRCounters.MRCounter;
import org.apache.tez.runtime.api.TezProcessorContext;
public class TezTaskContext extends TaskContext<TezProcessorContext> {
@@ -36,12 +37,20 @@ public class TezTaskContext extends Task
@Override
public Counter getCounter(Enum<?> name) {
- throw new UnsupportedOperationException();
+ if (context == null) {
+ return null;
+ }
+ TezCounter tezCounter = context.getCounters().findCounter(name);
+ return new MRCounter(tezCounter);
}
@Override
public Counter getCounter(String group, String name) {
- throw new UnsupportedOperationException();
+ if (context == null) {
+ return null;
+ }
+ TezCounter tezCounter = context.getCounters().getGroup(group).findCounter(name);
+ return new MRCounter(tezCounter);
}
@Override