/** Ensures that methods of the from() / fromQuery() family are called at most once. */ private void ensureFromNotCalledYet() { checkState( getJsonTableRef() == null && getQuery() == null, "from() or fromQuery() already called"); }
BigQueryHelpers.verifyDatasetPresence(datasetService, table.get()); BigQueryHelpers.verifyTablePresence(datasetService, table.get()); } else if (getQuery() != null) { checkArgument( getQuery().isAccessible(), "Cannot call validate if query is dynamically set."); JobService jobService = getBigQueryServices().getJobService(bqOptions); try { bqOptions.getProject(), new JobConfigurationQuery() .setQuery(getQuery().get()) .setFlattenResults(getFlattenResults()) .setUseLegacySql(getUseLegacySql()), } catch (Exception e) { throw new IllegalArgumentException( String.format(QUERY_VALIDATION_FAILURE_ERROR, getQuery().get()), e);
private BigQuerySourceBase<T> createSource(String jobUuid, Coder<T> coder) { BigQuerySourceBase<T> source; if (getQuery() == null) { source = BigQueryTableSource.create( jobUuid, getTableProvider(), getBigQueryServices(), coder, getParseFn()); } else { source = BigQueryQuerySource.create( jobUuid, getQuery(), getFlattenResults(), getUseLegacySql(), getBigQueryServices(), coder, getParseFn(), MoreObjects.firstNonNull(getQueryPriority(), QueryPriority.BATCH), getQueryLocation()); } return source; }
@Override public void populateDisplayData(DisplayData.Builder builder) { super.populateDisplayData(builder); builder .addIfNotNull( DisplayData.item("table", BigQueryHelpers.displayTable(getTableProvider())) .withLabel("Table")) .addIfNotNull(DisplayData.item("query", getQuery()).withLabel("Query")) .addIfNotNull( DisplayData.item("flattenResults", getFlattenResults()) .withLabel("Flatten Query Results")) .addIfNotNull( DisplayData.item("useLegacySql", getUseLegacySql()) .withLabel("Use Legacy SQL Dialect")) .addIfNotDefault( DisplayData.item("validation", getValidate()).withLabel("Validation Enabled"), true); }
ValueProvider<String> getQuery() { return this.inner.getQuery(); }