From 9ddabe0acf76e3acaa026c77cd46d484cc6afb60 Mon Sep 17 00:00:00 2001
From: Kent Yao <yaooqinn@hotmail.com>
Date: Tue, 20 Mar 2018 16:56:12 +0800
Subject: [PATCH] toLocalIterator

---
 src/main/scala/yaooqinn/kyuubi/operation/KyuubiOperation.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/scala/yaooqinn/kyuubi/operation/KyuubiOperation.scala b/src/main/scala/yaooqinn/kyuubi/operation/KyuubiOperation.scala
index 7ac92d1b6..155a2df4e 100644
--- a/src/main/scala/yaooqinn/kyuubi/operation/KyuubiOperation.scala
+++ b/src/main/scala/yaooqinn/kyuubi/operation/KyuubiOperation.scala
@@ -383,7 +383,7 @@ class KyuubiOperation(session: KyuubiSession, statement: String) extends Logging
         _.onStatementParsed(statementId, result.queryExecution.toString())
       }
       debug(result.queryExecution.toString())
-      iter = result.collect().iterator
+      iter = result.toLocalIterator().asScala
       dataTypes = result.queryExecution.analyzed.output.map(_.dataType).toArray
       setState(FINISHED)
       KyuubiServerMonitor.getListener(session.getUserName).foreach(_.onStatementFinish(statementId))