pan3793 commented on code in PR #38357:
URL: https://github.com/apache/spark/pull/38357#discussion_r1054620568
##########
core/src/main/scala/org/apache/spark/scheduler/SchedulerBackend.scala:
##########
@@ -74,14 +76,24 @@ private[spark] trait SchedulerBackend {
* Executors tab for the driver.
* @return Map containing the log names and their respective URLs
*/
- def getDriverLogUrls: Option[Map[String, String]] = None
+ def getDriverLogUrls: Option[Map[String, String]] = {
+ val prefix = "SPARK_DRIVER_LOG_URL_"
+ val driverLogUrls = sys.env.filterKeys(_.startsWith(prefix))
+ .map(e => (e._1.substring(prefix.length).toLowerCase(Locale.ROOT),
e._2)).toMap
+ if (driverLogUrls.nonEmpty) Some(driverLogUrls) else None
+ }
/**
* Get the attributes on driver. These attributes are used to replace log
URLs when
* custom log url pattern is specified.
* @return Map containing attributes on driver.
*/
- def getDriverAttributes: Option[Map[String, String]] = None
+ def getDriverAttributes: Option[Map[String, String]] = {
+ val prefix = "SPARK_DRIVER_ATTRIBUTE_"
+ val driverAttributes = sys.env.filterKeys(_.startsWith(prefix))
+ .map(e => (e._1.substring(prefix.length).toUpperCase(Locale.ROOT),
e._2)).toMap
+ if (driverAttributes.nonEmpty) Some(driverAttributes) else None
Review Comment:
There are few differences, e.g. `toLowerCase` and `toUpperCase`, this is
following `CoarseGrainedExecutorBackend#extractLogUrls` and
`CoarseGrainedExecutorBackend#extractAttributes`
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
