Fix Aggregation push down NPE issue (pingcap#164)

wanfangdata · Dec 27, 2017 · fe033c1 · fe033c1
1 parent 5e1c1d5
commit fe033c1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/scala/org/apache/spark/sql/TiStrategy.scala b/src/main/scala/org/apache/spark/sql/TiStrategy.scala
@@ -94,6 +94,8 @@ class TiStrategy(context: SQLContext) extends Strategy with Logging {
                                dagRequest: TiDAGRequest): SparkPlan = {
     val table = source.table
     dagRequest.setTableInfo(table)
+    // Need to resolve column info after add aggregation push downs
+    dagRequest.resolve()
     if (dagRequest.getFields.isEmpty) {
       dagRequest.addRequiredColumn(TiColumnRef.create(table.getColumns.get(0).getName))
     }