Merge pull request #1 from ivantopo/wip/context-management-reloaded

remove context management from the Tracer

1 files changed, 45 insertions, 0 deletions

diff --git a/kamon-core/src/main/scala/kamon/context/Mixin.scala b/kamon-core/src/main/scala/kamon/context/Mixin.scala
new file mode 100644
index 00000000..64e03748
--- /dev/null
+++ b/kamon-core/src/main/scala/kamon/context/Mixin.scala
@@ -0,0 +1,45 @@
+/* =========================================================================================
+ * Copyright © 2013-2017 the kamon project <http://kamon.io/>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+ * except in compliance with the License. You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the
+ * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+ * either express or implied. See the License for the specific language governing permissions
+ * and limitations under the License.
+ * =========================================================================================
+ */
+
+package kamon.context
+
+import kamon.Kamon
+
+
+/**
+  * Utility trait that marks objects carrying a reference to a Span.
+  *
+  */
+trait HasContext {
+  def context: Context
+}
+
+object HasContext {
+  private case class Default(context: Context) extends HasContext
+
+  /**
+    * Construct a HasSpan instance that references the provided Span.
+    *
+    */
+  def from(context: Context): HasContext =
+    Default(context)
+
+  /**
+    * Construct a HasSpan instance that references the currently ActiveSpan in Kamon's tracer.
+    *
+    */
+  def fromCurrentContext(): HasContext =
+    Default(Kamon.currentContext())
+}