blob: 64eed1035c1b2a4d8521a86e780c5c119f3bb8af (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
package dotty
package tools
package dotc
package vulpix
import java.io.{
File => JFile,
InputStream, ObjectInputStream,
OutputStream, ObjectOutputStream
}
import scala.concurrent.duration.Duration
import scala.collection.mutable
trait RunnerOrchestration {
/** The maximum amount of active runners, which contain a child JVM */
val numberOfSlaves: Int
/** The maximum duration the child process is allowed to consume before
* getting destroyed
*/
val maxDuration: Duration
/** Destroy and respawn process after each test */
val safeMode: Boolean
/** Running a `Test` class's main method from the specified `dir` */
def runMain(dir: JFile): Status = monitor.runMain(dir)
private[this] val monitor = new RunnerMonitor
private class RunnerMonitor {
def runMain(dir: JFile): Status = withRunner(_.runMain(dir))
private class Runner(private var process: Process) {
def kill(): Unit = ???
def runMain(dir: JFile): Status = ???
}
private def createProcess: Process = ???
private[this] val allRunners =
List.fill(numberOfSlaves)(new Runner(createProcess))
private[this] val freeRunners = mutable.Queue(allRunners: _*)
private[this] val busyRunners = mutable.Set.empty[Runner]
private def getRunner(): Runner = synchronized {
while (freeRunners.isEmpty) wait()
val runner = freeRunners.dequeue()
busyRunners += runner
notify()
runner
}
private def freeRunner(runner: Runner): Unit = synchronized {
freeRunners.enqueue(runner)
busyRunners -= runner
notify()
}
private def withRunner[T](op: Runner => T): T = {
val runner = getRunner()
val result = op(runner)
freeRunner(runner)
result
}
private def killAll(): Unit = allRunners.foreach(_.kill())
// On shutdown, we need to kill all runners:
sys.addShutdownHook(killAll())
// If for some reason the test runner (i.e. sbt) doesn't kill the VM, we
// need to clean up ourselves.
SummaryReport.addCleanup(killAll)
}
}
|