diff options
Diffstat (limited to 'src/main/scala/kamon/TraceContext.scala')
-rw-r--r-- | src/main/scala/kamon/TraceContext.scala | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/main/scala/kamon/TraceContext.scala b/src/main/scala/kamon/TraceContext.scala index 351446f3..0bfcd74b 100644 --- a/src/main/scala/kamon/TraceContext.scala +++ b/src/main/scala/kamon/TraceContext.scala @@ -53,12 +53,14 @@ object ThreadLocalTraceEntryStorage extends TraceEntryStorage { private val storage = new ThreadLocal[List[TraceEntry]] { override def initialValue(): List[TraceEntry] = Nil - def update(f: List[TraceEntry] => List[TraceEntry]) = set(f(get())) } + def update(f: List[TraceEntry] => List[TraceEntry]) = storage set f(storage.get) + def store(entry: TraceEntry): Boolean = { - storage.update(entry :: _) + update(entry :: _) true } } + |