[CWS] fix unmarshalling of regular int to time.Duration (DataDog#31769

)
kkpan11 · Dec 5, 2024 · fef1eb7 · fef1eb7
1 parent 6b84c87
commit fef1eb7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/security/secl/rules/model.go b/pkg/security/secl/rules/model.go
@@ -7,7 +7,7 @@
 package rules
 
 import (
-	"errors"
+	"fmt"
 	"time"
 
 	"gopkg.in/yaml.v3"
@@ -219,7 +219,7 @@ func (d *HumanReadableDuration) UnmarshalYAML(n *yaml.Node) error {
 		return err
 	}
 	switch value := v.(type) {
-	case float64:
+	case int:
 		d.Duration = time.Duration(value)
 		return nil
 	case string:
@@ -230,7 +230,7 @@ func (d *HumanReadableDuration) UnmarshalYAML(n *yaml.Node) error {
 		}
 		return nil
 	default:
-		return errors.New("invalid duration")
+		return fmt.Errorf("invalid duration: (yaml type: %T)", v)
 	}
 }