This is an automated email from the ASF dual-hosted git repository. hequn pushed a commit to branch asf-site in repository https://gitbox.apache.org/repos/asf/flink-web.git
The following commit(s) were added to refs/heads/asf-site by this push: new 133f472 Add blog post: Introducing Python Support for UDFs in Flink's Table API. (#316) 133f472 is described below commit 133f47228d4cfed2ebd6a9727edf33980f90ac44 Author: Jincheng Sun <sunjincheng...@gmail.com> AuthorDate: Thu Apr 9 23:23:22 2020 +0800 Add blog post: Introducing Python Support for UDFs in Flink's Table API. (#316) Co-authored-by: Markos Sfikas <mar...@data-artisans.com> --- _posts/2020-02-23-pyflink-udf-support-flink.md | 195 +++++++++++++++++++++ .../pyflink-udf-architecture.png | Bin 0 -> 928431 bytes .../2020-02-23-pyflink-udfs/roadmap-of-pyflink.png | Bin 0 -> 1191798 bytes 3 files changed, 195 insertions(+) diff --git a/_posts/2020-02-23-pyflink-udf-support-flink.md b/_posts/2020-02-23-pyflink-udf-support-flink.md new file mode 100644 index 0000000..b0e46ac --- /dev/null +++ b/_posts/2020-02-23-pyflink-udf-support-flink.md @@ -0,0 +1,195 @@ +--- +layout: post +title: "PyFlink: Introducing Python Support for UDFs in Flink's Table API" +date: 2020-02-23T12:00:00.000Z +authors: +- Jincheng: + name: "Jincheng Sun" + twitter: "sunjincheng121" +- markos: + name: "Markos Sfikas" + twitter: "MarkSfik" +excerpt: Flink 1.10 extends its support for Python by adding Python UDFs in PyFlink. This post explains how UDFs work in PyFlink and gives some practical examples of how to use UDFs in PyFlink. +--- + +Flink 1.9 introduced the Python Table API, allowing developers and data engineers to write Python Table API jobs for Table transformations and analysis, such as Python ETL or aggregate jobs. However, Python users faced some limitations when it came to support for Python UDFs in Flink 1.9, preventing them from extending the system’s built-in functionality. + +In Flink 1.10, the community further extended the support for Python by adding Python UDFs in PyFlink. Additionally, both the Python UDF environment and dependency management are now supported, allowing users to import third-party libraries in the UDFs, leveraging Python's rich set of third-party libraries. + + +# Python Support for UDFs in Flink 1.10 + +Before diving into how you can define and use Python UDFs, we explain the motivation and background behind how UDFs work in PyFlink and provide some additional context about the implementation of our approach. Below we give a brief introduction on the PyFlink architecture from job submission, all the way to executing the Python UDF. + +The PyFlink architecture mainly includes two parts — local and cluster — as shown in the architecture visual below. The local phase is the compilation of the job, and the cluster is the execution of the job. + +<center> +<img src="{{ site.baseurl }}/img/blog/2020-02-23-pyflink-udfs/pyflink-udf-architecture.png" width="600px" alt="PyFlink UDF Architecture"/> +</center> +<br> + +For the local part, the Python API is a mapping of the Java API: each time Python executes a method in the figure above, it will synchronously call the method corresponding to Java through Py4J, and finally generate a Java JobGraph, before submitting it to the cluster. + +For the cluster part, just like ordinary Java jobs, the JobMaster schedules tasks to TaskManagers. The tasks that include Python UDF in a TaskManager involve the execution of Java and Python operators. In the Python UDF operator, various gRPC services are used to provide different communications between the Java VM and the Python VM, such as DataService for data transmissions, StateService for state requirements, and Logging and Metrics Services. These services are built on Beam's Fn API [...] + +# How to use PyFlink with UDFs in Flink 1.10 + +This section provides some Python user defined function (UDF) examples, including how to install PyFlink, how to define/register/invoke UDFs in PyFlink and how to execute the job. + + +## Install PyFlink +Using Python in Apache Flink requires installing PyFlink. PyFlink is available through PyPI and can be easily installed using pip: + +```bash +$ python -m pip install apache-flink +``` +<div class="alert alert-info" markdown="1"> +<span class="label label-info" style="display: inline-block"><span class="glyphicon glyphicon-info-sign" aria-hidden="true"></span> Note</span> +Please note that Python 3.5 or higher is required to install and run PyFlink +</div> + +<br> + +## Define a Python UDF + +There are many ways to define a Python scalar function, besides extending the base class `ScalarFunction`. The following example shows the different ways of defining a Python scalar function that takes two columns of `BIGINT` as input parameters and returns the sum of them as the result. + +```python +# option 1: extending the base class `ScalarFunction` +class Add(ScalarFunction): + def eval(self, i, j): + return i + j + +add = udf(Add(), [DataTypes.BIGINT(), DataTypes.BIGINT()], DataTypes.BIGINT()) + +# option 2: Python function +@udf(input_types=[DataTypes.BIGINT(), DataTypes.BIGINT()], result_type=DataTypes.BIGINT()) +def add(i, j): + return i + j + +# option 3: lambda function +add = udf(lambda i, j: i + j, [DataTypes.BIGINT(), DataTypes.BIGINT()], DataTypes.BIGINT()) + +# option 4: callable function +class CallableAdd(object): + def __call__(self, i, j): + return i + j + +add = udf(CallableAdd(), [DataTypes.BIGINT(), DataTypes.BIGINT()], DataTypes.BIGINT()) + +# option 5: partial function +def partial_add(i, j, k): + return i + j + k + +add = udf(functools.partial(partial_add, k=1), [DataTypes.BIGINT(), DataTypes.BIGINT()], + DataTypes.BIGINT()) +``` + +## Register a Python UDF + +```python +# register the Python function +table_env.register_function("add", add) +``` + +## Invoke a Python UDF + +```python +# use the function in Python Table API +my_table.select("add(a, b)") +``` + +Below, you can find a complete example of using Python UDF. + +```python +from pyflink.datastream import StreamExecutionEnvironment +from pyflink.table import StreamTableEnvironment, DataTypes +from pyflink.table.descriptors import Schema, OldCsv, FileSystem +from pyflink.table.udf import udf + +env = StreamExecutionEnvironment.get_execution_environment() +env.set_parallelism(1) +t_env = StreamTableEnvironment.create(env) + +add = udf(lambda i, j: i + j, [DataTypes.BIGINT(), DataTypes.BIGINT()], DataTypes.BIGINT()) + +t_env.register_function("add", add) + +t_env.connect(FileSystem().path('/tmp/input')) \ + .with_format(OldCsv() + .field('a', DataTypes.BIGINT()) + .field('b', DataTypes.BIGINT())) \ + .with_schema(Schema() + .field('a', DataTypes.BIGINT()) + .field('b', DataTypes.BIGINT())) \ + .create_temporary_table('mySource') + +t_env.connect(FileSystem().path('/tmp/output')) \ + .with_format(OldCsv() + .field('sum', DataTypes.BIGINT())) \ + .with_schema(Schema() + .field('sum', DataTypes.BIGINT())) \ + .create_temporary_table('mySink') + +t_env.from_path('mySource')\ + .select("add(a, b)") \ + .insert_into('mySink') + +t_env.execute("tutorial_job") +``` + +## Submit the job + +Firstly, you need to prepare the input data in the “/tmp/input” file. For example, + +`$ echo "1,2" > /tmp/input` + +Next, you can run this example on the command line, + +`$ python python_udf_sum.py` + +The command builds and runs the Python Table API program in a local mini-cluster. You can also submit the Python Table API program to a remote cluster using different command lines, (see more details [here](https://ci.apache.org/projects/flink/flink-docs-release-1.10/ops/cli.html#job-submission-examples)). + +Finally, you can see the execution result on the command line: + +`$ cat /tmp/output + 3` + + +## Python UDF dependency management + +In many cases, you would like to import third-party dependencies in the Python UDF. The example below provides detailed guidance on how to manage such dependencies. + +Suppose you want to use the `mpmath` to perform the sum of the example above. The Python UDF may look like: + +```python +@udf(input_types=[DataTypes.BIGINT(), DataTypes.BIGINT()], result_type=DataTypes.BIGINT()) +def add(i, j): + from mpmath import fadd # add third-party dependency + return int(fadd(1, 2)) +``` + +To make it available on the worker node that does not contain the dependency, you can specify the dependencies with the following commands and API: + +```shell +$ cd /tmp +$ echo mpmath==1.1.0 > requirements.txt +$ pip download -d cached_dir -r requirements.txt --no-binary :all: +``` + +```python + +t_env.set_python_requirements("/tmp/requirements.txt", "/tmp/cached_dir") +``` + +A `requirements.txt` file that defines the third-party dependencies is used. If the dependencies cannot be accessed in the cluster, then you can specify a directory containing the installation packages of these dependencies by using the parameter "`requirements_cached_dir`", as illustrated in the example above. The dependencies will be uploaded to the cluster and installed offline. + + +# Conclusion & Upcoming work + +In this blog post, we introduced the architecture of Python UDFs in PyFlink and provided some examples on how to define, register and invoke UDFs. Flink 1.10 brings Python support in the framework to new levels, allowing Python users to write even more magic with their preferred language. The community is actively working towards continuously improving the functionality and performance of PyFlink. Future work in upcoming releases will introduce support for Pandas UDFs in scalar and aggre [...] + +<center> +<img src="{{ site.baseurl }}/img/blog/2020-02-23-pyflink-udfs/roadmap-of-pyflink.png" width="600px" alt="Roadmap of PyFlink"/> +</center> +<br> diff --git a/img/blog/2020-02-23-pyflink-udfs/pyflink-udf-architecture.png b/img/blog/2020-02-23-pyflink-udfs/pyflink-udf-architecture.png new file mode 100644 index 0000000..d8e56a0 Binary files /dev/null and b/img/blog/2020-02-23-pyflink-udfs/pyflink-udf-architecture.png differ diff --git a/img/blog/2020-02-23-pyflink-udfs/roadmap-of-pyflink.png b/img/blog/2020-02-23-pyflink-udfs/roadmap-of-pyflink.png new file mode 100644 index 0000000..b1f3d36 Binary files /dev/null and b/img/blog/2020-02-23-pyflink-udfs/roadmap-of-pyflink.png differ