chore: Extract tristate as an enum

bric3 · bric3 · commit 3d282555c692 · 2025-09-09T09:43:59.000+02:00
Signed-off-by: Brice Dutheil &lt;brice.dutheil@gmail.com&gt;
diff --git a/dd-trace-core/src/main/java/datadog/trace/common/metrics/SerializingMetricWriter.java b/dd-trace-core/src/main/java/datadog/trace/common/metrics/SerializingMetricWriter.java
@@ -36,11 +36,9 @@ public final class SerializingMetricWriter implements MetricWriter {
   private static final byte[] SPAN_KIND = "SpanKind".getBytes(ISO_8859_1);
   private static final byte[] PEER_TAGS = "PeerTags".getBytes(ISO_8859_1);
 
-  @SuppressWarnings("unused") // Kept for representing all possible states
-  public static final int TRISTATE_UNKNOWN = 0;
-
-  public static final int TRISTATE_TRUE = 1;
-  public static final int TRISTATE_FALSE = 2;
+  // Constant declared here for compile-time folding
+  public static final int TRISTATE_TRUE = TriState.TRUE.serialValue;
+  public static final int TRISTATE_FALSE = TriState.FALSE.serialValue;
 
   private final WellKnownTags wellKnownTags;
   private final WritableFormatter writer;
diff --git a/dd-trace-core/src/main/java/datadog/trace/common/metrics/TriState.java b/dd-trace-core/src/main/java/datadog/trace/common/metrics/TriState.java
@@ -0,0 +1,16 @@
+package datadog.trace.common.metrics;
+
+/**
+ * TriState is used to represent a three-valued logic: true, false, and unknown. The "integer"
+ * values are used for metrics serialization.
+ */
+public enum TriState {
+  UNKNOWN(0),
+  TRUE(1),
+  FALSE(2);
+  public final int serialValue;
+
+  TriState(int serialValue) {
+    this.serialValue = serialValue;
+  }
+}
diff --git a/dd-trace-core/src/test/groovy/datadog/trace/common/metrics/SerializingMetricWriterTest.groovy b/dd-trace-core/src/test/groovy/datadog/trace/common/metrics/SerializingMetricWriterTest.groovy
@@ -17,10 +17,6 @@ import static java.util.concurrent.TimeUnit.MILLISECONDS
 import static java.util.concurrent.TimeUnit.SECONDS
 
 class SerializingMetricWriterTest extends DDSpecification {
-
-  public static final int TRISTATE_TRUE = 1
-  public static final int TRISTATE_FALSE = 2
-
   def "should produce correct message #iterationIndex with process tags enabled #withProcessTags" () {
     setup:
     if (withProcessTags) {
@@ -180,7 +176,7 @@ class SerializingMetricWriterTest extends DDSpecification {
         assert unpacker.unpackBoolean() == key.isSynthetics()
         ++elementCount
         assert unpacker.unpackString() == "IsTraceRoot"
-        assert unpacker.unpackInt() == (key.isTraceRoot() ? TRISTATE_TRUE : TRISTATE_FALSE)
+        assert unpacker.unpackInt() == (key.isTraceRoot() ? TriState.TRUE.serialValue : TriState.FALSE.serialValue)
         ++elementCount
         assert unpacker.unpackString() == "SpanKind"
         assert unpacker.unpackString() == key.getSpanKind() as String
