aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/xyz/driver/tracing/google/api.scala
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/scala/xyz/driver/tracing/google/api.scala')
-rw-r--r--src/main/scala/xyz/driver/tracing/google/api.scala111
1 files changed, 111 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/tracing/google/api.scala b/src/main/scala/xyz/driver/tracing/google/api.scala
new file mode 100644
index 0000000..2dcac92
--- /dev/null
+++ b/src/main/scala/xyz/driver/tracing/google/api.scala
@@ -0,0 +1,111 @@
+package xyz.driver.tracing
+package google
+
+import spray.json._
+import spray.json.DefaultJsonProtocol._
+import java.util.UUID
+import java.nio.ByteBuffer
+import java.time._
+import java.time.format._
+
+case class TraceSpan(
+ spanId: Long,
+ kind: TraceSpan.SpanKind,
+ name: String,
+ startTime: Instant,
+ endTime: Instant,
+ parentSpanId: Option[Long],
+ labels: Map[String, String]
+)
+
+object TraceSpan {
+
+ sealed trait SpanKind
+ // Unspecified
+ case object Unspecified extends SpanKind
+ // Indicates that the span covers server-side handling of an RPC or other remote network request.
+ case object RpcServer extends SpanKind
+ // Indicates that the span covers the client-side wrapper around an RPC or other remote request.
+ case object RpcClient extends SpanKind
+
+ object SpanKind {
+ implicit val format: JsonFormat[SpanKind] = new JsonFormat[SpanKind] {
+ override def write(x: SpanKind): JsValue = x match {
+ case Unspecified => JsString("SPAN_KIND_UNSPECIFIED")
+ case RpcServer => JsString("RPC_SERVER")
+ case RpcClient => JsString("RPC_CLIENT")
+ }
+ override def read(x: JsValue): SpanKind = x match {
+ case JsString("SPAN_KIND_UNSPECIFIED") => Unspecified
+ case JsString("RPC_SERVER") => RpcServer
+ case JsString("RPC_CLIENT") => RpcClient
+ case other =>
+ spray.json.deserializationError(s"`$other` is not a valid span kind")
+ }
+ }
+ }
+
+ implicit val instantFormat = new JsonFormat[Instant] {
+ val formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd'T'HH:mm:ssXXXZ")
+ override def write(x: Instant): JsValue = JsString(formatter.format(x))
+ override def read(x: JsValue): Instant = x match {
+ case JsString(x) => Instant.parse(x)
+ case other =>
+ spray.json.deserializationError(s"`$other` is not a valid instant")
+ }
+ }
+
+ implicit val format: JsonFormat[TraceSpan] = jsonFormat7(TraceSpan.apply)
+
+ def fromSpan(span: Span) = TraceSpan(
+ span.spanId.getLeastSignificantBits,
+ Unspecified,
+ span.name,
+ span.startTime,
+ span.endTime,
+ span.parentSpanId.map(_.getLeastSignificantBits),
+ span.labels
+ )
+
+}
+
+case class Trace(
+ traceId: UUID,
+ projectId: String = "",
+ spans: Seq[TraceSpan] = Seq.empty
+)
+
+object Trace {
+
+ implicit val uuidFormat = new JsonFormat[UUID] {
+ override def write(x: UUID) = {
+ val buffer = ByteBuffer.allocate(16)
+ buffer.putLong(x.getMostSignificantBits)
+ buffer.putLong(x.getLeastSignificantBits)
+ val array = buffer.array()
+ val string = new StringBuilder
+ for (i <- 0 until 16) {
+ string ++= f"${array(i) & 0xff}%02x"
+ }
+ JsString(string.result)
+ }
+ override def read(x: JsValue): UUID = x match {
+ case JsString(str) if str.length == 32 =>
+ val (msb, lsb) = str.splitAt(16)
+ new UUID(java.lang.Long.decode(msb), java.lang.Long.decode(lsb))
+ case JsString(str) =>
+ spray.json.deserializationError(
+ "128-bit id string must be exactly 32 characters long")
+ case other =>
+ spray.json.deserializationError("expected 32 character hex string")
+ }
+ }
+
+ implicit val format: JsonFormat[Trace] = jsonFormat3(Trace.apply)
+
+}
+
+case class Traces(traces: Seq[Trace])
+object Traces {
+ implicit val format: JsonFormat[Traces] = jsonFormat1(Traces.apply)
+}