model_metric_threshold_serializer.go

package nextgen

import (
	"encoding/json"
	"fmt"
)

func (a *MetricThreshold) UnmarshalJSON(data []byte) error {

	type Alias MetricThreshold

	aux := &struct {
		*Alias
	}{
		Alias: (*Alias)(a),
	}

	err := json.Unmarshal(data, &aux)
	if err != nil {
		return err
	}

	switch a.Type_ {
	case MetricThresholdTypes.FailImmediately:
		err = json.Unmarshal(aux.Spec, &a.FailImmediately)
	case MetricThresholdTypes.IgnoreThreshold:
		err = json.Unmarshal(aux.Spec, &a.IgnoreThreshold)
	default:
		panic(fmt.Sprintf("unknown metric threshold type %s", a.Type_))
	}

	return err
}

func (a *MetricThreshold) MarshalJSON() ([]byte, error) {
	type Alias MetricThreshold

	var spec []byte
	var err error

	switch a.Type_ {
	case MetricThresholdTypes.FailImmediately:
		spec, err = json.Marshal(a.FailImmediately)
	case MetricThresholdTypes.IgnoreThreshold:
		spec, err = json.Marshal(a.IgnoreThreshold)
	default:
		panic(fmt.Sprintf("unknown metric threshold type %s", a.Type_))
	}

	if err != nil {
		return nil, err
	}

	a.Spec = json.RawMessage(spec)

	return json.Marshal((*Alias)(a))
}