com.google.cloud.spark.bigquery.write.context

Class BigQueryDirectDataWriterContext

java.lang.Object

com.google.cloud.spark.bigquery.write.context.BigQueryDirectDataWriterContext

All Implemented Interfaces:

DataWriterContext<org.apache.spark.sql.catalyst.InternalRow>, Closeable, AutoCloseable

public class BigQueryDirectDataWriterContext
extends Object
implements DataWriterContext<org.apache.spark.sql.catalyst.InternalRow>

Constructor Summary

Constructors

Constructor and Description

BigQueryDirectDataWriterContext(int partitionId, long taskId, long epochId, BigQueryClientFactory writeClientFactory, String tablePath, org.apache.spark.sql.types.StructType sparkSchema, com.google.cloud.bigquery.storage.v1.ProtoSchema protoSchema, com.google.api.gax.retrying.RetrySettings bigqueryDataWriterHelperRetrySettings, com.google.common.base.Optional<String> traceId, boolean writeAtLeastOnce)

Method Summary

Modifier and Type	Method and Description
void	abort()
void	close()
WriterCommitMessageContext	commit()
void	write(org.apache.spark.sql.catalyst.InternalRow record)

Methods inherited from class java.lang.Object

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

Constructor Detail

BigQueryDirectDataWriterContext

public BigQueryDirectDataWriterContext(int partitionId,
                                       long taskId,
                                       long epochId,
                                       BigQueryClientFactory writeClientFactory,
                                       String tablePath,
                                       org.apache.spark.sql.types.StructType sparkSchema,
                                       com.google.cloud.bigquery.storage.v1.ProtoSchema protoSchema,
                                       com.google.api.gax.retrying.RetrySettings bigqueryDataWriterHelperRetrySettings,
                                       com.google.common.base.Optional<String> traceId,
                                       boolean writeAtLeastOnce)

Method Detail

write

public void write(org.apache.spark.sql.catalyst.InternalRow record)
           throws IOException

Specified by:

write in interface DataWriterContext<org.apache.spark.sql.catalyst.InternalRow>

Throws:

IOException

commit

public WriterCommitMessageContext commit()
                                  throws IOException

Specified by:

commit in interface DataWriterContext<org.apache.spark.sql.catalyst.InternalRow>

Throws:

IOException

abort

public void abort()
           throws IOException

Specified by:

abort in interface DataWriterContext<org.apache.spark.sql.catalyst.InternalRow>

Throws:

IOException

close

public void close()
           throws IOException

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable

Throws:

IOException