blob: b7e0996f5fc14aa3c8da8e2b3b2d084472e2dd89 (
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
|
package dotty.tools
package dotc
package util
import core.Contexts._
import collection.mutable
@sharable object Stats {
final val enabled = false
/** The period in ms in which stack snapshots are displayed */
final val HeartBeatPeriod = 250
var monitored = false
@volatile private var stack: List[String] = Nil
val hits = new mutable.HashMap[String, Int] {
override def default(key: String): Int = 0
}
@inline
def record(fn: String, n: Int = 1) =
if (enabled) doRecord(fn, n)
private def doRecord(fn: String, n: Int) =
if (monitored) {
val name = if (fn.startsWith("member-")) "member" else fn
hits(name) += n
}
@inline
def track[T](fn: String)(op: => T) =
if (enabled) doTrack(fn)(op) else op
def doTrack[T](fn: String)(op: => T) =
if (monitored) {
stack = fn :: stack
record(fn)
try op
finally stack = stack.tail
} else op
class HeartBeat extends Thread() {
@volatile private[Stats] var continue = true
private def printStack(stack: List[String]): Unit = stack match {
case str :: rest =>
printStack(rest)
print(s"-> $str ")
case Nil =>
println()
print("|")
}
override final def run(): Unit = {
Thread.sleep(HeartBeatPeriod)
printStack(stack)
if (continue) run()
}
}
def monitorHeartBeat[T](op: => T)(implicit ctx: Context) = {
if (ctx.settings.Yheartbeat.value) {
var hb = new HeartBeat()
hb.start()
monitored = true
try op
finally {
hb.continue = false
println()
println(hits.toList.sortBy(_._2).map{ case (x, y) => s"$x -> $y" } mkString "\n")
println(s"sizes: ${ctx.base.uniquesSizes}")
}
} else op
}
}
|