From 3b80556fd6b4a60baaf32810abf4b94b57ee087e Mon Sep 17 00:00:00 2001
From: Gordon Hamilton <gordon.hamilton@wandisco.com>
Date: Thu, 12 Sep 2013 16:44:30 +0100
Subject: [PATCH] SHARK-172 - Make ppd2 unit test pass - Copy across
 numDistributionKeys for ReduceSinkOperator during merge.

---
 .../apache/hadoop/hive/ql/optimizer/ReduceSinkDeDuplication.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ql/src/java/org/apache/hadoop/hive/ql/optimizer/ReduceSinkDeDuplication.java b/ql/src/java/org/apache/hadoop/hive/ql/optimizer/ReduceSinkDeDuplication.java
index 323ca9027d6c..7151d565f606 100644
--- a/ql/src/java/org/apache/hadoop/hive/ql/optimizer/ReduceSinkDeDuplication.java
+++ b/ql/src/java/org/apache/hadoop/hive/ql/optimizer/ReduceSinkDeDuplication.java
@@ -365,6 +365,7 @@ protected boolean merge(ReduceSinkOperator cRS, ReduceSinkOperator pRS, int minR
       if (result[0] > 0) {
         ArrayList<ExprNodeDesc> childKCs = cRS.getConf().getKeyCols();
         pRS.getConf().setKeyCols(ExprNodeDescUtils.backtrack(childKCs, cRS, pRS));
+        pRS.getConf().setNumDistributionKeys(cRS.getConf().getNumDistributionKeys());
       }
       if (result[1] > 0) {
         ArrayList<ExprNodeDesc> childPCs = cRS.getConf().getPartitionCols();