svn commit: r1679437 - /pig/branches/branch-0.15/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java

[rohini]

Author: rohini
Date: Thu May 14 19:17:05 2015
New Revision: 1679437

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

Modified:
    
pig/branches/branch-0.15/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java

Modified: 
pig/branches/branch-0.15/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java
URL: 
http://svn.apache.org/viewvc/pig/branches/branch-0.15/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java?rev=1679437&r1=1679436&r2=1679437&view=diff
==============================================================================
--- 
pig/branches/branch-0.15/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java
 (original)
+++ 
pig/branches/branch-0.15/src/org/apache/pig/backend/hadoop/executionengine/tez/plan/optimizer/UnionOptimizer.java
 Thu May 14 19:17:05 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));