/* ===================================================
 * Copyright © 2013 the kamon project <http://kamon.io/>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ========================================================== */
package kamon.newrelic

import akka.actor._
import scala.concurrent.duration._
import kamon.Kamon
import kamon.metrics.{TickMetricSnapshotBuffer, TraceMetrics, Metrics}
import kamon.metrics.Subscriptions.TickMetricSnapshot
import akka.actor


class NewRelicExtension(system: ExtendedActorSystem) extends Kamon.Extension {
  val manager: ActorRef = system.actorOf(Props[NewRelicManager], "kamon-newrelic")

  Kamon(Metrics)(system).subscribe(TraceMetrics, "*", manager, permanently = true)
}

class NewRelicManager extends Actor with ActorLogging {
  log.info("Starting the Kamon(NewRelic) extension")

  val agent = context.actorOf(Props[Agent], "agent")
  val translator = context.actorOf(MetricTranslator.props(agent), "translator")
  val buffer = context.actorOf(TickMetricSnapshotBuffer.props(1 minute, translator), "metric-buffer")

  def receive = {
    case tick: TickMetricSnapshot => buffer.forward(tick)
  }
}


object NewRelic extends ExtensionId[NewRelicExtension] with ExtensionIdProvider {
  def lookup(): ExtensionId[_ <: actor.Extension] = NewRelic
  def createExtension(system: ExtendedActorSystem): NewRelicExtension = new NewRelicExtension(system)

  case class Metric(name: String, scope: Option[String], callCount: Long, total: Double, totalExclusive: Double,
                    min: Double, max: Double, sumOfSquares: Double) {

    def merge(that: Metric): Metric = {
      Metric(name, scope,
        callCount + that.callCount,
        total + that.total,
        totalExclusive + that.totalExclusive,
        math.min(min, that.min),
        math.max(max, that.max),
        sumOfSquares + that.sumOfSquares)
    }

  }
}