Skip to content

[SPARK-47818][CONNECT][FOLLOW-UP] Introduce plan cache in SparkConnectPlanner to improve performance of Analyze requests - #46098

Closed
xi-db wants to merge 3 commits into
apache:masterfrom
xi-db:SPARK-47818-plan-cache-followup
Closed

[SPARK-47818][CONNECT][FOLLOW-UP] Introduce plan cache in SparkConnectPlanner to improve performance of Analyze requests#46098
xi-db wants to merge 3 commits into
apache:masterfrom
xi-db:SPARK-47818-plan-cache-followup

Conversation

@xi-db

@xi-db xi-db commented Apr 17, 2024

Copy link
Copy Markdown
Contributor

What changes were proposed in this pull request?

In the previous PR, we cache plans of AnalyzePlan requests. We're also enabling it for ExecutePlan in this PR.

Why are the changes needed?

Some operations like spark.sql() issue ExecutePlan requests. By caching them, we can also improve performance if subsequent plans to be analyzed include the plan returned by ExecutePlan as a subtree.

Does this PR introduce any user-facing change?

No.

How was this patch tested?

Existing tests.

Was this patch authored or co-authored using generative AI tooling?

No.

@vicennial

Copy link
Copy Markdown
Contributor

cc @ueshin @zhengruifeng

@xi-db xi-db changed the title [WIP][SPARK-47818][CONNECT][FOLLOW-UP] Introduce plan cache in SparkConnectPlanner to improve performance of Analyze requests [SPARK-47818][CONNECT][FOLLOW-UP] Introduce plan cache in SparkConnectPlanner to improve performance of Analyze requests Apr 17, 2024
with self.sql_conf({"spark.sql.pyspark.udf.profiler": "perf"}):
_do_computation(self.spark, action=action)

# Without the plan cache, UDF ID will be different for each action

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

also cc @xinrong-meng to check the profiler tests

@hvanhovell

Copy link
Copy Markdown
Contributor

@xi-db please update the PR.

# Conflicts:
#	connector/connect/server/src/main/scala/org/apache/spark/sql/connect/execution/SparkConnectPlanExecution.scala
@xi-db

xi-db commented Apr 25, 2024

Copy link
Copy Markdown
Contributor Author

@xi-db please update the PR.

Hi @hvanhovell, the CI is green and its ready to merge.

@hvanhovell

Copy link
Copy Markdown
Contributor

Merging.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants