Upsert Kafka SQL Connector

Scan Source: Unbounded Sink: Streaming Upsert Mode

The Upsert Kafka connector allows for reading data from and writing data into Kafka topics in the upsert fashion.

As a source, the upsert-kafka connector produces a changelog stream, where each data record represents an update or delete event. More precisely, the value in a data record is interpreted as an UPDATE of the last value for the same key, if any (if a corresponding key doesn’t exist yet, the update will be considered an INSERT). Using the table analogy, a data record in a changelog stream is interpreted as an UPSERT aka INSERT/UPDATE because any existing row with the same key is overwritten. Also, null values are interpreted in a special way: a record with a null value represents a “DELETE”.

As a sink, the upsert-kafka connector can consume a changelog stream. It will write INSERT/UPDATE_AFTER data as normal Kafka messages value, and write DELETE data as Kafka messages with null values (indicate tombstone for the key). Flink will guarantee the message ordering on the primary key by partition data on the values of the primary key columns, so the update/deletion messages on the same key will fall into the same partition.

Dependencies

There is no connector (yet) available for Flink version 1.19.

The Upsert Kafka connector is not part of the binary distribution. See how to link with it for cluster execution here.

Full Example

The example below shows how to create and use an Upsert Kafka table:

  1. CREATE TABLE pageviews_per_region (
  2. user_region STRING,
  3. pv BIGINT,
  4. uv BIGINT,
  5. PRIMARY KEY (user_region) NOT ENFORCED
  6. ) WITH (
  7. 'connector' = 'upsert-kafka',
  8. 'topic' = 'pageviews_per_region',
  9. 'properties.bootstrap.servers' = '...',
  10. 'key.format' = 'avro',
  11. 'value.format' = 'avro'
  12. );
  13. CREATE TABLE pageviews (
  14. user_id BIGINT,
  15. page_id BIGINT,
  16. viewtime TIMESTAMP,
  17. user_region STRING,
  18. WATERMARK FOR viewtime AS viewtime - INTERVAL '2' SECOND
  19. ) WITH (
  20. 'connector' = 'kafka',
  21. 'topic' = 'pageviews',
  22. 'properties.bootstrap.servers' = '...',
  23. 'format' = 'json'
  24. );
  25. -- calculate the pv, uv and insert into the upsert-kafka sink
  26. INSERT INTO pageviews_per_region
  27. SELECT
  28. user_region,
  29. COUNT(*),
  30. COUNT(DISTINCT user_id)
  31. FROM pageviews
  32. GROUP BY user_region;

Attention Make sure to define the primary key in the DDL.

Available Metadata

See the regular Kafka connector for a list of all available metadata fields.

Connector Options

OptionRequiredDefaultTypeDescription
connector
required(none)StringSpecify which connector to use, for the Upsert Kafka use: ‘upsert-kafka’.
topic
required(none)StringThe Kafka topic name to read from and write to.
properties.bootstrap.servers
required(none)StringComma separated list of Kafka brokers.
properties.*
optional(none)StringThis can set and pass arbitrary Kafka configurations. Suffix names must match the configuration key defined in Kafka Configuration documentation. Flink will remove the “properties.” key prefix and pass the transformed key and values to the underlying KafkaClient. For example, you can disable automatic topic creation via ‘properties.allow.auto.create.topics’ = ‘false’. But there are some configurations that do not support to set, because Flink will override them, e.g. ‘key.deserializer’ and ‘value.deserializer’.
key.format
required(none)String

The format used to deserialize and serialize the key part of Kafka messages. Please refer to the formats page for more details and more format options.

Attention Compared to the regular Kafka connector, the key fields are specified by the PRIMARY KEY syntax.
key.fields-prefix
optional(none)StringDefines a custom prefix for all fields of the key format to avoid name clashes with fields of the value format. By default, the prefix is empty. If a custom prefix is defined, both the table schema and ‘key.fields’ will work with prefixed names. When constructing the data type of the key format, the prefix will be removed and the non-prefixed names will be used within the key format. Please note that this option requires that ‘value.fields-include’ must be set to ‘EXCEPT_KEY’.
value.format
required(none)StringThe format used to deserialize and serialize the value part of Kafka messages. Please refer to the formats page for more details and more format options.
value.fields-include
optionalALL

Enum

Possible values: [ALL, EXCEPT_KEY]
Defines a strategy how to deal with key columns in the data type of the value format. By default, ‘ALL’ physical columns of the table schema will be included in the value format which means that key columns appear in the data type for both the key and value format.
sink.parallelism
optional(none)IntegerDefines the parallelism of the upsert-kafka sink operator. By default, the parallelism is determined by the framework using the same parallelism of the upstream chained operator.
sink.buffer-flush.max-rows
optional0IntegerThe max size of buffered records before flush. When the sink receives many updates on the same key, the buffer will retain the last record of the same key. This can help to reduce data shuffling and avoid possible tombstone messages to Kafka topic. Can be set to ‘0’ to disable it. By default, this is disabled. Note both ‘sink.buffer-flush.max-rows’ and ‘sink.buffer-flush.interval’ must be set to be greater than zero to enable sink buffer flushing.
sink.buffer-flush.interval
optional0DurationThe flush interval mills, over this time, asynchronous threads will flush data. When the sink receives many updates on the same key, the buffer will retain the last record of the same key. This can help to reduce data shuffling and avoid possible tombstone messages to Kafka topic. Can be set to ‘0’ to disable it. By default, this is disabled. Note both ‘sink.buffer-flush.max-rows’ and ‘sink.buffer-flush.interval’ must be set to be greater than zero to enable sink buffer flushing.

Features

Key and Value Formats

See the regular Kafka connector for more explanation around key and value formats. However, note that this connector requires both a key and value format where the key fields are derived from the PRIMARY KEY constraint.

The following example shows how to specify and configure key and value formats. The format options are prefixed with either the 'key' or 'value' plus format identifier.

SQL

  1. CREATE TABLE KafkaTable (
  2. `ts` TIMESTAMP(3) METADATA FROM 'timestamp',
  3. `user_id` BIGINT,
  4. `item_id` BIGINT,
  5. `behavior` STRING,
  6. PRIMARY KEY (`user_id`) NOT ENFORCED
  7. ) WITH (
  8. 'connector' = 'upsert-kafka',
  9. ...
  10. 'key.format' = 'json',
  11. 'key.json.ignore-parse-errors' = 'true',
  12. 'value.format' = 'json',
  13. 'value.json.fail-on-missing-field' = 'false',
  14. 'value.fields-include' = 'EXCEPT_KEY'
  15. )

Primary Key Constraints

The Upsert Kafka always works in the upsert fashion and requires to define the primary key in the DDL. With the assumption that records with the same key should be ordered in the same partition, the primary key semantic on the changelog source means the materialized changelog is unique on the primary keys. The primary key definition will also control which fields should end up in Kafka’s key.

Consistency Guarantees

By default, an Upsert Kafka sink ingests data with at-least-once guarantees into a Kafka topic if the query is executed with checkpointing enabled.

This means, Flink may write duplicate records with the same key into the Kafka topic. But as the connector is working in the upsert mode, the last record on the same key will take effect when reading back as a source. Therefore, the upsert-kafka connector achieves idempotent writes just like the HBase sink.

Source Per-Partition Watermarks

Flink supports to emit per-partition watermarks for Upsert Kafka. Watermarks are generated inside the Kafka consumer. The per-partition watermarks are merged in the same way as watermarks are merged during streaming shuffles. The output watermark of the source is determined by the minimum watermark among the partitions it reads. If some partitions in the topics are idle, the watermark generator will not advance. You can alleviate this problem by setting the ‘table.exec.source.idle-timeout’ option in the table configuration.

Please refer to Kafka watermark strategies for more details.

Data Type Mapping

Upsert Kafka stores message keys and values as bytes, so Upsert Kafka doesn’t have schema or data types. The messages are serialized and deserialized by formats, e.g. csv, json, avro. Thus, the data type mapping is determined by specific formats. Please refer to Formats pages for more details.