diff --git beeline/src/java/org/apache/hive/beeline/Commands.java beeline/src/java/org/apache/hive/beeline/Commands.java index 3a204c0..7160788 100644 --- beeline/src/java/org/apache/hive/beeline/Commands.java +++ beeline/src/java/org/apache/hive/beeline/Commands.java @@ -742,10 +742,12 @@ public boolean sql(String line) { private Map getHiveVariables() { Map result = new HashMap<>(); BufferedRows rows = getConfInternal(true); - while (rows.hasNext()) { - Rows.Row row = (Rows.Row) rows.next(); - if (!row.isMeta) { - result.put(row.values[0], row.values[1]); + if (rows != null) { + while (rows.hasNext()) { + Rows.Row row = (Rows.Row) rows.next(); + if (!row.isMeta) { + result.put(row.values[0], row.values[1]); + } } } return result; @@ -784,13 +786,19 @@ private BufferedRows getConfInternal(boolean call) { Statement stmnt = null; BufferedRows rows = null; try { - boolean hasResults; - if (call) { - stmnt = beeLine.getDatabaseConnection().getConnection().prepareCall("set"); - hasResults = ((CallableStatement) stmnt).execute(); - } else { - stmnt = beeLine.createStatement(); - hasResults = stmnt.execute("set"); + boolean hasResults = false; + DatabaseConnection dbconn = beeLine.getDatabaseConnection(); + Connection conn = null; + if (dbconn != null) + conn = dbconn.getConnection(); + if (conn != null) { + if (call) { + stmnt = conn.prepareCall("set"); + hasResults = ((CallableStatement) stmnt).execute(); + } else { + stmnt = beeLine.createStatement(); + hasResults = stmnt.execute("set"); + } } if (hasResults) { ResultSet rs = stmnt.getResultSet(); @@ -823,7 +831,8 @@ private void addConf(Rows.Row r, HiveConf hiveConf) { return; } else { String[] kv = val.split("=", 2); - hiveConf.set(kv[0], kv[1]); + if (kv.length == 2) + hiveConf.set(kv[0], kv[1]); } } @@ -1088,7 +1097,8 @@ public boolean sh(String line) { } line = line.substring("sh".length()).trim(); - line = substituteVariables(getHiveConf(false), line.trim()); + if (!beeLine.isBeeLine()) + line = substituteVariables(getHiveConf(false), line.trim()); try { ShellCmdExecutor executor = new ShellCmdExecutor(line, beeLine.getOutputStream(), diff --git itests/hive-unit/src/test/java/org/apache/hive/beeline/TestBeeLineWithArgs.java itests/hive-unit/src/test/java/org/apache/hive/beeline/TestBeeLineWithArgs.java index ae68f62..9159957 100644 --- itests/hive-unit/src/test/java/org/apache/hive/beeline/TestBeeLineWithArgs.java +++ itests/hive-unit/src/test/java/org/apache/hive/beeline/TestBeeLineWithArgs.java @@ -864,4 +864,12 @@ public void testConnectionWithURLParams() throws Throwable { testScriptFile( SCRIPT_TEXT, EXPECTED_PATTERN, true, argList); } + + @Test + public void testBeelineShellCommandWithoutConn() throws Throwable { + List argList = new ArrayList(); + final String SCRIPT_TEXT = "!sh echo hello world"; + final String EXPECTED_PATTERN = "hello world"; + testScriptFile(SCRIPT_TEXT, EXPECTED_PATTERN, true, argList,true,false); + } }