/
yaml.go
50 lines (44 loc) · 1.23 KB
/
yaml.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package util
import (
"bufio"
"bytes"
k8sJson "k8s.io/apimachinery/pkg/runtime/serializer/json"
"sigs.k8s.io/yaml"
)
const YamlContentHeader string = "# DO NOT MODIFY! This file was generated by the Riser platform. Changes will be lost\n"
// ToYaml serializes to pretty YAML with JSON tags using the k8s encoding packages for consistency
func ToYaml(in interface{}) ([]byte, error) {
jsonIterator := k8sJson.CaseSensitiveJsonIterator()
jsonData, err := jsonIterator.Marshal(in)
if err != nil {
return nil, err
}
yamlData, err := yaml.JSONToYAML(jsonData)
if err != nil {
return nil, err
}
buffer := &bytes.Buffer{}
writer := bufio.NewWriter(buffer)
_, err = writer.WriteString(YamlContentHeader)
if err != nil {
return nil, err
}
_, err = writer.Write(yamlData)
if err != nil {
return nil, err
}
err = writer.Flush()
if err != nil {
return nil, err
}
return buffer.Bytes(), nil
}
// ParseYaml deserializes YAML into a struct with JSON tags. This uses k8s encoding packages for consistenct
func ParseYaml(in []byte, out interface{}) error {
jsonBytes, err := yaml.YAMLToJSONStrict(in)
if err != nil {
return err
}
jsonIterator := k8sJson.CaseSensitiveJsonIterator()
return jsonIterator.Unmarshal(jsonBytes, out)
}