gengliangwang commented on code in PR #45729: URL: https://github.com/apache/spark/pull/45729#discussion_r1540122993
########## common/utils/src/main/scala/org/apache/spark/internal/Logging.scala: ########## @@ -55,6 +76,37 @@ trait Logging { log_ } + implicit class LogStringContext(val sc: StringContext) { + def log(args: Any*): (String, Instance) = { + val processedParts = sc.parts.iterator + val sb = new StringBuilder(processedParts.next()) + lazy val map = new java.util.HashMap[String, String]() + + args.foreach { arg => + arg match { + case mdc: MDC => + sb.append(mdc.value) + if (Logging.isStructuredLoggingEnabled) { + map.put(mdc.key.toString.toLowerCase(Locale.ROOT), mdc.value) + } + case other => + throw new IllegalArgumentException(s"Argument $other is not a MDC") + } + if (processedParts.hasNext) { + sb.append(processedParts.next()) + } + } + + // Create a CloseableThreadContext and apply the context map + val closeableContext = if (Logging.isStructuredLoggingEnabled) { + CloseableThreadContext.putAll(map) + } else { + null Review Comment: This is to avoid the extra overhead of creating a CloseableThreadContext instance. I am changing this to None for safety. -- 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: reviews-unsubscr...@spark.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org