diff --git a/beeline/src/java/org/apache/hive/beeline/Commands.java b/beeline/src/java/org/apache/hive/beeline/Commands.java index a25e09b35393e0bc5486d8f96096a209e0e3597a..039e3549209682b5ebdde937607b77c3ec094527 100644 --- a/beeline/src/java/org/apache/hive/beeline/Commands.java +++ b/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]); } } @@ -1085,7 +1094,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 a/itests/hive-unit/src/test/java/org/apache/hive/beeline/TestBeeLineWithArgs.java b/itests/hive-unit/src/test/java/org/apache/hive/beeline/TestBeeLineWithArgs.java index 49c1120689a33b4336bb04ca19ed4d398f6feca7..1ca76232a9d689a18945125550278d47ce98591f 100644 --- a/itests/hive-unit/src/test/java/org/apache/hive/beeline/TestBeeLineWithArgs.java +++ b/itests/hive-unit/src/test/java/org/apache/hive/beeline/TestBeeLineWithArgs.java @@ -903,4 +903,12 @@ public void testShowDbInPrompt() 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); + } }