fix: don't assume attribute type in oteltracing (#1483)

honeycombio · Feb 12, 2025 · d9224c4 · d9224c4
1 parent a81349d
commit d9224c4
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/cmd/refinery/main.go b/cmd/refinery/main.go
@@ -244,7 +244,18 @@ func main() {
 		// add telemetry callback so husky can enrich spans with attributes
 		husky.AddTelemetryAttributeFunc = func(ctx context.Context, key string, value any) {
 			span := trace.SpanFromContext(ctx)
-			span.SetAttributes(attribute.String(key, value.(string)))
+			switch v := value.(type) {
+			case string:
+				span.SetAttributes(attribute.String(key, v))
+			case bool:
+				span.SetAttributes(attribute.Bool(key, v))
+			case int:
+				span.SetAttributes(attribute.Int(key, v))
+			case float64:
+				span.SetAttributes(attribute.Float64(key, v))
+			default:
+				span.SetAttributes(attribute.String(key, fmt.Sprintf("%v", v)))
+			}
 		}
 	}
 	defer shutdown()