diff options
Diffstat (limited to 'kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala')
-rw-r--r-- | kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala b/kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala new file mode 100644 index 00000000..937200f5 --- /dev/null +++ b/kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala @@ -0,0 +1,106 @@ +package kamon.trace + +import java.nio.ByteBuffer +import java.util.concurrent.ThreadLocalRandom + +import kamon.util.HexCodec + +import scala.util.Try + +trait IdentityProvider { + def traceIdGenerator(): IdentityProvider.Generator + def spanIdGenerator(): IdentityProvider.Generator +} + +object IdentityProvider { + case class Identifier(string: String, bytes: Array[Byte]) { + + override def equals(obj: Any): Boolean = { + if(obj != null && obj.isInstanceOf[Identifier]) + obj.asInstanceOf[Identifier].string == string + else false + } + } + + val NoIdentifier = Identifier("", new Array[Byte](0)) + + trait Generator { + def generate(): Identifier + def from(string: String): Identifier + def from(bytes: Array[Byte]): Identifier + } + + + class Default extends IdentityProvider { + protected val longGenerator = new Generator { + override def generate(): Identifier = { + val data = ByteBuffer.wrap(new Array[Byte](8)) + val random = ThreadLocalRandom.current().nextLong() + data.putLong(random) + + Identifier(HexCodec.toLowerHex(random), data.array()) + } + + override def from(string: String): Identifier = Try { + val identifierLong = HexCodec.lowerHexToUnsignedLong(string) + val data = ByteBuffer.allocate(8) + data.putLong(identifierLong) + + Identifier(string, data.array()) + } getOrElse(IdentityProvider.NoIdentifier) + + override def from(bytes: Array[Byte]): Identifier = Try { + val buffer = ByteBuffer.wrap(bytes) + val identifierLong = buffer.getLong + + Identifier(HexCodec.toLowerHex(identifierLong), bytes) + } getOrElse(IdentityProvider.NoIdentifier) + } + + override def traceIdGenerator(): Generator = longGenerator + override def spanIdGenerator(): Generator = longGenerator + } + + object Default { + def apply(): Default = new Default() + } + + + class DoubleSizeTraceID extends Default { + private val doubleLongGenerator = new Generator { + override def generate(): Identifier = { + val data = ByteBuffer.wrap(new Array[Byte](16)) + val highLong = ThreadLocalRandom.current().nextLong() + val lowLong = ThreadLocalRandom.current().nextLong() + data.putLong(highLong) + data.putLong(lowLong) + + Identifier(HexCodec.toLowerHex(highLong) + HexCodec.toLowerHex(lowLong), data.array()) + } + + override def from(string: String): Identifier = Try { + val highPart = HexCodec.lowerHexToUnsignedLong(string.substring(0, 16)) + val lowPart = HexCodec.lowerHexToUnsignedLong(string.substring(16, 32)) + val data = ByteBuffer.allocate(16) + data.putLong(highPart) + data.putLong(lowPart) + + Identifier(string, data.array()) + } getOrElse(IdentityProvider.NoIdentifier) + + override def from(bytes: Array[Byte]): Identifier = Try { + val buffer = ByteBuffer.wrap(bytes) + val highLong = buffer.getLong + val lowLong = buffer.getLong + + Identifier(HexCodec.toLowerHex(highLong) + HexCodec.toLowerHex(lowLong), bytes) + } getOrElse(IdentityProvider.NoIdentifier) + } + + override def traceIdGenerator(): Generator = doubleLongGenerator + } + + object DoubleSizeTraceID { + def apply(): DoubleSizeTraceID = new DoubleSizeTraceID() + } +}
\ No newline at end of file |