Intro to the Python DataStream API

DataStream programs in Flink are regular programs that implement transformations on data streams (e.g., filtering, updating state, defining windows, aggregating). The data streams are initially created from various sources (e.g., message queues, socket streams, files). Results are returned via sinks, which may for example write the data to files, or to standard output (for example the command line terminal).

Python DataStream API is a Python version of DataStream API which allows Python users could write Python DatStream API jobs.

Common Structure of Python DataStream API Programs

The following code example shows the common structure of Python DataStream API programs.

  1. from pyflink.common import WatermarkStrategy, Row
  2. from pyflink.common.serialization import Encoder
  3. from pyflink.common.typeinfo import Types
  4. from pyflink.datastream import StreamExecutionEnvironment
  5. from pyflink.datastream.connectors import FileSink, OutputFileConfig, NumberSequenceSource
  6. from pyflink.datastream.functions import RuntimeContext, MapFunction
  7. from pyflink.datastream.state import ValueStateDescriptor
  10. class MyMapFunction(MapFunction):
  12.     def open(self, runtime_context: RuntimeContext):
  13.         state_desc = ValueStateDescriptor('cnt', Types.PICKLED_BYTE_ARRAY())
  14.         self.cnt_state = runtime_context.get_state(state_desc)
  16.     def map(self, value):
  17.         cnt = self.cnt_state.value()
  18.         if cnt is None or cnt < 2:
  19.             self.cnt_state.update(1 if cnt is None else cnt + 1)
  20.             return value[0], value[1] + 1
  21.         else:
  22.             return value[0], value[1]
  25. def state_access_demo():
  26.     # 1. create a StreamExecutionEnvironment
  27.     env = StreamExecutionEnvironment.get_execution_environment()
  29.     # 2. create source DataStream
  30.     seq_num_source = NumberSequenceSource(1, 10000)
  31.     ds = env.from_source(
  32.         source=seq_num_source,
  33.         watermark_strategy=WatermarkStrategy.for_monotonous_timestamps(),
  34.         source_name='seq_num_source',
  35.         type_info=Types.LONG())
  37.     # 3. define the execution logic
  38.     ds = ds.map(lambda a: Row(a % 4, 1), output_type=Types.ROW([Types.LONG(), Types.LONG()])) \
  39.            .key_by(lambda a: a[0]) \
  40.            .map(MyMapFunction(), output_type=Types.TUPLE([Types.LONG(), Types.LONG()]))
  42.     # 4. create sink and emit result to sink
  43.     output_path = '/opt/output/'
  44.     file_sink = FileSink \
  45.         .for_row_format(output_path, Encoder.simple_string_encoder()) \
  46.         .with_output_file_config(OutputFileConfig.builder().with_part_prefix('pre').with_part_suffix('suf').build()) \
  47.         .build()
  48.     ds.sink_to(file_sink)
  50.     # 5. execute the job
  51.     env.execute('state_access_demo')
  54. if __name__ == '__main__':
  55.     state_access_demo()

Create a StreamExecutionEnvironment

The StreamExecutionEnvironment is a central concept of the DataStream API program. The following code example shows how to create a StreamExecutionEnvironment:

  1. from pyflink.datastream import StreamExecutionEnvironment
  3. env = StreamExecutionEnvironment.get_execution_environment()

Create a DataStream

The DataStream API gets its name from the special DataStream class that is used to represent a collection of data in a Flink program. You can think of them as immutable collections of data that can contain duplicates. This data can either be finite or unbounded, the API that you use to work on them is the same.

A DataStream is similar to a regular Python Collection in terms of usage but is quite different in some key ways. They are immutable, meaning that once they are created you cannot add or remove elements. You can also not simply inspect the elements inside but only work on them using the DataStream API operations, which are also called transformations.

You can create an initial DataStream by adding a source in a Flink program. Then you can derive new streams from this and combine them by using API methods such as map, filter, and so on.

Create from a list object

You can create a DataStream from a list object:

  1. from pyflink.common.typeinfo import Types
  2. from pyflink.datastream import StreamExecutionEnvironment
  4. env = StreamExecutionEnvironment.get_execution_environment()
  5. ds = env.from_collection(
  6.     collection=[(1, 'aaa|bb'), (2, 'bb|a'), (3, 'aaa|a')],
  7.     type_info=Types.ROW([Types.INT(), Types.STRING()]))

The parameter type_info is optional, if not specified, the output type of the returned DataStream will be Types.PICKLED_BYTE_ARRAY().

Create using DataStream connectors

You can also create a DataStream using DataStream connectors with method add_source as following:

  1. from pyflink.common.serialization import JsonRowDeserializationSchema
  2. from pyflink.common.typeinfo import Types
  3. from pyflink.datastream import StreamExecutionEnvironment
  4. from pyflink.datastream.connectors import FlinkKafkaConsumer
  6. env = StreamExecutionEnvironment.get_execution_environment()
  7. # the sql connector for kafka is used here as it's a fat jar and could avoid dependency issues
  8. env.add_jars("file:///path/to/flink-sql-connector-kafka.jar")
  10. deserialization_schema = JsonRowDeserializationSchema.builder() \
  11.     .type_info(type_info=Types.ROW([Types.INT(), Types.STRING()])).build()
  13. kafka_consumer = FlinkKafkaConsumer(
  14.     topics='test_source_topic',
  15.     deserialization_schema=deserialization_schema,
  16.     properties={'bootstrap.servers': 'localhost:9092', 'group.id': 'test_group'})
  18. ds = env.add_source(kafka_consumer)

Note It currently only supports FlinkKafkaConsumer to be used as DataStream source connectors with method add_source.

Note The DataStream created using add_source could only be executed in streaming executing mode.

You could also call the from_source method to create a DataStream using unified DataStream source connectors:

  1. from pyflink.common.typeinfo import Types
  2. from pyflink.common.watermark_strategy import WatermarkStrategy
  3. from pyflink.datastream import StreamExecutionEnvironment
  4. from pyflink.datastream.connectors import NumberSequenceSource
  6. env = StreamExecutionEnvironment.get_execution_environment()
  7. seq_num_source = NumberSequenceSource(1, 1000)
  8. ds = env.from_source(
  9.     source=seq_num_source,
  10.     watermark_strategy=WatermarkStrategy.for_monotonous_timestamps(),
  11.     source_name='seq_num_source',
  12.     type_info=Types.LONG())

Note Currently, it only supports NumberSequenceSource and FileSource as unified DataStream source connectors.

Note The DataStream created using from_source could be executed in both batch and streaming executing mode.

Create using Table & SQL connectors

Table & SQL connectors could also be used to create a DataStream. You could firstly create a Table using Table & SQL connectors and then convert it to a DataStream.

  1. from pyflink.common.typeinfo import Types
  2. from pyflink.datastream import StreamExecutionEnvironment
  3. from pyflink.table import StreamTableEnvironment
  5. env = StreamExecutionEnvironment.get_execution_environment()
  6. t_env = StreamTableEnvironment.create(stream_execution_environment=env)
  8. t_env.execute_sql("""
  9.         CREATE TABLE my_source (
  10.           a INT,
  11.           b VARCHAR
  12.         ) WITH (
  13.           'connector' = 'datagen',
  14.           'number-of-rows' = '10'
  15.         )
  16.     """)
  18. ds = t_env.to_append_stream(
  19.     t_env.from_path('my_source'),
  20.     Types.ROW([Types.INT(), Types.STRING()]))

Note The StreamExecutionEnvironment env should be specified when creating the TableEnvironment t_env.

Note As all the Java Table & SQL connectors could be used in PyFlink Table API, this means that all of them could also be used in PyFlink DataStream API.

DataStream Transformations

Operators transform one or more DataStream into a new DataStream. Programs can combine multiple transformations into sophisticated dataflow topologies.

The following example shows a simple example about how to convert a DataStream into another DataStream using map transformation:

  1. ds = ds.map(lambda a: a + 1)

Please see operators for an overview of the available DataStream transformations.

Conversion between DataStream and Table

It also supports to convert a DataStream to a Table and vice verse.

  1. # convert a DataStream to a Table
  2. table = t_env.from_data_stream(ds, 'a, b, c')
  4. # convert a Table to a DataStream
  5. ds = table.to_append_stream(table, Types.ROW([Types.INT(), Types.STRING()]))
  6. # or
  7. ds = table.to_retract_stream(table, Types.ROW([Types.INT(), Types.STRING()]))

Emit Results

Print

You can call the print method to print the data of a DataStream to the standard output:

  1. ds.print()

Collect results to client

You can call the execute_and_collect method to collect the data of a DataStream to client:

  1. with ds.execute_and_collect() as results:
  2.     for result in results:
  3.         print(result)

Note The method execute_and_collect will collect the data of the DataStream to the memory of the client and so it’s a good practice to limit the number of rows collected.

Emit results to a DataStream sink connector

You can call the add_sink method to emit the data of a DataStream to a DataStream sink connector:

  1. from pyflink.common.typeinfo import Types
  2. from pyflink.datastream.connectors import FlinkKafkaProducer
  3. from pyflink.common.serialization import JsonRowSerializationSchema
  5. serialization_schema = JsonRowSerializationSchema.builder().with_type_info(
  6.     type_info=Types.ROW([Types.INT(), Types.STRING()])).build()
  8. kafka_producer = FlinkKafkaProducer(
  9.     topic='test_sink_topic',
  10.     serialization_schema=serialization_schema,
  11.     producer_config={'bootstrap.servers': 'localhost:9092', 'group.id': 'test_group'})
  13. ds.add_sink(kafka_producer)

Note It currently only supports FlinkKafkaProducer, JdbcSink and StreamingFileSink to be used as DataStream sink connectors with method add_sink.

Note The method add_sink could only be used in streaming executing mode.

You could also call the sink_to method to emit the data of a DataStream to a unified DataStream sink connector:

  1. from pyflink.datastream.connectors import FileSink, OutputFileConfig
  2. from pyflink.common.serialization import Encoder
  4. output_path = '/opt/output/'
  5. file_sink = FileSink \
  6.     .for_row_format(output_path, Encoder.simple_string_encoder()) \
  7.     .with_output_file_config(OutputFileConfig.builder().with_part_prefix('pre').with_part_suffix('suf').build()) \
  8.     .build()
  9. ds.sink_to(file_sink)

Note It currently only supports FileSink as unified DataStream sink connectors.

Note The method sink_to could be used in both batch and streaming executing mode.

Emit results to a Table & SQL sink connector

Table & SQL connectors could also be used to write out a DataStream. You need firstly convert a DataStream to a Table and then write it to a Table & SQL sink connector.

  1. from pyflink.common import Row
  2. from pyflink.common.typeinfo import Types
  3. from pyflink.datastream import StreamExecutionEnvironment
  4. from pyflink.table import StreamTableEnvironment
  6. env = StreamExecutionEnvironment.get_execution_environment()
  7. t_env = StreamTableEnvironment.create(stream_execution_environment=env)
  9. # option 1：the result type of ds is Types.ROW
  10. def split(s):
  11.     splits = s[1].split("|")
  12.     for sp in splits:
  13.         yield Row(s[0], sp)
  15. ds = ds.map(lambda i: (i[0] + 1, i[1])) \
  16.        .flat_map(split, Types.ROW([Types.INT(), Types.STRING()])) \
  17.        .key_by(lambda i: i[1]) \
  18.        .reduce(lambda i, j: Row(i[0] + j[0], i[1]))
  20. # option 1：the result type of ds is Types.TUPLE
  21. def split(s):
  22.     splits = s[1].split("|")
  23.     for sp in splits:
  24.         yield s[0], sp
  26. ds = ds.map(lambda i: (i[0] + 1, i[1])) \
  27.        .flat_map(split, Types.TUPLE([Types.INT(), Types.STRING()])) \
  28.        .key_by(lambda i: i[1]) \
  29.        .reduce(lambda i, j: (i[0] + j[0], i[1]))
  31. # emit ds to print sink
  32. t_env.execute_sql("""
  33.         CREATE TABLE my_sink (
  34.           a INT,
  35.           b VARCHAR
  36.         ) WITH (
  37.           'connector' = 'print'
  38.         )
  39.     """)
  41. table = t_env.from_data_stream(ds)
  42. table_result = table.execute_insert("my_sink")

Note The output type of DataStream ds must be composite type.

Submit Job

Finally, you should call the StreamExecutionEnvironment.execute method to submit the DataStream API job for execution:

  1. env.execute()

If you convert the DataStream to a Table and then write it to a Table API & SQL sink connector, it may happen that you need to submit the job using TableEnvironment.execute method.

  1. t_env.execute()