public ResultSet getNativeResultSet(ResultSet rs) throws SQLException { return (ResultSet) rs.unwrap(ResultSet.class); } }
@Override public <T> T unwrap(Class<T> iface) throws SQLException { return results.unwrap(iface); }
@Override public <T> T unwrap(Class<T> iface) throws SQLException { return rs.unwrap(iface); }
@Override public <T> T unwrap(Class<T> iface) throws SQLException { return delegate.unwrap(iface); }
public static ROWID getROWID(ResultSet rs, int columnIndex) throws SQLException { OracleResultSet oracleResultSet = rs.unwrap(OracleResultSet.class); return oracleResultSet.getROWID(columnIndex); }
/** {@inheritDoc} */ @Override @SuppressWarnings("unchecked") public final <T> T unwrap(Class<T> iface) throws SQLException { if (iface.isInstance(delegate)) { return (T) delegate; } else if (delegate != null) { return delegate.unwrap(iface); } throw new SQLException("Wrapped ResultSet is not an instance of " + iface); } }
protected long getProcessedLinesCount(String sqlStatement, QueryResult queryResult) throws SQLException { String queryId; if (queryResult.getJdbcResultSet().isPresent() && queryResult.getJdbcResultSet().get().isWrapperFor(PrestoResultSet.class)) { // if PrestoResult is available, just unwrap it from ResultSet and extract query id queryId = queryResult.getJdbcResultSet().get().unwrap(PrestoResultSet.class).getQueryId(); } else { // if there is no ResultSet (UPDATE statements), try to find it in system.runtime.queries table queryId = (String) query(format("select query_id from system.runtime.queries where query = '%s'", sqlStatement)).row(0).get(0); } return queryStatsClient.getQueryStats(queryId).get().getRawInputPositions(); } }
private static void assertConnectionSource(Connection connection, String expectedSource) throws SQLException { String queryId; try (Statement statement = connection.createStatement(); ResultSet rs = statement.executeQuery("SELECT 123")) { queryId = rs.unwrap(PrestoResultSet.class).getQueryId(); } try (PreparedStatement statement = connection.prepareStatement( "SELECT source FROM system.runtime.queries WHERE query_id = ?")) { statement.setString(1, queryId); try (ResultSet rs = statement.executeQuery()) { assertTrue(rs.next()); assertThat(rs.getString("source")).isEqualTo(expectedSource); assertFalse(rs.next()); } } } }
Statement statement = connection.createStatement(); ResultSet resultSet = statement.executeQuery("SELECT * FROM slow_test_table")) { queryId.set(resultSet.unwrap(PrestoResultSet.class).getQueryId()); queryStarted.countDown(); try {
queryId.set(resultSet.unwrap(PrestoResultSet.class).getQueryId()); queryStarted.countDown(); resultSet.next();
queryId.set(resultSet.unwrap(PrestoResultSet.class).getQueryId()); queryStarted.countDown(); resultSet.next();
Statement s = conn.createStatement(); ResultSet r = s.executeQuery(sql); sb.append(String.format("QueryId: %s\n", r.unwrap(DrillResultSet.class).getQueryId())); boolean first = true; while (r.next()) {
@Override public <T> T unwrap(Class<T> iface) throws SQLException { return delegate.unwrap(iface); }
@Override public <T> T unwrap(Class<T> iface) throws SQLException { return innerResultSet.unwrap(iface); }
@Override public <T> T unwrap(Class<T> iface) throws SQLException { return innerResultSet.unwrap(iface); }
public ResultSet getNativeResultSet(ResultSet rs) throws SQLException { return (ResultSet) rs.unwrap(ResultSet.class); } }
@Override public <T> T unwrap(Class<T> iface) throws SQLException { return rs.unwrap(iface); }
/** * Reset the read metrics collected in the result set. * * @see {@link #getRequestReadMetrics(ResultSet)} {@link #getOverAllReadRequestMetrics(ResultSet)} * @param rs * @throws SQLException */ public static void resetMetrics(ResultSet rs) throws SQLException { PhoenixResultSet prs = rs.unwrap(PhoenixResultSet.class); prs.resetMetrics(); }
private static void assertQualifierRangesNotPresent(ResultSet rs) throws SQLException { Scan scan = rs.unwrap(PhoenixResultSet.class).getStatement().getQueryPlan().getContext().getScan(); assertNull(scan.getAttribute(MIN_QUALIFIER)); assertNull(scan.getAttribute(MAX_QUALIFIER)); } }
private static void assertQualifierRanges(ResultSet rs, int minQualifier, int maxQualifier) throws SQLException { Scan scan = rs.unwrap(PhoenixResultSet.class).getStatement().getQueryPlan().getContext().getScan(); assertNotNull(scan.getAttribute(MIN_QUALIFIER)); assertNotNull(scan.getAttribute(MAX_QUALIFIER)); assertEquals(minQualifier, Bytes.toInt(scan.getAttribute(MIN_QUALIFIER))); assertEquals(maxQualifier, Bytes.toInt(scan.getAttribute(MAX_QUALIFIER))); }