diff options
author | Ivan Topolnjak <ivantopo@gmail.com> | 2017-07-18 10:59:45 +0200 |
---|---|---|
committer | Ivan Topolnjak <ivantopo@gmail.com> | 2017-07-18 10:59:45 +0200 |
commit | ec01aa2661df31d58b0b894dcae81c982012e3a7 (patch) | |
tree | 35798a3b23911e97ba82dbaf3ad922136698842c /kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala | |
parent | 9b6878da2fbfb1328e972a885a6fdc077e08aaf9 (diff) | |
download | Kamon-ec01aa2661df31d58b0b894dcae81c982012e3a7.tar.gz Kamon-ec01aa2661df31d58b0b894dcae81c982012e3a7.tar.bz2 Kamon-ec01aa2661df31d58b0b894dcae81c982012e3a7.zip |
add a double length trace identifier implementation of IdentityProvider
Diffstat (limited to 'kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala')
-rw-r--r-- | kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala | 45 |
1 files changed, 42 insertions, 3 deletions
diff --git a/kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala b/kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala index 25e8f3c0..0e7d3e7d 100644 --- a/kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala +++ b/kamon-core/src/main/scala/kamon/trace/IdentityProvider.scala @@ -25,7 +25,7 @@ object IdentityProvider { class Default extends IdentityProvider { - private val generator = new Generator { + protected val longGenerator = new Generator { override def generate(): Identifier = { val data = ByteBuffer.wrap(new Array[Byte](8)) val random = ThreadLocalRandom.current().nextLong() @@ -50,11 +50,50 @@ object IdentityProvider { } getOrElse(IdentityProvider.NoIdentifier) } - override def traceIdentifierGenerator(): Generator = generator - override def spanIdentifierGenerator(): Generator = generator + override def traceIdentifierGenerator(): Generator = longGenerator + override def spanIdentifierGenerator(): 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 traceIdentifierGenerator(): Generator = doubleLongGenerator + } + + object DoubleSizeTraceID { + def apply(): DoubleSizeTraceID = new DoubleSizeTraceID() + } }
\ No newline at end of file |