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

Class FakeBigQueryServices

java.lang.Object

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

All Implemented Interfaces:

Serializable, BigQueryServices

@Internal
public class FakeBigQueryServices
extends 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, BigQueryServices.WriteStreamService

Constructor Summary

Constructors

Constructor	Description
FakeBigQueryServices()

Method Summary

Modifier and Type	Method	Description
static TableRow	convertNumbers(TableRow tableRow)
static KV<Table,List<TableRow>>	decodeQueryResult(String queryResult)
static String	encodeQueryResult(Table table)
static String	encodeQueryResult(Table table, List<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.
BigQueryServices.WriteStreamService	getWriteStreamService(BigQueryOptions bqOptions)	Returns a real, mock, or fake BigQueryServices.WriteStreamService.
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 Details

FakeBigQueryServices

public FakeBigQueryServices()

Method Details

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

getWriteStreamService

public BigQueryServices.WriteStreamService getWriteStreamService(BigQueryOptions bqOptions)

Description copied from interface: BigQueryServices

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

Specified by:

getWriteStreamService 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 String encodeQueryResult(Table table)
                                throws IOException

Throws:

IOException

encodeQueryResult

public static String encodeQueryResult(Table table,
 List<TableRow> rows)
                                throws IOException

Throws:

IOException

decodeQueryResult

public static KV<Table,List<TableRow>> decodeQueryResult(String queryResult)
                                                  throws IOException

Throws:

IOException

convertNumbers

public static TableRow convertNumbers(TableRow tableRow)