Examples with Column com.facebook.presto.client.Column used on opensource projects

Example 6 with Column

use of com.facebook.presto.client.Column in project presto by prestodb.

the class PrestoResultSet method getColumns.

private static List<Column> getColumns(StatementClient client, Consumer<QueryStats> progressCallback) throws SQLException {
    while (client.isValid()) {
        QueryResults results = client.current();
        progressCallback.accept(QueryStats.create(results.getId(), results.getStats()));
        List<Column> columns = results.getColumns();
        if (columns != null) {
            return columns;
        }
        client.advance();
    }
    QueryResults results = client.finalResults();
    if (!client.isFailed()) {
        throw new SQLException(format("Query has no columns (#%s)", results.getId()));
    }
    throw resultsException(results);
}

Example 7 with Column

use of com.facebook.presto.client.Column in project airpal by airbnb.

the class QueryResource method getHistory.

@GET
@Path("history")
@Produces(MediaType.APPLICATION_JSON)
public Response getHistory(@Auth AirpalUser user, @QueryParam("table") List<Table> tables) {
    Iterable<Job> recentlyRun;
    if (tables.size() < 1) {
        recentlyRun = jobHistoryStore.getRecentlyRun(200);
    } else {
        Table[] tablesArray = tables.toArray(new Table[tables.size()]);
        Table[] restTables = Arrays.copyOfRange(tablesArray, 1, tablesArray.length);
        recentlyRun = jobHistoryStore.getRecentlyRun(200, tablesArray[0], restTables);
    }
    ImmutableList.Builder<Job> filtered = ImmutableList.builder();
    for (Job job : recentlyRun) {
        if (job.getTablesUsed().isEmpty() && (job.getState() == JobState.FAILED)) {
            filtered.add(job);
            continue;
        }
        for (Table table : job.getTablesUsed()) {
            if (AuthorizationUtil.isAuthorizedRead(user, table)) {
                filtered.add(new Job(job.getUser(), job.getQuery(), job.getUuid(), job.getOutput(), job.getQueryStats(), job.getState(), Collections.<Column>emptyList(), Collections.<Table>emptySet(), job.getQueryStartedDateTime(), job.getError(), job.getQueryFinishedDateTime()));
            }
        }
    }
    List<Job> sortedResult = Ordering.natural().nullsLast().onResultOf(JOB_ORDERING).reverse().immutableSortedCopy(filtered.build());
    return Response.ok(sortedResult).build();
}