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 and Description
static class	FakeBigQueryServices.FakeBigQueryServerStream<T> An implementation of 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

Constructor Summary

Constructors

Constructor and Description
FakeBigQueryServices()

Method Summary

Modifier and Type	Method and Description
static TableRow	convertNumbers(TableRow tableRow)
static KV<Table,java.util.List<TableRow>>	decodeQueryResult(java.lang.String queryResult)
static java.lang.String	encodeQueryResult(Table table)
static java.lang.String	encodeQueryResult(Table table, java.util.List<TableRow> rows)
BigQueryServices.DatasetService	getDatasetService(BigQueryOptions bqOptions) Returns a real, mock, or fake BigQueryServices.DatasetService.
BigQueryServices.DatasetService	getDatasetService(BigQueryOptions bqOptions, org.apache.beam.sdk.util.Histogram requestLatencies) 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

getDatasetService

public BigQueryServices.DatasetService getDatasetService(BigQueryOptions bqOptions,
                                                         org.apache.beam.sdk.util.Histogram requestLatencies)

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(Table table)
                                          throws java.io.IOException

Throws:

java.io.IOException

encodeQueryResult

public static java.lang.String encodeQueryResult(Table table,
                                                 java.util.List<TableRow> rows)
                                          throws java.io.IOException

Throws:

java.io.IOException

decodeQueryResult

public static KV<Table,java.util.List<TableRow>> decodeQueryResult(java.lang.String queryResult)
                                                            throws java.io.IOException

Throws:

java.io.IOException

convertNumbers

public static TableRow convertNumbers(TableRow tableRow)