/*
* =========================================================================================
* Copyright © 2013-2014 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.metrics
import akka.actor.ActorSystem
import com.typesafe.config.Config
import kamon.metric._
import kamon.metric.instrument.Gauge.CurrentValueCollector
import kamon.metric.instrument.{ Gauge, Histogram }
import kamon.system.SigarExtensionProvider
import org.hyperic.sigar.{ NetInterfaceStat, SigarProxy }
case class NetworkMetrics(name: String) extends MetricGroupIdentity {
val category = NetworkMetrics
}
object NetworkMetrics extends MetricGroupCategory {
val name = "network"
case object RxBytes extends MetricIdentity { val name = "rx-bytes" }
case object TxBytes extends MetricIdentity { val name = "tx-bytes" }
case object RxErrors extends MetricIdentity { val name = "rx-errors" }
case object TxErrors extends MetricIdentity { val name = "tx-errors" }
case class NetworkMetricRecorder(rxBytes: Gauge, txBytes: Gauge, rxErrors: Gauge, txErrors: Gauge)
extends MetricGroupRecorder {
def collect(context: CollectionContext): MetricGroupSnapshot = {
NetworkMetricSnapshot(rxBytes.collect(context), txBytes.collect(context), rxErrors.collect(context), txErrors.collect(context))
}
def cleanup: Unit = {}
}
case class NetworkMetricSnapshot(rxBytes: Histogram.Snapshot, txBytes: Histogram.Snapshot, rxErrors: Histogram.Snapshot, txErrors: Histogram.Snapshot)
extends MetricGroupSnapshot {
type GroupSnapshotType = NetworkMetricSnapshot
def merge(that: GroupSnapshotType, context: CollectionContext): GroupSnapshotType = {
NetworkMetricSnapshot(rxBytes.merge(that.rxBytes, context), txBytes.merge(that.txBytes, context), rxErrors.merge(that.rxErrors, context), txErrors.merge(that.txErrors, context))
}
val metrics: Map[MetricIdentity, MetricSnapshot] = Map(
RxBytes -> rxBytes,
TxBytes -> txBytes,
RxErrors -> rxErrors,
TxErrors -> txErrors)
}
val Factory = new MetricGroupFactory with SigarExtensionProvider {
val interfaces: Set[String] = sigar.getNetInterfaceList.toSet
type GroupRecorder = NetworkMetricRecorder
def create(config: Config, system: ActorSystem): GroupRecorder = {
val settings = config.getConfig("precision.system.network")
val rxBytesConfig = settings.getConfig("rx-bytes")
val txBytesConfig = settings.getConfig("tx-bytes")
val rxErrorsConfig = settings.getConfig("rx-errors")
val txErrorsConfig = settings.getConfig("tx-errors")
new NetworkMetricRecorder(
Gauge.fromConfig(rxBytesConfig, system)(collect(sigar, interfaces)(net ⇒ net.getRxBytes)),
Gauge.fromConfig(txBytesConfig, system)(collect(sigar, interfaces)(net ⇒ net.getTxBytes)),
Gauge.fromConfig(rxErrorsConfig, system)(collect(sigar, interfaces)(net ⇒ net.getRxErrors)),
Gauge.fromConfig(txErrorsConfig, system)(collect(sigar, interfaces)(net ⇒ net.getTxErrors)))
}
private def collect(sigar: SigarProxy, interfaces: Set[String])(block: NetInterfaceStat ⇒ Long) = () ⇒ {
interfaces.foldLeft(0L) { (totalBytes, interface) ⇒
{
val net = sigar.getNetInterfaceStat(interface)
totalBytes + block(net)
}
}
}
}
}