@Test public void testOf() { DatasetInfo datasetInfo = DatasetInfo.of(DATASET_ID.getDataset()); assertEquals(DATASET_ID, datasetInfo.getDatasetId()); assertNull(datasetInfo.getAcl()); assertNull(datasetInfo.getCreationTime()); assertNull(datasetInfo.getDefaultTableLifetime()); assertNull(datasetInfo.getDescription()); assertNull(datasetInfo.getEtag()); assertNull(datasetInfo.getFriendlyName()); assertNull(datasetInfo.getGeneratedId()); assertNull(datasetInfo.getLastModified()); assertNull(datasetInfo.getLocation()); assertNull(datasetInfo.getSelfLink()); assertTrue(datasetInfo.getLabels().isEmpty()); datasetInfo = DatasetInfo.of(DATASET_ID); assertEquals(DATASET_ID, datasetInfo.getDatasetId()); assertNull(datasetInfo.getAcl()); assertNull(datasetInfo.getCreationTime()); assertNull(datasetInfo.getDefaultTableLifetime()); assertNull(datasetInfo.getDescription()); assertNull(datasetInfo.getEtag()); assertNull(datasetInfo.getFriendlyName()); assertNull(datasetInfo.getGeneratedId()); assertNull(datasetInfo.getLastModified()); assertNull(datasetInfo.getLocation()); assertNull(datasetInfo.getSelfLink()); assertTrue(datasetInfo.getLabels().isEmpty()); }
private void createBqDatasetIfMissing(String bqDatasetName) { if (bigQuery.getDataset(bqDatasetName) != null) { return; } // create dataset bigQuery.create(com.google.cloud.bigquery.DatasetInfo.of(bqDatasetName)); }
private QueryResponse queryWithLarge(BigQuery bigquery, QueryRequest queryRequest, String projectId) { String tempDataset = genTempName("dataset"); String tempTable = genTempName("table"); bigquery.create(DatasetInfo.of(tempDataset)); TableId tableId = TableId.of(projectId, tempDataset, tempTable); QueryJobConfiguration jobConfiguration = QueryJobConfiguration .newBuilder(queryRequest.getQuery()) .setAllowLargeResults(true) .setUseLegacySql(queryRequest.useLegacySql()) .setDestinationTable(tableId) .build(); Job job = bigquery.create(JobInfo.of(jobConfiguration)); QueryResponse queryResponse = bigquery.getQueryResults(job.getJobId()); queryResponse = loopQueryResponse(bigquery, queryResponse); bigquery.delete(tableId); return queryResponse; }
private QueryResponse queryWithLarge(BigQuery bigquery, QueryRequest queryRequest, String projectId) { String tempDataset = genTempName("dataset"); String tempTable = genTempName("table"); bigquery.create(DatasetInfo.of(tempDataset)); TableId tableId = TableId.of(projectId, tempDataset, tempTable); QueryJobConfiguration jobConfiguration = QueryJobConfiguration .newBuilder(queryRequest.getQuery()) .setAllowLargeResults(true) .setUseLegacySql(queryRequest.useLegacySql()) .setDestinationTable(tableId) .build(); Job job = bigquery.create(JobInfo.of(jobConfiguration)); QueryResponse queryResponse = bigquery.getQueryResults(job.getJobId()); queryResponse = loopQueryResponse(bigquery, queryResponse); bigquery.delete(tableId); return queryResponse; }