config_serde.go

package dasmon

import (
	"encoding/json"
	"fmt"
	"time"

	"gopkg.in/yaml.v3"
)

// DurationSer wraps time.DurationSer to provide JSON unmarshaling from strings
type DurationSer time.Duration

// UnmarshalJSON parses a duration string from JSON
func (d *DurationSer) UnmarshalJSON(b []byte) error {
	var v interface{}
	if err := json.Unmarshal(b, &v); err != nil {
		return err
	}
	switch value := v.(type) {
	case string:
		parsed, err := time.ParseDuration(value)
		if err != nil {
			return err
		}
		*d = DurationSer(parsed)
		return nil
	case float64:
		*d = DurationSer(time.Duration(value))
		return nil
	default:
		return fmt.Errorf("invalid duration type")
	}
}

// MarshalJSON converts duration to JSON string
func (d DurationSer) MarshalJSON() ([]byte, error) {
	return json.Marshal(time.Duration(d).String())
}

// UnmarshalYAML parses a duration string from YAML
func (d *DurationSer) UnmarshalYAML(node *yaml.Node) error {
	var s string
	if err := node.Decode(&s); err != nil {
		return err
	}
	parsed, err := time.ParseDuration(s)
	if err != nil {
		return err
	}
	*d = DurationSer(parsed)
	return nil
}

// MarshalYAML converts duration to YAML string
func (d DurationSer) MarshalYAML() (interface{}, error) {
	return time.Duration(d).String(), nil
}