com.google.cloud.bigquery.storage.v1

Class ConnectionWorker

java.lang.Object

com.google.cloud.bigquery.storage.v1.ConnectionWorker

All Implemented Interfaces:

AutoCloseable

public class ConnectionWorker
extends Object
implements AutoCloseable

A BigQuery Stream Writer that can be used to write data into BigQuery Table.

TODO: Support batching.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ConnectionWorker.Load Represent the current workload for this worker.

Constructor Summary

Constructors

Constructor and Description
ConnectionWorker(String streamName, ProtoSchema writerSchema, long maxInflightRequests, long maxInflightBytes, com.google.api.gax.batching.FlowController.LimitExceededBehavior limitExceededBehavior, String traceId, BigQueryWriteClient client, boolean ownsBigQueryWriteClient)

Method Summary

Modifier and Type	Method and Description
com.google.api.core.ApiFuture<AppendRowsResponse>	append(ProtoRows rows) Schedules the writing of rows at the end of current stream.
com.google.api.core.ApiFuture<AppendRowsResponse>	append(ProtoRows rows, long offset) Schedules the writing of rows at given offset.
void	close() Close the stream writer.
static long	getApiMaxRequestBytes() The maximum size of one request.
long	getInflightWaitSeconds()
TableSchema	getUpdatedSchema() Thread-safe getter of updated TableSchema
String	getWriterId()

Methods inherited from class java.lang.Object

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

Constructor Detail

ConnectionWorker

public ConnectionWorker(String streamName,
                        ProtoSchema writerSchema,
                        long maxInflightRequests,
                        long maxInflightBytes,
                        com.google.api.gax.batching.FlowController.LimitExceededBehavior limitExceededBehavior,
                        String traceId,
                        BigQueryWriteClient client,
                        boolean ownsBigQueryWriteClient)
                 throws IOException

Throws:

IOException

Method Detail

getApiMaxRequestBytes

public static long getApiMaxRequestBytes()

The maximum size of one request. Defined by the API.

append

public com.google.api.core.ApiFuture<AppendRowsResponse> append(ProtoRows rows)

Schedules the writing of rows at the end of current stream.

append

public com.google.api.core.ApiFuture<AppendRowsResponse> append(ProtoRows rows,
                                                                long offset)

Schedules the writing of rows at given offset.

getInflightWaitSeconds

public long getInflightWaitSeconds()

getWriterId

public String getWriterId()

Returns:

a unique Id for the writer.

close

public void close()

Close the stream writer. Shut down all resources.

Specified by:

close in interface AutoCloseable

getUpdatedSchema

public TableSchema getUpdatedSchema()

Thread-safe getter of updated TableSchema