@Override public TableDataList listTableData( String projectId, String datasetId, String tableId, Map<Option, ?> options) { try { return bigquery .tabledata() .list(projectId, datasetId, tableId) .setMaxResults(Option.MAX_RESULTS.getLong(options)) .setPageToken(Option.PAGE_TOKEN.getString(options)) .setStartIndex( Option.START_INDEX.getLong(options) != null ? BigInteger.valueOf(Option.START_INDEX.getLong(options)) : null) .execute(); } catch (IOException ex) { throw translate(ex); } }
private void onTableList(TableDataList result) throws IOException { when(mockClient.tabledata()).thenReturn(mockTabledata); when(mockTabledata.list(anyString(), anyString(), anyString())).thenReturn(mockTabledataList); when(mockTabledataList.execute()).thenReturn(result); }
private List<TableRow> getTableRows(Bigquery bq) { try { return bq.tabledata() .list( pipelineOptions.getProject(), pipelineOptions.getTargetDataset(), table.getTableReference().getTableId()) .execute() .getRows(); } catch (IOException e) { throw new RuntimeException(e); } }
@Override public TableDataList listTableData( String projectId, String datasetId, String tableId, Map<Option, ?> options) { try { return bigquery .tabledata() .list(projectId, datasetId, tableId) .setMaxResults(Option.MAX_RESULTS.getLong(options)) .setPageToken(Option.PAGE_TOKEN.getString(options)) .setStartIndex( Option.START_INDEX.getLong(options) != null ? BigInteger.valueOf(Option.START_INDEX.getLong(options)) : null) .execute(); } catch (IOException ex) { throw translate(ex); } }
@Override public Tuple<String, Iterable<TableRow>> listTableData(String datasetId, String tableId, Map<Option, ?> options) { try { TableDataList tableDataList = bigquery.tabledata() .list(this.options.projectId(), datasetId, tableId) .setMaxResults(MAX_RESULTS.getLong(options)) .setPageToken(PAGE_TOKEN.getString(options)) .setStartIndex(START_INDEX.getLong(options) != null ? BigInteger.valueOf(START_INDEX.getLong(options)) : null) .execute(); return Tuple.<String, Iterable<TableRow>>of(tableDataList.getPageToken(), tableDataList.getRows()); } catch (IOException ex) { throw translate(ex); } }
@Override public Tuple<String, Iterable<TableRow>> listTableData(String datasetId, String tableId, Map<Option, ?> options) { try { TableDataList tableDataList = bigquery.tabledata() .list(this.options.projectId(), datasetId, tableId) .setMaxResults(MAX_RESULTS.getLong(options)) .setPageToken(PAGE_TOKEN.getString(options)) .setStartIndex(START_INDEX.getLong(options) != null ? BigInteger.valueOf(START_INDEX.getLong(options)) : null) .execute(); return Tuple.<String, Iterable<TableRow>>of(tableDataList.getPageToken(), tableDataList.getRows()); } catch (IOException ex) { throw translate(ex); } }