blob: a535f203c55bd7ad3cb9da1aed15a6288d29ab99 (
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
|
package dotty.tools.dotc
package util
import core.Contexts._
import collection.mutable
object Stats {
final val enabled = true
/** The period in ms in which stack snapshots are displayed */
final val HeartBeatPeriod = 250
@volatile private var stack: List[String] = Nil
val hits = new mutable.HashMap[String, Int] {
override def default(key: String): Int = 0
}
def record(fn: String, n: Int = 1) = {
if (monitored) {
val name = if (fn.startsWith("member-")) "member" else fn
hits(name) += n
}
}
var monitored = false
def track[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.uniquesSize}")
}
} else op
}
}
|