Add traces to all effectful methods for zio.http.codec package (#2317)

zio · Jul 28, 2023 · 500cbb7 · 500cbb7
1 parent dd07127
commit 500cbb7
Show file tree

Hide file tree

Showing 28 changed files with 50 additions and 20 deletions.
diff --git a/zio-http/src/main/scala/zio/http/codec/Alternator.scala b/zio-http/src/main/scala/zio/http/codec/Alternator.scala
@@ -17,6 +17,7 @@
 package zio.http.codec
 
 import zio.ZNothing
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
 /**
  * A alternator is a type class responsible for combining invariant type

diff --git a/zio-http/src/main/scala/zio/http/codec/Combiner.scala b/zio-http/src/main/scala/zio/http/codec/Combiner.scala
@@ -16,6 +16,8 @@
 
 package zio.http.codec
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
+
 /**
  * A combiner is a type class responsible for combining invariant type
  * parameters using a tuple. It is used to compose the parameters of the

diff --git a/zio-http/src/main/scala/zio/http/codec/ContentCodec.scala b/zio-http/src/main/scala/zio/http/codec/ContentCodec.scala
@@ -15,5 +15,5 @@
  */
 
 package zio.http.codec
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 object ContentCodec extends ContentCodecs
diff --git a/zio-http/src/main/scala/zio/http/codec/ContentCodecs.scala b/zio-http/src/main/scala/zio/http/codec/ContentCodecs.scala
@@ -16,6 +16,8 @@
 
 package zio.http.codec
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
+
 import zio.stream.ZStream
 
 import zio.schema.Schema

diff --git a/zio-http/src/main/scala/zio/http/codec/Doc.scala b/zio-http/src/main/scala/zio/http/codec/Doc.scala
@@ -16,6 +16,8 @@
 
 package zio.http.codec
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
+
 import zio.http.html
 
 /**

diff --git a/zio-http/src/main/scala/zio/http/codec/HeaderCodec.scala b/zio-http/src/main/scala/zio/http/codec/HeaderCodec.scala
@@ -15,5 +15,5 @@
  */
 
 package zio.http.codec
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 object HeaderCodec extends HeaderCodecs
diff --git a/zio-http/src/main/scala/zio/http/codec/HeaderCodecs.scala b/zio-http/src/main/scala/zio/http/codec/HeaderCodecs.scala
@@ -18,6 +18,8 @@ package zio.http.codec
 
 import scala.util.Try
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
+
 import zio.http.Header
 import zio.http.Header.HeaderType
 

diff --git a/zio-http/src/main/scala/zio/http/codec/HttpCodec.scala b/zio-http/src/main/scala/zio/http/codec/HttpCodec.scala
@@ -20,6 +20,7 @@ import scala.language.implicitConversions
 import scala.reflect.ClassTag
 
 import zio._
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
 import zio.stream.ZStream
 

diff --git a/zio-http/src/main/scala/zio/http/codec/HttpCodecError.scala b/zio-http/src/main/scala/zio/http/codec/HttpCodecError.scala
@@ -19,6 +19,7 @@ package zio.http.codec
 import scala.util.control.NoStackTrace
 
 import zio.Cause
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
 import zio.http.{Path, Status}
 

diff --git a/zio-http/src/main/scala/zio/http/codec/HttpCodecType.scala b/zio-http/src/main/scala/zio/http/codec/HttpCodecType.scala
@@ -15,7 +15,7 @@
  */
 
 package zio.http.codec
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 sealed trait HttpCodecType
 
 object HttpCodecType {

diff --git a/zio-http/src/main/scala/zio/http/codec/MethodCodec.scala b/zio-http/src/main/scala/zio/http/codec/MethodCodec.scala
@@ -15,5 +15,5 @@
  */
 
 package zio.http.codec
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 object MethodCodec extends MethodCodecs
diff --git a/zio-http/src/main/scala/zio/http/codec/MethodCodecs.scala b/zio-http/src/main/scala/zio/http/codec/MethodCodecs.scala
@@ -15,7 +15,7 @@
  */
 
 package zio.http.codec
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 private[codec] trait MethodCodecs {
   import HttpCodecType.Method
 

diff --git a/zio-http/src/main/scala/zio/http/codec/PathCodec.scala b/zio-http/src/main/scala/zio/http/codec/PathCodec.scala
@@ -20,6 +20,7 @@ import scala.annotation.tailrec
 import scala.collection.immutable.ListMap
 import scala.language.implicitConversions
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 import zio.{Chunk, NonEmptyChunk}
 
 import zio.http.Path

diff --git a/zio-http/src/main/scala/zio/http/codec/QueryCodec.scala b/zio-http/src/main/scala/zio/http/codec/QueryCodec.scala
@@ -15,5 +15,5 @@
  */
 
 package zio.http.codec
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 object QueryCodec extends QueryCodecs
diff --git a/zio-http/src/main/scala/zio/http/codec/QueryCodecs.scala b/zio-http/src/main/scala/zio/http/codec/QueryCodecs.scala
@@ -15,7 +15,7 @@
  */
 
 package zio.http.codec
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 private[codec] trait QueryCodecs {
   def query(name: String): QueryCodec[String] =
     HttpCodec.Query(name, TextCodec.string)

diff --git a/zio-http/src/main/scala/zio/http/codec/RichTextCodec.scala b/zio-http/src/main/scala/zio/http/codec/RichTextCodec.scala
@@ -21,6 +21,7 @@ import java.lang.Integer.parseInt
 import scala.annotation.tailrec
 import scala.collection.immutable.BitSet
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 import zio.{Chunk, NonEmptyChunk}
 
 /**

diff --git a/zio-http/src/main/scala/zio/http/codec/SegmentCodec.scala b/zio-http/src/main/scala/zio/http/codec/SegmentCodec.scala
@@ -18,6 +18,7 @@ package zio.http.codec
 import scala.language.implicitConversions
 
 import zio.Chunk
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
 import zio.http.Path
 

diff --git a/zio-http/src/main/scala/zio/http/codec/SimpleCodec.scala b/zio-http/src/main/scala/zio/http/codec/SimpleCodec.scala
@@ -15,6 +15,7 @@
  */
 
 package zio.http.codec
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
 /**
  * A simple codec is either equal to a given value, or unconstrained within a

diff --git a/zio-http/src/main/scala/zio/http/codec/StatusCodec.scala b/zio-http/src/main/scala/zio/http/codec/StatusCodec.scala
@@ -15,5 +15,5 @@
  */
 
 package zio.http.codec
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 object StatusCodec extends StatusCodecs
diff --git a/zio-http/src/main/scala/zio/http/codec/StatusCodecs.scala b/zio-http/src/main/scala/zio/http/codec/StatusCodecs.scala
@@ -16,8 +16,9 @@
 
 package zio.http.codec
 
-import zio.http.Status
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
+import zio.http.Status
 private[codec] trait StatusCodecs {
   def status(status: Status): StatusCodec[Unit] = HttpCodec.Status(SimpleCodec.Specified(status))
 

diff --git a/zio-http/src/main/scala/zio/http/codec/TextCodec.scala b/zio-http/src/main/scala/zio/http/codec/TextCodec.scala
@@ -18,6 +18,8 @@ package zio.http.codec
 
 import java.util.UUID
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
+
 /**
  * A [[zio.http.codec.TextCodec]] defines a codec for a text fragment. The text
  * fragment can be decoded into a value, or the value can be encoded into a text

diff --git a/zio-http/src/main/scala/zio/http/codec/internal/Atomized.scala b/zio-http/src/main/scala/zio/http/codec/internal/Atomized.scala
@@ -16,8 +16,9 @@
 
 package zio.http.codec.internal
 
-import zio.http.codec.HttpCodec
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
+import zio.http.codec.HttpCodec
 private[http] final case class Atomized[A](
   method: A,
   status: A,

diff --git a/zio-http/src/main/scala/zio/http/codec/internal/AtomizedCodecs.scala b/zio-http/src/main/scala/zio/http/codec/internal/AtomizedCodecs.scala
@@ -17,6 +17,7 @@
 package zio.http.codec.internal
 
 import zio.Chunk
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
 import zio.http.codec.HttpCodec._
 import zio.http.codec._

diff --git a/zio-http/src/main/scala/zio/http/codec/internal/BodyCodec.scala b/zio-http/src/main/scala/zio/http/codec/internal/BodyCodec.scala
@@ -17,6 +17,7 @@
 package zio.http.codec.internal
 
 import zio._
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
 import zio.stream.ZStream
 
@@ -41,12 +42,12 @@ private[internal] sealed trait BodyCodec[A] { self =>
   /**
    * Attempts to decode the `A` from a body using the given codec.
    */
-  def decodeFromBody(body: Body, codec: BinaryCodec[Element]): IO[Throwable, A]
+  def decodeFromBody(body: Body, codec: BinaryCodec[Element])(implicit trace: Trace): IO[Throwable, A]
 
   /**
    * Encodes the `A` to a body in the given codec.
    */
-  def encodeToBody(value: A, codec: BinaryCodec[Element]): Body
+  def encodeToBody(value: A, codec: BinaryCodec[Element])(implicit trace: Trace): Body
 
   /**
    * Erases the type for easier use in the internal implementation.
@@ -78,9 +79,9 @@ private[internal] object BodyCodec {
   case object Empty extends BodyCodec[Unit] {
     type Element = Unit
 
-    def decodeFromBody(body: Body, codec: BinaryCodec[Unit]): IO[Nothing, Unit] = ZIO.unit
+    def decodeFromBody(body: Body, codec: BinaryCodec[Unit])(implicit trace: Trace): IO[Nothing, Unit] = ZIO.unit
 
-    def encodeToBody(value: Unit, codec: BinaryCodec[Unit]): Body = Body.empty
+    def encodeToBody(value: Unit, codec: BinaryCodec[Unit])(implicit trace: Trace): Body = Body.empty
 
     def schema: Schema[Unit] = Schema[Unit]
 
@@ -91,26 +92,28 @@ private[internal] object BodyCodec {
 
   final case class Single[A](schema: Schema[A], mediaType: Option[MediaType], name: Option[String])
       extends BodyCodec[A] {
-    def decodeFromBody(body: Body, codec: BinaryCodec[A]): IO[Throwable, A] = {
+    def decodeFromBody(body: Body, codec: BinaryCodec[A])(implicit trace: Trace): IO[Throwable, A] = {
       if (schema == Schema[Unit]) ZIO.unit.asInstanceOf[IO[Throwable, A]]
       else
         body.asChunk.flatMap { chunk =>
           ZIO.fromEither(codec.decode(chunk))
         }
     }
 
-    def encodeToBody(value: A, codec: BinaryCodec[A]): Body =
+    def encodeToBody(value: A, codec: BinaryCodec[A])(implicit trace: Trace): Body =
       Body.fromChunk(codec.encode(value))
 
     type Element = A
   }
 
   final case class Multiple[E](schema: Schema[E], mediaType: Option[MediaType], name: Option[String])
       extends BodyCodec[ZStream[Any, Nothing, E]] {
-    def decodeFromBody(body: Body, codec: BinaryCodec[E]): IO[Throwable, ZStream[Any, Nothing, E]] =
+    def decodeFromBody(body: Body, codec: BinaryCodec[E])(implicit
+      trace: Trace,
+    ): IO[Throwable, ZStream[Any, Nothing, E]] =
       ZIO.succeed((body.asStream >>> codec.streamDecoder).orDie)
 
-    def encodeToBody(value: ZStream[Any, Nothing, E], codec: BinaryCodec[E]): Body =
+    def encodeToBody(value: ZStream[Any, Nothing, E], codec: BinaryCodec[E])(implicit trace: Trace): Body =
       Body.fromStream(value >>> codec.streamEncoder)
 
     type Element = E

diff --git a/zio-http/src/main/scala/zio/http/codec/internal/EncoderDecoder.scala b/zio-http/src/main/scala/zio/http/codec/internal/EncoderDecoder.scala
@@ -20,6 +20,7 @@ import java.time._
 import java.util.UUID
 
 import zio._
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 
 import zio.stream.ZStream
 
@@ -145,7 +146,7 @@ private[codec] object EncoderDecoder                   {
         erased.encodeToBody(_, jsonCodec)
       }
 
-    private val formFieldEncoders: Chunk[(String, Any) => FormField]      =
+    private val formFieldEncoders: Chunk[(String, Any) => FormField] =
       flattened.content.map { bodyCodec => (name: String, value: Any) =>
         {
           val erased = bodyCodec.erase
@@ -162,6 +163,9 @@ private[codec] object EncoderDecoder                   {
           }
         }
       }
+
+    implicit val trace: Trace = Trace.empty
+
     private val jsonDecoders: Chunk[Body => IO[Throwable, _]]             =
       flattened.content.map { bodyCodec =>
         val jsonCodec = JsonCodec.schemaBasedBinaryCodec(bodyCodec.schema)

diff --git a/zio-http/src/main/scala/zio/http/codec/internal/HaltException.scala b/zio-http/src/main/scala/zio/http/codec/internal/HaltException.scala
@@ -18,6 +18,7 @@ package zio.http.codec.internal
 
 import scala.util.control.NoStackTrace
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 case object HaltException
     extends RuntimeException("This exception is used only for control flow purposes")
     with NoStackTrace
diff --git a/zio-http/src/main/scala/zio/http/codec/internal/Mechanic.scala b/zio-http/src/main/scala/zio/http/codec/internal/Mechanic.scala
@@ -16,6 +16,8 @@
 
 package zio.http.codec.internal
 
+import zio.stacktracer.TracingImplicits.disableAutoTrace
+
 import zio.http.codec.HttpCodec._
 import zio.http.codec.{HttpCodec, HttpCodecError}
 

diff --git a/zio-http/src/main/scala/zio/http/codec/package.scala b/zio-http/src/main/scala/zio/http/codec/package.scala
@@ -15,7 +15,7 @@
  */
 
 package zio.http
-
+import zio.stacktracer.TracingImplicits.disableAutoTrace
 package object codec {
   type ContentCodec[A] = HttpCodec[HttpCodecType.Content, A]
   type HeaderCodec[A]  = HttpCodec[HttpCodecType.Header, A]