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