diff --git a/kyuubi-hive-beeline/src/main/java/org/apache/hive/beeline/KyuubiCommands.java b/kyuubi-hive-beeline/src/main/java/org/apache/hive/beeline/KyuubiCommands.java index 311495212..572417847 100644 --- a/kyuubi-hive-beeline/src/main/java/org/apache/hive/beeline/KyuubiCommands.java +++ b/kyuubi-hive-beeline/src/main/java/org/apache/hive/beeline/KyuubiCommands.java @@ -45,7 +45,7 @@ public class KyuubiCommands extends Commands { /** Extract and clean up the first command in the input. */ private String getFirstCmd(String cmd, int length) { - return cmd.substring(length); + return cmd.substring(length).trim(); } private String[] tokenizeCmd(String cmd) { @@ -82,9 +82,8 @@ public class KyuubiCommands extends Commands { BufferedReader reader = null; try { reader = new BufferedReader(new FileReader(sourceFile)); - String extra = reader.readLine(); - String lines = null; - while (extra != null) { + String lines = null, extra; + while ((extra = reader.readLine()) != null) { if (beeLine.isComment(extra)) { continue; } @@ -93,7 +92,6 @@ public class KyuubiCommands extends Commands { } else { lines += "\n" + extra; } - extra = reader.readLine(); } String[] cmds = lines.split(";"); for (String c : cmds) {