aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/Span.scala
blob: fcd52b6258f4e7b3b2510eb064c0187f8d9680df (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
package xyz.driver.tracing

import java.util.UUID
import java.time._

case class Span(
    name: String,
    traceId: UUID = UUID.randomUUID(),
    spanId: UUID = UUID.randomUUID(),
    parentSpanId: Option[UUID] = None,
    labels: Map[String, String] = Map.empty,
    startTime: Instant = Instant.now,
    endTime: Instant = Instant.now
) {

  def start(clock: Clock = Clock.systemUTC): Span =
    this.copy(startTime = clock.instant())
  def end(clock: Clock = Clock.systemUTC): Span =
    this.copy(endTime = clock.instant())

  def withLabels(extraLabels: (String, String)*) =
    this.copy(labels = this.labels ++ extraLabels)

}