org.apache.beam.sdk.extensions.sql.meta.provider.kafka

Class BeamKafkaTable

java.lang.Object

org.apache.beam.sdk.extensions.sql.meta.BaseBeamTable

org.apache.beam.sdk.extensions.sql.meta.SchemaBaseBeamTable

org.apache.beam.sdk.extensions.sql.meta.provider.kafka.BeamKafkaTable

All Implemented Interfaces:

java.io.Serializable, BeamSqlTable

Direct Known Subclasses:

BeamKafkaCSVTable, PayloadSerializerKafkaTable

public abstract class BeamKafkaTable
extends SchemaBaseBeamTable

BeamKafkaTable represent a Kafka topic, as source or target. Need to extend to convert between BeamSqlRow and KV<byte[], byte[]>.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected int	numberOfRecordsForRate

Fields inherited from class org.apache.beam.sdk.extensions.sql.meta.SchemaBaseBeamTable

schema

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BeamKafkaTable(org.apache.beam.sdk.schemas.Schema beamSchema)
	BeamKafkaTable(org.apache.beam.sdk.schemas.Schema beamSchema, java.util.List<org.apache.kafka.common.TopicPartition> topicPartitions, java.lang.String bootstrapServers)
	BeamKafkaTable(org.apache.beam.sdk.schemas.Schema beamSchema, java.lang.String bootstrapServers, java.util.List<java.lang.String> topics)

Method Summary

Modifier and Type	Method and Description
org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.values.Row>	buildIOReader(org.apache.beam.sdk.values.PBegin begin) create a PCollection<Row> from source.
org.apache.beam.sdk.values.POutput	buildIOWriter(org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.values.Row> input) create a IO.write() instance to write to target.
protected org.apache.beam.sdk.io.kafka.KafkaIO.Read<byte[],byte[]>	createKafkaRead()
java.lang.String	getBootstrapServers()
protected abstract org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.io.kafka.KafkaRecord<byte[],byte[]>>,org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.values.Row>>	getPTransformForInput()
protected abstract org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.values.Row>,org.apache.beam.sdk.values.PCollection<org.apache.kafka.clients.producer.ProducerRecord<byte[],byte[]>>>	getPTransformForOutput()
BeamTableStatistics	getTableStatistics(org.apache.beam.sdk.options.PipelineOptions options) Estimates the number of rows or the rate for unbounded Tables.
java.util.List<java.lang.String>	getTopics()
org.apache.beam.sdk.values.PCollection.IsBounded	isBounded() Whether this table is bounded (known to be finite) or unbounded (may or may not be finite).
BeamKafkaTable	updateConsumerProperties(java.util.Map<java.lang.String,java.lang.Object> configUpdates)

Methods inherited from class org.apache.beam.sdk.extensions.sql.meta.SchemaBaseBeamTable

getSchema

Methods inherited from class org.apache.beam.sdk.extensions.sql.meta.BaseBeamTable

buildIOReader, constructFilter, supportsProjects

Methods inherited from class java.lang.Object

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

Field Detail

numberOfRecordsForRate

protected int numberOfRecordsForRate

Constructor Detail

BeamKafkaTable

protected BeamKafkaTable(org.apache.beam.sdk.schemas.Schema beamSchema)

BeamKafkaTable

public BeamKafkaTable(org.apache.beam.sdk.schemas.Schema beamSchema,
                      java.lang.String bootstrapServers,
                      java.util.List<java.lang.String> topics)

BeamKafkaTable

public BeamKafkaTable(org.apache.beam.sdk.schemas.Schema beamSchema,
                      java.util.List<org.apache.kafka.common.TopicPartition> topicPartitions,
                      java.lang.String bootstrapServers)

Method Detail

updateConsumerProperties

public BeamKafkaTable updateConsumerProperties(java.util.Map<java.lang.String,java.lang.Object> configUpdates)

isBounded

public org.apache.beam.sdk.values.PCollection.IsBounded isBounded()

Description copied from interface: BeamSqlTable

Whether this table is bounded (known to be finite) or unbounded (may or may not be finite).

getPTransformForInput

protected abstract org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.io.kafka.KafkaRecord<byte[],byte[]>>,org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.values.Row>> getPTransformForInput()

getPTransformForOutput

protected abstract org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.values.Row>,org.apache.beam.sdk.values.PCollection<org.apache.kafka.clients.producer.ProducerRecord<byte[],byte[]>>> getPTransformForOutput()

buildIOReader

public org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.values.Row> buildIOReader(org.apache.beam.sdk.values.PBegin begin)

Description copied from interface: BeamSqlTable

create a PCollection<Row> from source.

createKafkaRead

protected org.apache.beam.sdk.io.kafka.KafkaIO.Read<byte[],byte[]> createKafkaRead()

buildIOWriter

public org.apache.beam.sdk.values.POutput buildIOWriter(org.apache.beam.sdk.values.PCollection<org.apache.beam.sdk.values.Row> input)

Description copied from interface: BeamSqlTable

create a IO.write() instance to write to target.

getBootstrapServers

public java.lang.String getBootstrapServers()

getTopics

public java.util.List<java.lang.String> getTopics()

getTableStatistics

public BeamTableStatistics getTableStatistics(org.apache.beam.sdk.options.PipelineOptions options)

Description copied from interface: BeamSqlTable

Estimates the number of rows or the rate for unbounded Tables. If it is not possible to estimate the row count or rate it will return BeamTableStatistics.BOUNDED_UNKNOWN.

Specified by:

getTableStatistics in interface BeamSqlTable

Overrides:

getTableStatistics in class BaseBeamTable