Refine search
@Override public boolean delete(DatasetId datasetId, DatasetDeleteOption... options) { final DatasetId completeDatasetId = datasetId.setProjectId(getOptions().getProjectId()); final Map<BigQueryRpc.Option, ?> optionsMap = optionMap(options); try { return runWithRetries( new Callable<Boolean>() { @Override public Boolean call() { return bigQueryRpc.deleteDataset( completeDatasetId.getProject(), completeDatasetId.getDataset(), optionsMap); } }, getOptions().getRetrySettings(), EXCEPTION_HANDLER, getOptions().getClock()); } catch (RetryHelper.RetryHelperException e) { throw BigQueryException.translateAndThrow(e); } }
@Override public BigQuery create(BigQueryOptions options) { return new BigQueryImpl(options); } }
@Override public boolean cancel(String jobId) { return cancel(JobId.of(jobId)); }
@Override public QueryResponse getQueryResults(JobId jobId, QueryResultsOption... options) { Map<BigQueryRpc.Option, ?> optionsMap = optionMap(options); return getQueryResults(jobId, getOptions(), optionsMap); }
@Override public Page<Job> listJobs(JobListOption... options) { return listJobs(getOptions(), optionMap(options)); }
@Override public Page<Dataset> listDatasets(DatasetListOption... options) { return listDatasets(getOptions().getProjectId(), options); }
@Override public boolean delete(final DatasetId datasetId, DatasetDeleteOption... options) { final Map<BigQueryRpc.Option, ?> optionsMap = optionMap(options); try { return runWithRetries(new Callable<Boolean>() { @Override public Boolean call() { return bigQueryRpc.deleteDataset(datasetId.dataset(), optionsMap); } }, options().retryParams(), EXCEPTION_HANDLER, options().clock()); } catch (RetryHelper.RetryHelperException e) { throw BigQueryException.translateAndThrow(e); } }
PROJECT, JOB, null, BigQueryImpl.optionMap(Job.DEFAULT_QUERY_WAIT_OPTIONS))) .andReturn(responsePb1); EasyMock.expect( bigqueryRpcMock.getQueryResults( PROJECT, JOB, null, BigQueryImpl.optionMap(Job.DEFAULT_QUERY_WAIT_OPTIONS))) .andReturn(responsePb2); EasyMock.expect( bigquery = options.getService(); TableResult result = bigquery.query(QUERY_JOB_CONFIGURATION_FOR_QUERY, queryJob); assertThat(result.getSchema()).isEqualTo(TABLE_SCHEMA);
serviceOptions.getRetrySettings(), EXCEPTION_HANDLER, serviceOptions.getClock()); String cursor = result.x(); return new PageImpl<>(
@Test public void testCreateJobTryGet() { final String id = "testCreateJobTryGet-id"; String query = "SELECT * in FOO"; Supplier<JobId> idProvider = new Supplier<JobId>() { @Override public JobId get() { return JobId.of(id); } }; Capture<com.google.api.services.bigquery.model.Job> jobCapture = EasyMock.newCapture(); EasyMock.expect( bigqueryRpcMock.create(EasyMock.capture(jobCapture), EasyMock.eq(EMPTY_RPC_OPTIONS))) .andThrow(new BigQueryException(409, "already exists, for some reason")); EasyMock.expect( bigqueryRpcMock.getJob( anyString(), EasyMock.eq(id), EasyMock.eq((String) null), EasyMock.eq(EMPTY_RPC_OPTIONS))) .andReturn(newJobPb()); EasyMock.replay(bigqueryRpcMock); bigquery = options.getService(); ((BigQueryImpl) bigquery).create(JobInfo.of(QueryJobConfiguration.of(query)), idProvider); assertThat(jobCapture.getValue().getJobReference().getJobId()).isEqualTo(id); }
@Override public Page<Table> listTables(String datasetId, TableListOption... options) { return listTables( DatasetId.of(getOptions().getProjectId(), datasetId), getOptions(), optionMap(options)); }
@Override public TableResult listTableData(TableId tableId, Schema schema, TableDataListOption... options) { Tuple<? extends Page<FieldValueList>, Long> data = listTableData(tableId, getOptions(), optionMap(options)); return new TableResult(schema, data.y(), data.x()); }
@Override public Page<List<FieldValue>> listTableData(TableId tableId, TableDataListOption... options) { return listTableData(tableId, options(), optionMap(options)); }
@Override public Page<Table> listTables(String datasetId, TableListOption... options) { return listTables(datasetId, options(), optionMap(options)); }
@Override public QueryResponse getQueryResults(JobId job, QueryResultsOption... options) { Map<BigQueryRpc.Option, ?> optionsMap = optionMap(options); return getQueryResults(job, options(), optionsMap); }
@Override public Page<Dataset> listDatasets(DatasetListOption... options) { return listDatasets(options(), optionMap(options)); }
@Override public Job create(JobInfo jobInfo, JobOption... options) { Supplier<JobId> idProvider = new Supplier<JobId>() { @Override public JobId get() { return JobId.of(); } }; return create(jobInfo, idProvider, options); }
@Override public boolean delete(String datasetId, DatasetDeleteOption... options) { return delete(DatasetId.of(datasetId), options); }
@Override public Table getTable(final String datasetId, final String tableId, TableOption... options) { return getTable(TableId.of(datasetId, tableId), options); }