HeartSaVioR commented on a change in pull request #27664:
URL: https://github.com/apache/spark/pull/27664#discussion_r429313023
##########
File path:
sql/core/src/test/scala/org/apache/spark/sql/execution/streaming/FileStreamSinkLogSuite.scala
##########
@@ -267,4 +308,38 @@ class FileStreamSinkLogSuite extends SparkFunSuite with
SharedSparkSession {
val log = new FileStreamSinkLog(FileStreamSinkLog.VERSION, spark,
input.toString)
log.allFiles()
}
+
+ private def withCountOpenLocalFileSystemAsLocalFileSystem(body: => Unit):
Unit = {
+ val optionKey = s"fs.${CountOpenLocalFileSystem.scheme}.impl"
+ val originClassForLocalFileSystem = spark.conf.getOption(optionKey)
+ try {
+ spark.conf.set(optionKey, classOf[CountOpenLocalFileSystem].getName)
+ body
+ } finally {
+ originClassForLocalFileSystem match {
+ case Some(fsClazz) => spark.conf.set(optionKey, fsClazz)
+ case _ => spark.conf.unset(optionKey)
+ }
+ }
+ }
+}
+
+class CountOpenLocalFileSystem extends RawLocalFileSystem {
+ import CountOpenLocalFileSystem._
+
+ override def getUri: URI = {
+ URI.create(s"$scheme:///")
+ }
+
+ override def open(f: Path, bufferSize: Int): FSDataInputStream = {
+ val path = f.toUri.getPath
+ val curVal = pathToNumOpenCalled.getOrElseUpdate(path, new AtomicLong(0))
+ curVal.incrementAndGet()
+ super.open(f, bufferSize)
+ }
+}
+
+object CountOpenLocalFileSystem {
+ val scheme = s"FileStreamSinkLogSuite${math.abs(Random.nextInt)}fs"
+ val pathToNumOpenCalled = new mutable.HashMap[String, AtomicLong]
Review comment:
Yeah, I feel it's safer to make the map be thread-safe. I thought it's a
contradiction to assume because we allow "resetting" the count, but it can be
handled with caution.
----------------------------------------------------------------
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.
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
