svn commit: r1679436 - /pig/trunk/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java

[rohini]

Author: rohini
Date: Thu May 14 19:15:56 2015
New Revision: 1679436

URL: http://svn.apache.org/r1679436
Log:
PIG-4503: Additional fix to correct the store vertex group members (rohini)

Modified:
    
pig/trunk/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java

Modified: 
pig/trunk/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java
URL: 
http://svn.apache.org/viewvc/pig/trunk/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java?rev=1679436&r1=1679435&r2=1679436&view=diff
==============================================================================
--- 
pig/trunk/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java
 (original)
+++ 
pig/trunk/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java
 Thu May 14 19:15:56 2015
@@ -161,6 +161,7 @@ public class UnionOptimizer extends TezO
             if (existingVertexGroup != null) {
                 storeVertexGroupOps[i] = existingVertexGroup;
                 
existingVertexGroup.getVertexGroupMembers().remove(unionOp.getOperatorKey());
+                
existingVertexGroup.getVertexGroupMembers().addAll(unionOp.getUnionMembers());
                 
existingVertexGroup.getVertexGroupInfo().removeInput(unionOp.getOperatorKey());
             } else {
                 storeVertexGroupOps[i] = new 
TezOperator(OperatorKey.genOpKey(scope));