epic.util.ProgressLog.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of epic_2.10 Show documentation

epic

The newest version!

package epic.util

import com.typesafe.scalalogging.slf4j.Logger
import java.util.concurrent.atomic.AtomicInteger

/**
 *
 *
 * @author dlwh
 */
class ProgressLog(log: Logger, items: Int, frequency: Int = 100, name: String = "Progress") {
  val initialTime = System.currentTimeMillis()
  val item = new AtomicInteger()


  def reportProgress() = {
    val x = item.incrementAndGet()
    if(x % frequency == 0 || x == items) {
      log.info(s"$name $x/$items (${(System.currentTimeMillis() - initialTime)/1000.0}s elapsed.)")
    }
  }

  def info(msg: =>String) = {
    val x = item.incrementAndGet()
    if(x % frequency == 0 || x == items) {
      val m = msg
      log.info(s"$name $x/$items: $m (${(System.currentTimeMillis() - initialTime)/1000.0}s elapsed.)")
    }
  }

  def debug(msg: =>String) = {
    val x = item.incrementAndGet()
    if(x % frequency == 0 || x == items) {
      val m = msg
      log.debug(s"$name $x/$items: $m (${(System.currentTimeMillis() - initialTime)/1000.0}s elapsed.)")
    }
  }

}