New lazy vals implementation

Author: olsdavis

compiler/src/dotty/tools/dotc/transform/LazyVals.scala

@@ -16,6 +16,8 @@
       val pathEntries = System.getenv("PATH").split(psep).toList
       System.err.printf("sun.java.command: %s\n", sys.props("sun.java.command"))
       System.err.printf("first 5 PATH entries:\n%s\n",pathEntries.take(5).mkString("\n"))
+      printf("script.path: %s\n",path.norm)
+      assert(path.endsWith("scriptPath.sc"),s"actual path [$path]")
     }
 
   extension(s: String)

compiler/test-resources/scripting/scriptPath.sc

@@ -597,7 +597,7 @@ class DottyBytecodeTests extends DottyBytecodeTest {
       val clsIn   = dir.lookupName("Test.class", directory = false).input
       val clsNode = loadClassNode(clsIn)
       val method  = getMethod(clsNode, "test")
-      assertEquals(88, instructionsFromMethod(method).size)
+      assertEquals(122, instructionsFromMethod(method).size)
     }
   }
 

compiler/test/dotty/tools/backend/jvm/DottyBytecodeTests.scala

@@ -22,6 +22,7 @@ object LazyVals {
     val processors = java.lang.Runtime.getRuntime.nn.availableProcessors()
     8 * processors * processors
   }
+
   private[this] val monitors: Array[Object] =
     Array.tabulate(base)(_ => new Object)
 
@@ -37,6 +38,41 @@ object LazyVals {
 
   /* ------------- Start of public API ------------- */
 
+  /**
+   * Used to indicate the state of a lazy val that is being
+   * evaluated and of which other threads await the result.
+   */
+  final class Waiting:
+    private var done = false
+    
+    /**
+     * Wakes up waiting threads. Called on completion of the evaluation
+     * of lazy val's right-hand side.
+     */
+    def release(): Unit = synchronized {
+      done = true
+      notifyAll()
+    }
+
+    /**
+     * Awaits the completion of the evaluation of lazy val's right-hand side.
+     */
+    def awaitRelease(): Unit = synchronized {
+      while !done do wait()
+    }
+
+  /**
+   * Used to indicate the state of a lazy val that is currently being
+   * evaluated with no other thread awaiting its result.
+   */
+  object Evaluating
+
+  /**
+   * Used to indicate the state of a lazy val that has been evaluated to
+   * `null`.
+   */
+  object NULL
+
   final val BITS_PER_LAZY_VAL = 2L
 
   def STATE(cur: Long, ord: Int): Long = {
@@ -54,6 +90,12 @@ object LazyVals {
     unsafe.compareAndSwapLong(t, offset, e, n)
   }
 
+  def objCAS(t: Object, offset: Long, exp: Object, n: Object): Boolean = {
+    if (debug)
+      println(s"objCAS($t, $exp, $n)")
+    unsafe.compareAndSwapObject(t, offset, exp, n)
+  }
+
   def setFlag(t: Object, offset: Long, v: Int, ord: Int): Unit = {
     if (debug)
       println(s"setFlag($t, $offset, $v, $ord)")
@@ -106,6 +148,13 @@ object LazyVals {
     r
   }
 
+  def getStaticOffset(clz: Class[_], name: String): Long = {
+    val r = unsafe.staticFieldOffset(clz.getDeclaredField(name))
+    if (debug)
+      println(s"getStaticOffset($clz, $name) = $r")
+    r
+  }
+
   def getOffsetStatic(field: java.lang.reflect.Field) =
     val r = unsafe.objectFieldOffset(field)
     if (debug)
@@ -114,11 +163,18 @@ object LazyVals {
 
 
   object Names {
+    final val waiting = "Waiting"
+    final val evaluating = "Evaluating"
+    final val nullValued = "NULL"
+    final val waitingAwaitRelease = "awaitRelease"
+    final val waitingRelease = "release"
     final val state = "STATE"
     final val cas = "CAS"
+    final val objCas = "objCAS"
     final val setFlag = "setFlag"
     final val wait4Notification = "wait4Notification"
     final val get = "get"
     final val getOffset = "getOffset"
+    final val getStaticOffset = "getStaticOffset"
   }
 }

library/src/scala/runtime/LazyVals.scala

@@ -10,5 +10,28 @@ object MiMaFilters {
     ProblemFilters.exclude[DirectMissingMethodProblem]("scala.quoted.runtime.QuoteUnpickler.unpickleTypeV2"),
 
     ProblemFilters.exclude[MissingClassProblem]("scala.annotation.since"),
+
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals.getStaticOffset"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals.getStaticOffset"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals.objCAS"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals.objCAS"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals#Names.evaluating"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals#Names.getStaticOffset"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals#Names.nullValued"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals#Names.objCas"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals#Names.waiting"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals#Names.waitingAwaitRelease"),
+    ProblemFilters.exclude[DirectMissingMethodProblem]("scala.runtime.LazyVals#Names.waitingRelease"),
+    ProblemFilters.exclude[MissingClassProblem]("scala.runtime.LazyVals$Evaluating$"),
+    ProblemFilters.exclude[MissingClassProblem]("scala.runtime.LazyVals$NULL$"),
+    ProblemFilters.exclude[MissingClassProblem]("scala.runtime.LazyVals$Waiting"),
+    ProblemFilters.exclude[MissingFieldProblem]("scala.runtime.LazyVals.Evaluating"),
+    ProblemFilters.exclude[MissingFieldProblem]("scala.runtime.LazyVals.NULL"),
+
+    // Experimental `MainAnnotation` APIs. Can be added in 3.3.0 or later.
+    // MiMa bug: classes nested in an experimental object should be ignored
+    ProblemFilters.exclude[MissingClassProblem]("scala.annotation.MainAnnotation$Info"),
+    ProblemFilters.exclude[MissingClassProblem]("scala.annotation.MainAnnotation$Parameter"),
+    ProblemFilters.exclude[MissingClassProblem]("scala.annotation.MainAnnotation$ParameterAnnotation"),
   )
 }

project/MiMaFilters.scala

@@ -1,7 +1,11 @@
 -- Error: tests/init/neg/promotion-loop.scala:16:10 --------------------------------------------------------------------
 16 |  println(b)      // error
    |          ^
+<<<<<<< HEAD
+   |          Cannot prove that the value is fully-initialized. Only initialized values may be used as arguments.
+=======
    |          Cannot prove that the value is fully initialized. Only initialized values may be used as arguments.
+>>>>>>> 91c03b1f7835d574290eba1ecf236e0f59fc2bf0
    |
    |          The unsafe promotion may cause the following problem:
    |          Cannot prove that the value is fully initialized. Only initialized values may be used as arguments.

tests/init/neg/promotion-loop.check

@@ -1,7 +1,7 @@
 -- Error: tests/init/neg/t3273.scala:4:42 ------------------------------------------------------------------------------
 4 |  val num1: LazyList[Int] = 1 #:: num1.map(_ + 1)                        // error
   |                                  ^^^^^^^^^^^^^^^
-  |              Cannot prove that the value is fully initialized. Only initialized values may be used as arguments.
+  |             Cannot prove that the value is fully-initialized. Only initialized values may be used as arguments.
   |
   |              The unsafe promotion may cause the following problem:
   |              Access non-initialized value num1. Calling trace:
@@ -10,7 +10,7 @@
 -- Error: tests/init/neg/t3273.scala:5:61 ------------------------------------------------------------------------------
 5 |  val num2: LazyList[Int] = 1 #:: num2.iterator.map(_ + 1).to(LazyList)  // error
   |                                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
-  |              Cannot prove that the value is fully initialized. Only initialized values may be used as arguments.
+  |             Cannot prove that the value is fully-initialized. Only initialized values may be used as arguments.
   |
   |              The unsafe promotion may cause the following problem:
   |              Access non-initialized value num2. Calling trace:

tests/init/neg/t3273.check

@@ -0,0 +1,4 @@
+computing x
+x
+computing y
+y

tests/run/lazyVals_c3.0.0.check

@@ -0,0 +1,13 @@
+// Compiled with 3.0.0 and run with current compiler
+class Foo:
+  lazy val x =
+    println("computing x")
+    "x"
+  lazy val y =
+    println("computing y")
+    "y"
+
+@main def Test =
+  val foo = new Foo
+  println(foo.x)
+  println(foo.y)

tests/run/lazyVals_c3.0.0.scala

@@ -0,0 +1,4 @@
+computing x
+x
+computing y
+y

tests/run/lazyVals_c3.1.0.check

@@ -0,0 +1,13 @@
+// Compiled with 3.1.0 and run with current compiler
+class Foo:
+  lazy val x =
+    println("computing x")
+    "x"
+  lazy val y =
+    println("computing y")
+    "y"
+
+@main def Test =
+  val foo = new Foo
+  println(foo.x)
+  println(foo.y)