cache line number and position comment parses

Author: lihaoyi

sourcecode/src-2/sourcecode/Macros.scala

@@ -1,5 +1,6 @@
 package sourcecode
 
+import java.util.concurrent.ConcurrentHashMap
 import language.experimental.macros
 
 trait NameMacros {
@@ -95,36 +96,47 @@ object Macros {
   }
 
   private val filePrefix = "//SOURCECODE_ORIGINAL_FILE_PATH="
+  private val filePrefixCache =
+    new ConcurrentHashMap[scala.reflect.internal.util.SourceFile, Option[String]]()
   private def findOriginalFile(chars: Array[Char]): Option[String] = {
     new String(chars).linesIterator.find(_.contains(filePrefix)).map(_.split(filePrefix).last)
   }
 
   def fileImpl(c: Compat.Context): c.Expr[sourcecode.File] = {
     import c.universe._
 
-    val file = findOriginalFile(c.enclosingPosition.source.content)
+    val file = filePrefixCache
+      .computeIfAbsent(c.enclosingPosition.source, source => findOriginalFile(source.content))
       .getOrElse(c.enclosingPosition.source.path)
 
     c.Expr[sourcecode.File](q"""${c.prefix}($file)""")
   }
 
   def fileNameImpl(c: Compat.Context): c.Expr[sourcecode.FileName] = {
     import c.universe._
-    val fileName = findOriginalFile(c.enclosingPosition.source.content)
+    val fileName = filePrefixCache
+      .computeIfAbsent(c.enclosingPosition.source, source => findOriginalFile(source.content))
       .getOrElse(c.enclosingPosition.source.path)
       .split('/').last
     c.Expr[sourcecode.FileName](q"""${c.prefix}($fileName)""")
   }
 
   private val linePrefix = "//SOURCECODE_ORIGINAL_CODE_START_MARKER"
+  private val linePrefixCache =
+    new ConcurrentHashMap[scala.reflect.internal.util.SourceFile, Int]()
+  
   def lineImpl(c: Compat.Context): c.Expr[sourcecode.Line] = {
     import c.universe._
-    val offset = new String(c.enclosingPosition.source.content)
-      .linesIterator
-      .indexWhere(_.contains(linePrefix)) match{
-      case -1 => 0
-      case n => n
-    }
+    val offset = linePrefixCache.computeIfAbsent(
+      c.enclosingPosition.source,
+      source => new String(source.content)
+        .linesIterator
+        .indexWhere(_.contains(linePrefix)) match{
+        case -1 => 0
+        case n => n
+      }
+    )
+
     val line = c.enclosingPosition.line - offset
     c.Expr[sourcecode.Line](q"""${c.prefix}($line)""")
   }

sourcecode/src-3/sourcecode/Macros.scala

@@ -1,7 +1,8 @@
 package sourcecode
 
+import java.util.concurrent.ConcurrentHashMap
 import scala.language.implicitConversions
-import scala.quoted._
+import scala.quoted.*
 
 trait NameMacros {
   inline implicit def generate: Name =
@@ -157,35 +158,48 @@ object Macros {
   }
 
   private val filePrefix = "//SOURCECODE_ORIGINAL_FILE_PATH="
-
+  private val filePrefixCache =
+    new ConcurrentHashMap[Any, Option[String]]()
   private def findOriginalFile(chars: Array[Char]): Option[String] = {
     new String(chars).linesIterator.find(_.contains(filePrefix)).map(_.split(filePrefix).last)
   }
   def fileImpl(using Quotes): Expr[sourcecode.File] = {
     import quotes.reflect._
-    val file = quotes.reflect.Position.ofMacroExpansion.sourceFile.content
+    val file = filePrefixCache.computeIfAbsent(
+      quotes.reflect.Position.ofMacroExpansion.sourceFile,
+      _ => quotes.reflect.Position.ofMacroExpansion.sourceFile.content
       .flatMap(s => findOriginalFile(s.toCharArray))
+    )
       .getOrElse(quotes.reflect.Position.ofMacroExpansion.sourceFile.path)
     '{new sourcecode.File(${Expr(file)})}
   }
 
   def fileNameImpl(using Quotes): Expr[sourcecode.FileName] = {
-    val name = quotes.reflect.Position.ofMacroExpansion.sourceFile.content
-      .flatMap(s => findOriginalFile(s.toCharArray).map(_.split('/').last))
+    val name = filePrefixCache.computeIfAbsent(
+      quotes.reflect.Position.ofMacroExpansion.sourceFile,
+      _ => quotes.reflect.Position.ofMacroExpansion.sourceFile.content
+        .flatMap(s => findOriginalFile(s.toCharArray).map(_.split('/').last)
+    ).flatMap(s => findOriginalFile(s.toCharArray)))
       .getOrElse(quotes.reflect.Position.ofMacroExpansion.sourceFile.name)
 
     '{new sourcecode.FileName(${Expr(name)})}
   }
 
   private val linePrefix = "//SOURCECODE_ORIGINAL_CODE_START_MARKER"
+  private val linePrefixCache =
+    new ConcurrentHashMap[Any, Int]()
   def lineImpl(using Quotes): Expr[sourcecode.Line] = {
-    val offset = quotes.reflect.Position.ofMacroExpansion.sourceFile.content
-      .iterator
-      .flatMap(_.linesIterator)
-      .indexWhere(_.contains(linePrefix)) match{
-      case -1 => 0
-      case n => n
-    }
+    val offset = linePrefixCache.computeIfAbsent(
+      quotes.reflect.Position.ofMacroExpansion.sourceFile,
+      _ =>
+        quotes.reflect.Position.ofMacroExpansion.sourceFile.content
+          .iterator
+          .flatMap(_.linesIterator)
+          .indexWhere(_.contains(linePrefix)) match{
+          case -1 => 0
+          case n => n
+        }
+    )
     val line = quotes.reflect.Position.ofMacroExpansion.startLine + 1 - offset
     '{new sourcecode.Line(${Expr(line)})}
   }