Throttler.scala

package org.sdo.alg.other

import java.util.concurrent.atomic.AtomicReference

/**
  * Created by dsemenov
  * Date: 1/28/16.
  */
class Throttler(tps: Int) {
  val maxInterval = tps / 1000
  val lastMessageTimeStamp = new AtomicReference[Long](-1L)

  def acquire(onj: Any) = {
    val lastTimestamp = lastMessageTimeStamp.get()
    val currentMessageTime: Long = System.currentTimeMillis()
    if (lastTimestamp == -1) {
      lastMessageTimeStamp.compareAndSet(lastTimestamp, currentMessageTime)
    } else {
      if (currentMessageTime - lastTimestamp > maxInterval) {
        lastMessageTimeStamp.compareAndSet(lastTimestamp, currentMessageTime)
      } else {
        false
      }
    }
  }
}