diff options
author | Diego <diegolparra@gmail.com> | 2015-05-03 11:30:31 -0300 |
---|---|---|
committer | Diego <diegolparra@gmail.com> | 2015-05-03 11:30:31 -0300 |
commit | 3019e38462a2aad8b0d76802263636f5096a01ca (patch) | |
tree | 69d462f3a23d15f0f3186d69981e9cef4da5f497 /kamon-spray/src | |
parent | 2915de00984e31eda67f10348aa31ed8fdf8d71a (diff) | |
download | Kamon-3019e38462a2aad8b0d76802263636f5096a01ca.tar.gz Kamon-3019e38462a2aad8b0d76802263636f5096a01ca.tar.bz2 Kamon-3019e38462a2aad8b0d76802263636f5096a01ca.zip |
! spray: use SameThreadExecutionContext in ClientRequestInstrumentation and closes #143
Diffstat (limited to 'kamon-spray/src')
-rw-r--r-- | kamon-spray/src/main/scala/kamon/spray/instrumentation/ClientRequestInstrumentation.scala | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/kamon-spray/src/main/scala/kamon/spray/instrumentation/ClientRequestInstrumentation.scala b/kamon-spray/src/main/scala/kamon/spray/instrumentation/ClientRequestInstrumentation.scala index 22d6e354..4c408567 100644 --- a/kamon-spray/src/main/scala/kamon/spray/instrumentation/ClientRequestInstrumentation.scala +++ b/kamon-spray/src/main/scala/kamon/spray/instrumentation/ClientRequestInstrumentation.scala @@ -17,6 +17,7 @@ package spray.can.client import kamon.Kamon +import kamon.util.SameThreadExecutionContext import org.aspectj.lang.annotation._ import org.aspectj.lang.ProceedingJoinPoint import spray.http._ @@ -123,13 +124,10 @@ class ClientRequestInstrumentation { request.asInstanceOf[SegmentAware].segment = segment val responseFuture = originalSendReceive.apply(request) - responseFuture.onComplete { result ⇒ - segment.finish() - }(ec) - + responseFuture.map(result ⇒ segment.finish())(SameThreadExecutionContext) responseFuture - } getOrElse (originalSendReceive.apply(request)) + } getOrElse originalSendReceive.apply(request) } } |