aboutsummaryrefslogtreecommitdiff
path: root/kamon-newrelic/src/main/scala/kamon/newrelic/AgentJsonProtocol.scala
blob: b133aee028e6b8ae492bd7fcefe07c79092111bf (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/* ===================================================
 * 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 spray.json._
import kamon.newrelic.Agent._

object AgentJsonProtocol extends DefaultJsonProtocol {

  implicit object ConnectJsonWriter extends RootJsonWriter[AgentInfo] {
    def write(obj: AgentInfo): JsValue =
      JsArray(
        JsObject(
          "agent_version" -> JsString("3.1.0"),
          "app_name"      -> JsArray(JsString(obj.appName)),
          "host"          -> JsString(obj.host),
          "identifier"    -> JsString(s"java:${obj.appName}"),
          "language"      -> JsString("java"),
          "pid"           -> JsNumber(obj.pid)
        )
      )
  }

  import NewRelicMetric._

  implicit def listWriter[T : JsonWriter] = new JsonWriter[List[T]] {
    def write(list: List[T]) = JsArray(list.map(_.toJson))
  }

  implicit object MetricDetailWriter extends JsonWriter[(ID, Data)] {
    def write(obj: (ID, Data)): JsValue = {
      val (id, data) = obj
      JsArray(
        JsObject(
          "name" -> JsString(id.name) // TODO Include scope
        ),
        JsArray(
          JsNumber(data.callCount),
          JsNumber(data.total),
          JsNumber(data.totalExclusive),
          JsNumber(data.min),
          JsNumber(data.max),
          JsNumber(data.sumOfSquares)
        )
      )
    }
  }

  implicit object MetricDataWriter extends RootJsonWriter[MetricData] {
    def write(obj: MetricData): JsValue =
      JsArray(
        JsNumber(obj.runId),
        JsNumber(obj.start),
        JsNumber(obj.end),
        obj.metrics.toJson
      )
  }
}