test changes

Author: nhulston

internal/trace/listener.go

@@ -13,6 +13,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"os"
+	"strconv"
 	"strings"
 
 	"github.com/DataDog/datadog-lambda-go/internal/extension"
@@ -55,36 +56,61 @@ var tracerInitialized = false
 
 // MakeListener initializes a new trace lambda Listener
 func MakeListener(config Config, extensionManager *extension.ExtensionManager) Listener {
-	fmt.Printf("MakeListener()")
+	fmt.Printf("MakeListener()\n")
 	if config.TracerOptions == nil {
 		config.TracerOptions = []tracer.StartOption{}
 	}
 
 	// Read DD_TRACE_SAMPLING_RULES first
-	fmt.Printf("Trying to get sampling rules env var")
+	fmt.Printf("Trying to get sampling rules env var\n")
 	var samplingRules []tracer.SamplingRule
 	if ruleStr := os.Getenv("DD_TRACE_SAMPLING_RULES"); ruleStr != "" {
-		fmt.Printf("Env: %s", ruleStr)
+		fmt.Printf("Env: %s\n", ruleStr)
 		var rules []map[string]interface{}
 		if err := json.Unmarshal([]byte(ruleStr), &rules); err == nil {
+			fmt.Printf("Successfully unmarshaled rules: %+v\n", rules)
 			for _, rule := range rules {
-				fmt.Printf("Rule: %s", rule)
-				if rate, ok := rule["sample_rate"].(float64); ok {
-					fmt.Printf("Appending sampling rule")
+				fmt.Printf("Processing rule: %+v\n", rule)
+				if rateVal, ok := rule["sample_rate"]; ok {
+					fmt.Printf("Found sample_rate of type %T with value %v\n", rateVal, rateVal)
+
+					// Try different type conversions
+					rate, ok := rateVal.(float64)
+					if !ok {
+						// Try converting from json.Number
+						if numStr, ok := rateVal.(json.Number); ok {
+							rate, _ = numStr.Float64()
+							fmt.Printf("Converted json.Number to float64: %v\n", rate)
+						} else if str, ok := rateVal.(string); ok {
+							rate, _ = strconv.ParseFloat(str, 64)
+							fmt.Printf("Converted string to float64: %v\n", rate)
+						}
+					}
+					fmt.Printf("Final rate value: %v\n", rate)
 					samplingRules = append(samplingRules, tracer.RateRule(rate))
 				}
 			}
+		} else {
+			fmt.Printf("Error unmarshaling rules: %v\n", err)
 		}
+	} else {
+		fmt.Printf("No DD_TRACE_SAMPLING_RULES env var found\n")
 	}
 
 	// If we have sampling rules from environment, add them first
 	if len(samplingRules) > 0 {
-		fmt.Printf("Sample rule found, adding.")
+		fmt.Printf("Adding %d sampling rules to config\n", len(samplingRules))
+		for i, rule := range samplingRules {
+			fmt.Printf("Rule %d: %+v\n", i, rule)
+		}
 		config.TracerOptions = append([]tracer.StartOption{
 			tracer.WithSamplingRules(samplingRules),
 		}, config.TracerOptions...)
 	}
 
+	// Print final config
+	fmt.Printf("Final config has %d tracer options\n", len(config.TracerOptions))
+
 	return Listener{
 		ddTraceEnabled:           config.DDTraceEnabled,
 		mergeXrayTraces:          config.MergeXrayTraces,