diff --git a/llap-ext-client/src/java/org/apache/hadoop/hive/llap/LlapBaseInputFormat.java b/llap-ext-client/src/java/org/apache/hadoop/hive/llap/LlapBaseInputFormat.java index de9a031..c53e6cb 100644 --- a/llap-ext-client/src/java/org/apache/hadoop/hive/llap/LlapBaseInputFormat.java +++ b/llap-ext-client/src/java/org/apache/hadoop/hive/llap/LlapBaseInputFormat.java @@ -109,6 +109,7 @@ public static final String USER_KEY = "llap.if.user"; public static final String PWD_KEY = "llap.if.pwd"; public static final String HANDLE_ID = "llap.if.handleid"; + public static final String DB_KEY = "llap.if.database"; public final String SPLIT_QUERY = "select get_splits(\"%s\",%d)"; public static final LlapServiceInstance[] serviceInstanceArray = new LlapServiceInstance[0]; @@ -206,6 +207,7 @@ public LlapBaseInputFormat() {} if (query == null) query = job.get(QUERY_KEY); if (user == null) user = job.get(USER_KEY); if (pwd == null) pwd = job.get(PWD_KEY); + String database = job.get(DB_KEY); if (url == null || query == null) { throw new IllegalStateException(); @@ -235,6 +237,9 @@ public LlapBaseInputFormat() {} try ( Statement stmt = conn.createStatement(); ) { + if (database != null && !database.isEmpty()) { + stmt.execute("USE " + database); + } ResultSet res = stmt.executeQuery(sql); while (res.next()) { // deserialize split