This is an automated email from the ASF dual-hosted git repository.
gustavodemorais pushed a commit to branch release-2.2
in repository https://gitbox.apache.org/repos/asf/flink.git
The following commit(s) were added to refs/heads/release-2.2 by this push:
new 04efbf15445 [FLINK-39384][python] Fix Java errors when trying to load
or compile a CompiledPlan
04efbf15445 is described below
commit 04efbf15445cb4d7329aac1b8d72e45b784fc5fa
Author: Mika Naylor <[email protected]>
AuthorDate: Thu Jul 2 14:53:17 2026 +0200
[FLINK-39384][python] Fix Java errors when trying to load or compile a
CompiledPlan
This closes #28597.
---
flink-python/pyflink/table/table_environment.py | 4 +-
.../pyflink/table/tests/test_compiled_plan.py | 48 +++++++++++++++++++++-
2 files changed, 49 insertions(+), 3 deletions(-)
diff --git a/flink-python/pyflink/table/table_environment.py
b/flink-python/pyflink/table/table_environment.py
index 539815974b4..b9b95ab0b0b 100644
--- a/flink-python/pyflink/table/table_environment.py
+++ b/flink-python/pyflink/table/table_environment.py
@@ -1595,7 +1595,7 @@ class TableEnvironment(object):
"""
return CompiledPlan(
j_compiled_plan=self._j_tenv.loadPlan(plan_reference._j_plan_reference),
- t_env=self._j_tenv
+ t_env=self
)
def compile_plan_sql(self, stmt: str) -> CompiledPlan:
@@ -1621,7 +1621,7 @@ class TableEnvironment(object):
.. versionadded:: 2.1.0
"""
- return CompiledPlan(j_compiled_plan=self._j_tenv.compilePlanSql(stmt),
t_env=self._j_tenv)
+ return CompiledPlan(j_compiled_plan=self._j_tenv.compilePlanSql(stmt),
t_env=self)
def execute_plan(self, plan_reference: PlanReference) -> TableResult:
"""
diff --git a/flink-python/pyflink/table/tests/test_compiled_plan.py
b/flink-python/pyflink/table/tests/test_compiled_plan.py
index fe0675e5939..9660a36ff54 100644
--- a/flink-python/pyflink/table/tests/test_compiled_plan.py
+++ b/flink-python/pyflink/table/tests/test_compiled_plan.py
@@ -17,6 +17,7 @@
################################################################################
import os.path
import re
+import uuid
from pathlib import Path
from pyflink.table import Schema, DataTypes, TableDescriptor, PlanReference
@@ -26,6 +27,10 @@ THIS_DIR = os.path.dirname(os.path.abspath(__file__))
JSON_PLAN_DIR = os.path.join(THIS_DIR, "jsonplan")
+def generate_random_table_name():
+ return f"Table_{uuid.uuid4().hex}"
+
+
def _replace_flink_version(plan: str) -> str:
"""
Ignore the value for the Flink Version in the compiled plan.
@@ -82,7 +87,7 @@ class CompiledPlanTest(PyFlinkStreamTableTestCase,
PyFlinkTestCase):
table_pipeline = table.insert_into("RegisteredSink")
compiled_plan = table_pipeline.compile_plan()
- plan_path = Path(self.tempdir + "/plan.out")
+ plan_path = Path(self.tempdir) / "plan.out"
compiled_plan.write_to_file(plan_path)
plan_reference_from_file = PlanReference.from_file(plan_path)
@@ -97,6 +102,47 @@ class CompiledPlanTest(PyFlinkStreamTableTestCase,
PyFlinkTestCase):
compiled_plan.as_json_string(),
compiled_plan_from_string.as_json_string()
)
+ def test_load_plan_execute(self):
+ schema = Schema.new_builder().column("f0", DataTypes.STRING()).build()
+ table = self.t_env.from_descriptor(
+ TableDescriptor.for_connector("datagen")
+ .option("number-of-rows", "5")
+ .schema(schema)
+ .build()
+ )
+ self.t_env.create_temporary_table(
+ "Sink",
+ TableDescriptor.for_connector("blackhole").schema(schema).build(),
+ )
+
+ compiled_plan = table.insert_into("Sink").compile_plan()
+ plan_path = Path(self.tempdir) / "plan.json"
+ compiled_plan.write_to_file(plan_path)
+
+ loaded_plan = self.t_env.load_plan(PlanReference.from_file(plan_path))
+ loaded_plan.execute().wait()
+
+ def test_compile_plan_sql_execute(self):
+ source_table = generate_random_table_name()
+ sink_table = generate_random_table_name()
+
+ src = f"""
+ CREATE TABLE {source_table} (a BIGINT, b INT, c VARCHAR)
+ WITH ('connector' = 'datagen', 'number-of-rows' = '5')
+ """
+ self.t_env.execute_sql(src)
+
+ sink = f"""
+ CREATE TABLE {sink_table} (a BIGINT, b INT, c VARCHAR)
+ WITH ('connector' = 'blackhole')
+ """
+ self.t_env.execute_sql(sink)
+
+ compiled_plan = self.t_env.compile_plan_sql(
+ f"INSERT INTO {sink_table} SELECT * FROM {source_table}"
+ )
+ compiled_plan.execute().wait()
+
if __name__ == "__main__":
import unittest
