Skip to content

[AURON #1531] Enable SparkAuronConfiguration and deprecated AuronConf#1532

Merged
richox merged 5 commits intoapache:masterfrom
Tartarus0zm:enable_SparkAuronConfig
Oct 31, 2025
Merged

[AURON #1531] Enable SparkAuronConfiguration and deprecated AuronConf#1532
richox merged 5 commits intoapache:masterfrom
Tartarus0zm:enable_SparkAuronConfig

Conversation

@Tartarus0zm
Copy link
Contributor

@Tartarus0zm Tartarus0zm commented Oct 27, 2025

Which issue does this PR close?

Closes #1531.

Rationale for this change

We have enabled the AuronAdaptor introduced in auron-core, which serves as the bridge between the native engine and the upper-layer engine. We now need to enable SparkAuronConfiguration and deprecate AuronConf for the following reasons:

  • SparkAuronConfiguration is extends AuronConfiguration, facilitates interaction with native engines.
  • SparkAuronConfiguration is more standardized and includes description, facilitates configuration management.

What changes are included in this PR?

Replace AuronConf in classes such as AuronConverters/NativeConverters/NativeHelper/SparkUDAFWrapperContext/SparkUDTFWrapperContext with SparkAuronConfiguration.
and add @Deprecated for AuronConf.

Are there any user-facing changes?

No

How was this patch tested?

CI testing verification, SparkAuronConfiguration has been test with SparkAuronConfigurationTest.

@github-actions github-actions bot added the spark label Oct 27, 2025
@Tartarus0zm
Copy link
Contributor Author

@richox @merrily01 CI is green. PTAL

Copy link
Member

@merrily01 merrily01 left a comment

Choose a reason for hiding this comment

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

Thanks @Tartarus0zm

private val tc = TaskContext.get()
private val batchSize = AuronConf.BATCH_SIZE.intConf()
private val maxBatchMemorySize = AuronConf.SUGGESTED_BATCH_MEM_SIZE.intConf()
private val sparkAuronConfig: AuronConfiguration =
Copy link
Contributor

Choose a reason for hiding this comment

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

Should this be a method on AuronAdaptor? All the invocations of the adaptor to get the configuration requests a getInstance call; we can abstract this away by adding getSparkAuronConfig() as a public method in the Adaptor, WDYT?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Should this be a method on AuronAdaptor? All the invocations of the adaptor to get the configuration requests a getInstance call; we can abstract this away by adding getSparkAuronConfig() as a public method in the Adaptor, WDYT?

@ShreyeshArangath thanks for your comment. This is an excellent suggestion. I suggest we open a separate issue to discuss this. We introduced the AuronAdaptor to abstract away specific engines like Spark, Flink, and Trino, so the API does not carry the engine name. Let's discuss this issue in #1535.

@SteNicholas
Copy link
Member

@Tartarus0zm, you'd better to improve the description of this pull request. Current description is too simple.

private val extConvertProviders = ServiceLoader.load(classOf[AuronConvertProvider]).asScala

private val sparkAuronConfig: AuronConfiguration =
AuronAdaptor.getInstance.getAuronConfiguration
Copy link
Member

Choose a reason for hiding this comment

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

Could we add getAuronConfiguration in AuronAdaptor?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

@SteNicholas The API getAuronConfiguration already exists within the AuronAdaptor. Since the AuronAdaptor use the singleton pattern, the getInstance API must be called first each time.
Let's discuss this issue in #1535.

@Tartarus0zm
Copy link
Contributor Author

@Tartarus0zm, you'd better to improve the description of this pull request. Current description is too simple.

@SteNicholas Thanks for the reminder. I have updated the description.

@richox richox merged commit 923db4d into apache:master Oct 31, 2025
98 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Enable SparkAuronConfiguration and deprecated AuronConf

6 participants