blob: 09dfd894d0260a8cbc221384b519a72ed56573c5 (
plain) (
tree)
|
|
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.uniquesSizes}")
}
} else op
}
}
|