fixup! Add PassThroughPropagator

review tweaks
typelevel · Nov 28, 2023 · 1d9d185 · 1d9d185
1 parent 6d938e8
commit 1d9d185
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/...ommon/src/main/scala/org/typelevel/otel4s/context/propagation/PassThroughPropagator.scala b/...ommon/src/main/scala/org/typelevel/otel4s/context/propagation/PassThroughPropagator.scala
@@ -17,6 +17,7 @@
 package org.typelevel.otel4s.context
 package propagation
 
+import cats.Foldable
 import cats.effect.SyncIO
 import org.typelevel.otel4s.context.syntax._
 
@@ -44,7 +45,8 @@ final class PassThroughPropagator[Ctx, K[X] <: Key[X]] private (
       .getOrElse(entriesKey, Nil)
       .foldLeft(carrier) { case (c, k -> v) => updater.updated(c, k, v) }
 
-  override def toString: String = s"PassThroughPropagator{fields=$fields}"
+  override def toString: String =
+    s"PassThroughPropagator{fields=${fields.mkString("[", ", ", "]")}}"
 }
 
 object PassThroughPropagator {
@@ -71,9 +73,10 @@ object PassThroughPropagator {
     forDistinctFields(fields.distinct)
 
   /** Creates a `PassThroughPropagator` that propagates the given fields. */
-  def apply[Ctx, K[X] <: Key[X]](fields: IterableOnce[String])(implicit
+  def apply[F[_], Ctx, K[X] <: Key[X]](fields: F[String])(implicit
+      F: Foldable[F],
       c: Contextual.Keyed[Ctx, K],
       kp: Key.Provider[SyncIO, K]
   ): TextMapPropagator[Ctx] =
-    forDistinctFields(fields.iterator.distinct.toSeq)
+    forDistinctFields(F.toList(fields).distinct)
 }