[SPARK-48008] Support UDAF in Spark Connect

• Type: Sub-task
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 4.0.0
• Fix Version/s: 4.0.0
• Component/s: Connect
• Labels: pull-request-available
• Assignee: Pengfei Xu
• Reporter: Pengfei Xu
• Created: 26/Apr/24 11:04
• Updated: 15/Aug/24 15:01
• Resolved: 30/May/24 16:51

Description

Currently Spark Connect supports only UDFs. We need to add support for UDAFs, specifically `Aggregator[INT, BUF, OUT]`.

The user-facing API should not change, which includes Aggregator methods and the `spark.udf.register("agg", udaf(agg))` API.

Issue Links

• links to: GitHub Pull Request #46245