Package org.apache.beam.sdk.io.gcp.testing

Class FakeBigQueryServices

java.lang.Object

org.apache.beam.sdk.io.gcp.testing.FakeBigQueryServices

All Implemented Interfaces:

java.io.Serializable, BigQueryServices

@Internal
public class FakeBigQueryServices
extends java.lang.Object
implements BigQueryServices

A fake implementation of BigQuery's query service..

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	FakeBigQueryServices.FakeBigQueryServerStream<T>	An implementation of BigQueryServices.BigQueryServerStream which takes a List as the Iterable to simulate a server stream.

Nested classes/interfaces inherited from interface org.apache.beam.sdk.io.gcp.bigquery.BigQueryServices

BigQueryServices.BigQueryServerStream<T>, BigQueryServices.DatasetService, BigQueryServices.JobService, BigQueryServices.StorageClient, BigQueryServices.StreamAppendClient

Constructor Summary

Constructors

Constructor	Description
FakeBigQueryServices()

Method Summary

Modifier and Type	Method	Description
static com.google.api.services.bigquery.model.TableRow	convertNumbers(com.google.api.services.bigquery.model.TableRow tableRow)
static org.apache.beam.sdk.values.KV<com.google.api.services.bigquery.model.Table,java.util.List<com.google.api.services.bigquery.model.TableRow>>	decodeQueryResult(java.lang.String queryResult)
static java.lang.String	encodeQueryResult(com.google.api.services.bigquery.model.Table table)
static java.lang.String	encodeQueryResult(com.google.api.services.bigquery.model.Table table, java.util.List<com.google.api.services.bigquery.model.TableRow> rows)
BigQueryServices.DatasetService	getDatasetService(BigQueryOptions bqOptions)	Returns a real, mock, or fake BigQueryServices.DatasetService.
BigQueryServices.JobService	getJobService(BigQueryOptions bqOptions)	Returns a real, mock, or fake BigQueryServices.JobService.
BigQueryServices.StorageClient	getStorageClient(BigQueryOptions bqOptions)	Returns a real, mock, or fake BigQueryServices.StorageClient.
FakeBigQueryServices	withDatasetService(FakeDatasetService datasetService)
FakeBigQueryServices	withJobService(BigQueryServices.JobService jobService)
FakeBigQueryServices	withStorageClient(BigQueryServices.StorageClient storageClient)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

FakeBigQueryServices

public FakeBigQueryServices()

Method Detail

withJobService

public FakeBigQueryServices withJobService(BigQueryServices.JobService jobService)

withDatasetService

public FakeBigQueryServices withDatasetService(FakeDatasetService datasetService)

withStorageClient

public FakeBigQueryServices withStorageClient(BigQueryServices.StorageClient storageClient)

getJobService

public BigQueryServices.JobService getJobService(BigQueryOptions bqOptions)

Description copied from interface: BigQueryServices

Returns a real, mock, or fake BigQueryServices.JobService.

Specified by:

getJobService in interface BigQueryServices

getDatasetService

public BigQueryServices.DatasetService getDatasetService(BigQueryOptions bqOptions)

Description copied from interface: BigQueryServices

Returns a real, mock, or fake BigQueryServices.DatasetService.

Specified by:

getDatasetService in interface BigQueryServices

getStorageClient

public BigQueryServices.StorageClient getStorageClient(BigQueryOptions bqOptions)

Description copied from interface: BigQueryServices

Returns a real, mock, or fake BigQueryServices.StorageClient.

Specified by:

getStorageClient in interface BigQueryServices

encodeQueryResult

public static java.lang.String encodeQueryResult(com.google.api.services.bigquery.model.Table table)
                                          throws java.io.IOException

Throws:

java.io.IOException

encodeQueryResult

public static java.lang.String encodeQueryResult(com.google.api.services.bigquery.model.Table table,
                                                 java.util.List<com.google.api.services.bigquery.model.TableRow> rows)
                                          throws java.io.IOException

Throws:

java.io.IOException

decodeQueryResult

public static org.apache.beam.sdk.values.KV<com.google.api.services.bigquery.model.Table,java.util.List<com.google.api.services.bigquery.model.TableRow>> decodeQueryResult(java.lang.String queryResult)
                                                                                                                                                                           throws java.io.IOException

Throws:

java.io.IOException

convertNumbers

public static com.google.api.services.bigquery.model.TableRow convertNumbers(com.google.api.services.bigquery.model.TableRow tableRow)