Presto Jdbc

標簽(空格分隔)： Presto

一， 建立連接

傳統的JDBC方式類似，建立PrestoConnection”連接“，并且通過unwrap方法將connection轉換為PrestoConnection。實際上是賦值一些基本信息，并且建立新的OkHttpClient。

String url = "jdbc:presto://ip:port/hive/“; //默認連接hive

String user = "PRESTO";

Properties properties = new Properties();

properties.setProperty("user", user);

PrestoConnection conn = DriverManager.getConnection(prestoUrl, properties).unwrap(PrestoConnection.class);

conn.setClientInfo("ApplicationName", "group_1");? //指定資源組

conn.setSessionProperty("query_max_total_memory", "1GB"); //指定此次操作可使用的presto最大內存大小

conn.setSessionProperty("","");類似的屬性可以在 presto Client 中進入查看:

SET SESSION;

即可顯示 可以在SESSION 級別修改的 屬性。

二，建立Statement執行語句

指定SQL執行的相關屬性。在設置監聽器的時候需要注意！presto的任務監聽器會阻塞presto任務的執行，所以不建議在監聽器中做任何耗時的操作。如果需要使用監聽器記錄presto任務的狀態，可自己啟動一個線程使用prestoResultSet.getStats()獲取當前任務狀態，監聽任務進度。

PrestoStatement statement = conn.createStatement().unwrap(PrestoStatement.class);

statement.setQueryTimeout(10); ?//設置SQL語句可執行的時長(秒)

statement.setLargeMaxRows(1000);? //設置可獲取結果集的大小(分批獲取，直到超過此值后結束)

AtomicReference queryId = new AtomicReference<>();

statement.setProgressMonitor(queryStats -> { //設置監聽器(可選)，可監聽presto任務執行狀況

queryId.set(queryStats.getQueryId()); //獲取presto任務ID(可用該ID終止任務)

});

PrestoResultSet resultSet = statement.executeQuery("select * from table").unwrap(PrestoResultSet.class);

三，獲取結果集

將結果集轉換為json列表。這里需要注意的是resultSet.next()方法，Presto服務端并不會一次全部把結果返回給客戶端，而是不斷的通過next()方法調用HTTP接口獲取(每次獲取結果集大小默認1mb)，直到PrestoClient狀態不為Running時結束。

List results = new ArrayList<>();

int count = resultSet.getMetaData().getColumnCount();

String[] columns = new String[count];

for (int i = 0; i < count; i++) {

columns[i] = resultSet.getMetaData().getColumnName(i + 1);

}

while (resultSet.next()) {

JSONObject jsonObject = new JSONObject();

for (int j = 0; j < count; j++) {

jsonObject.put(columns[j], resultSet.getString(j + 1));

}

results.add(jsonObject);

}

創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的presto java_Presto Jdbc的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。