Experimental IOLocalContextStorage

build.sbt

@@ -26,7 +26,7 @@ ThisBuild / crossScalaVersions := Seq(Scala213, "3.2.2")
 ThisBuild / scalaVersion := Scala213 // the default Scala
 
 val CatsVersion = "2.9.0"
-val CatsEffectVersion = "3.5.0"
+val CatsEffectVersion = "3.5.0-31-2775064"
 val CatsMtlVersion = "1.3.1"
 val FS2Version = "3.6.1"
 val MUnitVersion = "1.0.0-M7"
@@ -150,11 +150,11 @@ lazy val testkit = crossProject(JVMPlatform)
 lazy val `java-common` = project
   .in(file("java/common"))
   .enablePlugins(BuildInfoPlugin)
-  .dependsOn(`core-common`.jvm, `testkit-common`.jvm)
+  .dependsOn(`core-common`.jvm, `testkit-common`.jvm % Test)
   .settings(
     name := "otel4s-java-common",
     libraryDependencies ++= Seq(
-      "org.typelevel" %%% "cats-effect-kernel" % CatsEffectVersion,
+      "org.typelevel" %%% "cats-effect" % CatsEffectVersion,
       "org.typelevel" %%% "cats-mtl" % CatsMtlVersion,
       "io.opentelemetry" % "opentelemetry-sdk" % OpenTelemetryVersion,
       "org.scalameta" %%% "munit" % MUnitVersion % Test
@@ -168,7 +168,7 @@ lazy val `java-common` = project
 
 lazy val `java-metrics` = project
   .in(file("java/metrics"))
-  .dependsOn(`java-common`, `core-metrics`.jvm, `testkit-metrics`.jvm)
+  .dependsOn(`java-common`, `core-metrics`.jvm, `testkit-metrics`.jvm % Test)
   .settings(munitDependencies)
   .settings(
     name := "otel4s-java-metrics",
@@ -222,7 +222,7 @@ lazy val benchmarks = project
   .enablePlugins(NoPublishPlugin)
   .enablePlugins(JmhPlugin)
   .in(file("benchmarks"))
-  .dependsOn(core.jvm, java)
+  .dependsOn(core.jvm, java, `testkit-metrics`.jvm)
   .settings(
     name := "otel4s-benchmarks"
   )
@@ -241,6 +241,7 @@ lazy val examples = project
     ),
     run / fork := true,
     javaOptions += "-Dotel.java.global-autoconfigure.enabled=true",
+    javaOptions += "-Dcats.effect.tracing.dumpLocals=true",
     envVars ++= Map(
       "OTEL_PROPAGATORS" -> "b3multi",
       "OTEL_SERVICE_NAME" -> "Trace Example"

examples/src/main/scala/ContextStorageExample.scala

@@ -0,0 +1,50 @@
+/*
+ * Copyright 2022 Typelevel
+ *
+ * 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.
+ */
+
+import cats.effect.IO
+import cats.effect.IOApp
+import cats.effect.Resource
+import io.opentelemetry.context.Context
+import io.opentelemetry.context.ContextKey
+import io.opentelemetry.context.ContextStorage
+import org.typelevel.otel4s.java.IOLocalContextStorageProvider
+
+import java.util.logging._
+
+object ContextStorageExample extends IOApp.Simple {
+
+  val key = ContextKey.named[String]("test")
+
+  val printKey =
+    IO(Option(Context.current().get(key))).flatMap(v => IO.println(v))
+
+  def run =
+    for {
+      _ <- IO {
+        val rootLog = Logger.getLogger("")
+        rootLog.setLevel(Level.FINE)
+        rootLog.getHandlers().head.setLevel(Level.FINE)
+      }
+      _ <- IOLocalContextStorageProvider.localContext.set(Context.root())
+      _ <- IO.println(ContextStorage.get.getClass())
+      _ <- Resource
+        .make(IO(Context.root().`with`(key, "hello").makeCurrent()))(scope =>
+          IO(scope.close())
+        )
+        .surround(printKey)
+      _ <- printKey
+    } yield ()
+}

java/common/src/main/resources/META-INF/services/io.opentelemetry.context.ContextStorageProvider

@@ -0,0 +1 @@
+org.typelevel.otel4s.java.IOLocalContextStorageProvider

java/common/src/main/scala/org/typelevel/otel4s/java/IOLocalContextStorage.scala

@@ -0,0 +1,41 @@
+/*
+ * Copyright 2022 Typelevel
+ *
+ * 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 org.typelevel.otel4s.java
+
+import cats.Eval
+import cats.effect.IOLocal
+import cats.effect.unsafe.IOLocals
+import io.opentelemetry.context.Context
+import io.opentelemetry.context.ContextStorage
+import io.opentelemetry.context.Scope
+
+class IOLocalContextStorage(
+    ioLocal: Eval[IOLocal[Context]]
+) extends ContextStorage {
+
+  override def attach(toAttach: Context): Scope = {
+    val previous = current()
+    IOLocals.set(ioLocal.value, toAttach)
+    new Scope {
+      def close() = IOLocals.set(ioLocal.value, previous)
+    }
+  }
+
+  override def current(): Context =
+    IOLocals.get(ioLocal.value)
+
+}

java/common/src/main/scala/org/typelevel/otel4s/java/IOLocalContextStorageProvider.scala

@@ -0,0 +1,46 @@
+/*
+ * Copyright 2022 Typelevel
+ *
+ * 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 org.typelevel.otel4s.java
+
+import cats.Eval
+import cats.effect.IOLocal
+import cats.effect.SyncIO
+import cats.syntax.all._
+import io.opentelemetry.context.Context
+import io.opentelemetry.context.ContextStorage
+import io.opentelemetry.context.ContextStorageProvider
+
+object IOLocalContextStorageProvider {
+  val localContext: IOLocal[Context] =
+    IOLocal[Context](Context.root())
+      .syncStep(100)
+      .flatMap(
+        _.leftMap(_ =>
+          new Error(
+            "Failed to initialize the local context of the IOLocalContextStorageProvider."
+          )
+        ).liftTo[SyncIO]
+      )
+      .unsafeRunSync()
+}
+
+class IOLocalContextStorageProvider extends ContextStorageProvider {
+  def get(): ContextStorage =
+    new IOLocalContextStorage(
+      Eval.later(IOLocalContextStorageProvider.localContext)
+    )
+}