31 lines
782 B
Go
31 lines
782 B
Go
|
package codec
|
||
|
|
||
|
import (
|
||
|
"encoding/json"
|
||
|
|
||
|
"github.com/gogo/protobuf/proto"
|
||
|
"gopkg.in/yaml.v2"
|
||
|
)
|
||
|
|
||
|
// MarshalYAML marshals the provided toPrint content with the provided JSON marshaler
|
||
|
// by encoding JSON, decoding JSON, and then encoding YAML.
|
||
|
func MarshalYAML(jsonMarshaler JSONMarshaler, toPrint proto.Message) ([]byte, error) {
|
||
|
// only the JSONMarshaler has full context as to how the JSON
|
||
|
// mashalling should look (which may be different for amino & proto codecs)
|
||
|
// so we need to use it to marshal toPrint first
|
||
|
bz, err := jsonMarshaler.MarshalJSON(toPrint)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
// generate YAML by decoding and re-encoding JSON as YAML
|
||
|
var j interface{}
|
||
|
|
||
|
err = json.Unmarshal(bz, &j)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return yaml.Marshal(j)
|
||
|
}
|