CLI bug fix

jackylk · jackylk · commit 28e228f374d4 · 2015-07-29T11:29:44.000-07:00
diff --git a/bin/hbase-sql b/bin/hbase-sql
@@ -32,8 +32,8 @@ echo "\$SPARK_HOME is not set"
 fi
 
 export FWDIR=$SPARK_HOME
-export SPARKHBASE_SOURCE="$(cd "`dirname "$0"`"/..; pwd)"
-export JAR="$SPARKHBASE_SOURCE/target/spark-sql-on-hbase-1.0.0.jar"
+export ASTRO_SOURCE="$(cd "`dirname "$0"`"/..; pwd)"
+export JAR="$ASTRO_SOURCE/target/spark-sql-on-hbase-1.0.0.jar"
 
 function usage {
   if [ -n "$1" ]; then
diff --git a/src/main/scala/org/apache/spark/sql/hbase/HBaseSQLCliDriver.scala b/src/main/scala/org/apache/spark/sql/hbase/HBaseSQLCliDriver.scala
@@ -27,7 +27,7 @@ import org.apache.spark.{Logging, SparkConf, SparkContext}
  *
  */
 object HBaseSQLCliDriver extends Logging {
-  private val prompt = "spark-hbaseql"
+  private val prompt = "astro"
   private val continuedPrompt = "".padTo(prompt.length, ' ')
   private val conf = new SparkConf()
   private val sc = new SparkContext(conf)
@@ -107,7 +107,7 @@ object HBaseSQLCliDriver extends Logging {
     // TODO: handle multiple command separated by ;
 
     // Since we are using SqlParser and it does not handle ';', just work around to omit the ';'
-    val input = line.trim.substring(0, line.length - 1)
+    val input = line.substring(0, line.length - 1)
 
     try {
       process(input.trim())
