Question

我想做这样的事情：

def warnEvery[A](duration: Duration)(block: => A): A = {
   val start = DateTime.now
   val f = Future(f)
   while(!f.isComplete) {
     Thread.sleep(duration)
     if (!f.isComplete) {
       // Trigger a stack trace warning message
       Log.warn(s"Block is running for ${DateTime.now() - start}", new Throwable())
     }   
   }
   f.result
}

目标是以这种方式使用上述帮助：

warnEvery(1 minute) {
  // slow operation
}

在我的日志中，我希望看到一个缓慢的代码块的堆栈跟踪。

在Scala中实现此目的的最佳方法是什么？指向现有图书馆的指针也很好。

Answer 1

您可以使用Akka的调度程序来安排检查未来完成情况的任务。这样，就不需要Thread.sleep，它会在任务完成之前阻塞其线程。

def warnEvery[A](duration: FiniteDuration)(block: => A): Future[A] = {
  val start = DateTime.now
  val f = Future(f)
  def warn {
    if (!f.isCompleted) {
      Log.warn(s"Block is running for ${DateTime.now() - start}")
      Akka.system.scheduler.scheduleOnce(duration)(warn)
    }
  }
  Akka.system.scheduler.scheduleOnce(duration)(warn)
  f
}

轻微的挑剔：从技术上讲，这个解决方案衡量的是创造未来和现在之间的时间。如果在创建未来时执行上下文中的所有线程都忙，则可能稍后开始执行该块。因此，更好的日志消息将是＆＃34; Block已经排队等待执行......几秒钟之前还没有完成运行。＆＃34;

Answer 2

只需运行该块，最后检查它是否已运行太长时间

import org.joda.time.{Interval, DateTime}
import scala.concurrent.duration.Duration

def warnEvery[A](duration: Duration)(block: => A): A = {
  val start = DateTime.now
  try {
    block
  }
  finally {
    val millis = new Interval(start, DateTime.now).toDurationMillis

    if (millis > duration.toMillis) {
      // Log here ...
    }
  }
}

编辑：对不起，没有正确理解你的问题。然后你可以使用计时器 -

import java.util.{TimerTask, Timer}

import org.joda.time.{Interval, DateTime}
import scala.concurrent.duration.Duration

def warnEvery[A](duration: Duration)(block: => A): A = {
  val task = new TimerTask() {
    @Override
    def run() {
      // Log here
    }
  }

  val timer = new Timer()
  val delay = duration.toMillis
  val intevalPeriod = duration.toMillis

  timer.scheduleAtFixedRate(task, delay, intevalPeriod)

  try {
    block
  }
  finally {
    timer.cancel()
  }
}

Scala：如果代码块执行的时间超过一定时间，请发出警告吗？

2 个答案: