Add ShouldTrace#shouldTrace

Add `ShouldTrace#shouldTrace`, to make boolean checks easier.
http4s · May 8, 2024 · a5dba72 · a5dba72
1 parent b4f9584
commit a5dba72
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/core/src/main/scala/org/http4s/otel4s/middleware/ShouldTrace.scala b/core/src/main/scala/org/http4s/otel4s/middleware/ShouldTrace.scala
@@ -17,13 +17,21 @@
 package org.http4s.otel4s.middleware
 
 /** Whether or not to trace something. */
-sealed trait ShouldTrace
+sealed trait ShouldTrace {
+
+  /** Whether or not to trace something. */
+  def shouldTrace: Boolean
+}
 
 object ShouldTrace {
 
   /** Trace the thing. */
-  case object Trace extends ShouldTrace
+  case object Trace extends ShouldTrace {
+    def shouldTrace: Boolean = true
+  }
 
   /** Do not trace the thing. */
-  case object DoNotTrace extends ShouldTrace
+  case object DoNotTrace extends ShouldTrace {
+    def shouldTrace: Boolean = false
+  }
 }