| // Package yaml implements YAML support for the Go language. |
| // |
| // Source code and other details for the project are available at GitHub: |
| // |
| // https://github.com/go-yaml/yaml |
| // |
| package yaml |
| |
| import ( |
| "errors" |
| "fmt" |
| "reflect" |
| "strings" |
| "sync" |
| ) |
| |
| // MapSlice encodes and decodes as a YAML map. |
| // The order of keys is preserved when encoding and decoding. |
| type MapSlice []MapItem |
| |
| // MapItem is an item in a MapSlice. |
| type MapItem struct { |
| Key, Value interface{} |
| } |
| |
| // The Unmarshaler interface may be implemented by types to customize their |
| // behavior when being unmarshaled from a YAML document. The UnmarshalYAML |
| // method receives a function that may be called to unmarshal the original |
| // YAML value into a field or variable. It is safe to call the unmarshal |
| // function parameter more than once if necessary. |
| type Unmarshaler interface { |
| UnmarshalYAML(unmarshal func(interface{}) error) error |
| } |
| |
| // The Marshaler interface may be implemented by types to customize their |
| // behavior when being marshaled into a YAML document. The returned value |
| // is marshaled in place of the original value implementing Marshaler. |
| // |
| // If an error is returned by MarshalYAML, the marshaling procedure stops |
| // and returns with the provided error. |
| type Marshaler interface { |
| MarshalYAML() (interface{}, error) |
| } |
| |
| // Unmarshal decodes the first document found within the in byte slice |
| // and assigns decoded values into the out value. |
| // |
| // Maps and pointers (to a struct, string, int, etc) are accepted as out |
| // values. If an internal pointer within a struct is not initialized, |
| // the yaml package will initialize it if necessary for unmarshalling |
| // the provided data. The out parameter must not be nil. |
| // |
| // The type of the decoded values should be compatible with the respective |
| // values in out. If one or more values cannot be decoded due to a type |
| // mismatches, decoding continues partially until the end of the YAML |
| // content, and a *yaml.TypeError is returned with details for all |
| // missed values. |
| // |
| // Struct fields are only unmarshalled if they are exported (have an |
| // upper case first letter), and are unmarshalled using the field name |
| // lowercased as the default key. Custom keys may be defined via the |
| // "yaml" name in the field tag: the content preceding the first comma |
| // is used as the key, and the following comma-separated options are |
| // used to tweak the marshalling process (see Marshal). |
| // Conflicting names result in a runtime error. |
| // |
| // For example: |
| // |
| // type T struct { |
| // F int `yaml:"a,omitempty"` |
| // B int |
| // } |
| // var t T |
| // yaml.Unmarshal([]byte("a: 1\nb: 2"), &t) |
| // |
| // See the documentation of Marshal for the format of tags and a list of |
| // supported tag options. |
| // |
| func Unmarshal(in []byte, out interface{}) (err error) { |
| defer handleErr(&err) |
| d := newDecoder() |
| p := newParser(in) |
| defer p.destroy() |
| node := p.parse() |
| if node != nil { |
| v := reflect.ValueOf(out) |
| if v.Kind() == reflect.Ptr && !v.IsNil() { |
| v = v.Elem() |
| } |
| d.unmarshal(node, v) |
| } |
| if len(d.terrors) > 0 { |
| return &TypeError{d.terrors} |
| } |
| return nil |
| } |
| |
| // Marshal serializes the value provided into a YAML document. The structure |
| // of the generated document will reflect the structure of the value itself. |
| // Maps and pointers (to struct, string, int, etc) are accepted as the in value. |
| // |
| // Struct fields are only unmarshalled if they are exported (have an upper case |
| // first letter), and are unmarshalled using the field name lowercased as the |
| // default key. Custom keys may be defined via the "yaml" name in the field |
| // tag: the content preceding the first comma is used as the key, and the |
| // following comma-separated options are used to tweak the marshalling process. |
| // Conflicting names result in a runtime error. |
| // |
| // The field tag format accepted is: |
| // |
| // `(...) yaml:"[<key>][,<flag1>[,<flag2>]]" (...)` |
| // |
| // The following flags are currently supported: |
| // |
| // omitempty Only include the field if it's not set to the zero |
| // value for the type or to empty slices or maps. |
| // Does not apply to zero valued structs. |
| // |
| // flow Marshal using a flow style (useful for structs, |
| // sequences and maps). |
| // |
| // inline Inline the field, which must be a struct or a map, |
| // causing all of its fields or keys to be processed as if |
| // they were part of the outer struct. For maps, keys must |
| // not conflict with the yaml keys of other struct fields. |
| // |
| // In addition, if the key is "-", the field is ignored. |
| // |
| // For example: |
| // |
| // type T struct { |
| // F int "a,omitempty" |
| // B int |
| // } |
| // yaml.Marshal(&T{B: 2}) // Returns "b: 2\n" |
| // yaml.Marshal(&T{F: 1}} // Returns "a: 1\nb: 0\n" |
| // |
| func Marshal(in interface{}) (out []byte, err error) { |
| defer handleErr(&err) |
| e := newEncoder() |
| defer e.destroy() |
| e.marshal("", reflect.ValueOf(in)) |
| e.finish() |
| out = e.out |
| return |
| } |
| |
| func handleErr(err *error) { |
| if v := recover(); v != nil { |
| if e, ok := v.(yamlError); ok { |
| *err = e.err |
| } else { |
| panic(v) |
| } |
| } |
| } |
| |
| type yamlError struct { |
| err error |
| } |
| |
| func fail(err error) { |
| panic(yamlError{err}) |
| } |
| |
| func failf(format string, args ...interface{}) { |
| panic(yamlError{fmt.Errorf("yaml: "+format, args...)}) |
| } |
| |
| // A TypeError is returned by Unmarshal when one or more fields in |
| // the YAML document cannot be properly decoded into the requested |
| // types. When this error is returned, the value is still |
| // unmarshaled partially. |
| type TypeError struct { |
| Errors []string |
| } |
| |
| func (e *TypeError) Error() string { |
| return fmt.Sprintf("yaml: unmarshal errors:\n %s", strings.Join(e.Errors, "\n ")) |
| } |
| |
| // -------------------------------------------------------------------------- |
| // Maintain a mapping of keys to structure field indexes |
| |
| // The code in this section was copied from mgo/bson. |
| |
| // structInfo holds details for the serialization of fields of |
| // a given struct. |
| type structInfo struct { |
| FieldsMap map[string]fieldInfo |
| FieldsList []fieldInfo |
| |
| // InlineMap is the number of the field in the struct that |
| // contains an ,inline map, or -1 if there's none. |
| InlineMap int |
| } |
| |
| type fieldInfo struct { |
| Key string |
| Num int |
| OmitEmpty bool |
| Flow bool |
| |
| // Inline holds the field index if the field is part of an inlined struct. |
| Inline []int |
| } |
| |
| var structMap = make(map[reflect.Type]*structInfo) |
| var fieldMapMutex sync.RWMutex |
| |
| func getStructInfo(st reflect.Type) (*structInfo, error) { |
| fieldMapMutex.RLock() |
| sinfo, found := structMap[st] |
| fieldMapMutex.RUnlock() |
| if found { |
| return sinfo, nil |
| } |
| |
| n := st.NumField() |
| fieldsMap := make(map[string]fieldInfo) |
| fieldsList := make([]fieldInfo, 0, n) |
| inlineMap := -1 |
| for i := 0; i != n; i++ { |
| field := st.Field(i) |
| if field.PkgPath != "" && !field.Anonymous { |
| continue // Private field |
| } |
| |
| info := fieldInfo{Num: i} |
| |
| tag := field.Tag.Get("yaml") |
| if tag == "" && strings.Index(string(field.Tag), ":") < 0 { |
| tag = string(field.Tag) |
| } |
| if tag == "-" { |
| continue |
| } |
| |
| inline := false |
| fields := strings.Split(tag, ",") |
| if len(fields) > 1 { |
| for _, flag := range fields[1:] { |
| switch flag { |
| case "omitempty": |
| info.OmitEmpty = true |
| case "flow": |
| info.Flow = true |
| case "inline": |
| inline = true |
| default: |
| return nil, errors.New(fmt.Sprintf("Unsupported flag %q in tag %q of type %s", flag, tag, st)) |
| } |
| } |
| tag = fields[0] |
| } |
| |
| if inline { |
| switch field.Type.Kind() { |
| case reflect.Map: |
| if inlineMap >= 0 { |
| return nil, errors.New("Multiple ,inline maps in struct " + st.String()) |
| } |
| if field.Type.Key() != reflect.TypeOf("") { |
| return nil, errors.New("Option ,inline needs a map with string keys in struct " + st.String()) |
| } |
| inlineMap = info.Num |
| case reflect.Struct: |
| sinfo, err := getStructInfo(field.Type) |
| if err != nil { |
| return nil, err |
| } |
| for _, finfo := range sinfo.FieldsList { |
| if _, found := fieldsMap[finfo.Key]; found { |
| msg := "Duplicated key '" + finfo.Key + "' in struct " + st.String() |
| return nil, errors.New(msg) |
| } |
| if finfo.Inline == nil { |
| finfo.Inline = []int{i, finfo.Num} |
| } else { |
| finfo.Inline = append([]int{i}, finfo.Inline...) |
| } |
| fieldsMap[finfo.Key] = finfo |
| fieldsList = append(fieldsList, finfo) |
| } |
| default: |
| //return nil, errors.New("Option ,inline needs a struct value or map field") |
| return nil, errors.New("Option ,inline needs a struct value field") |
| } |
| continue |
| } |
| |
| if tag != "" { |
| info.Key = tag |
| } else { |
| info.Key = strings.ToLower(field.Name) |
| } |
| |
| if _, found = fieldsMap[info.Key]; found { |
| msg := "Duplicated key '" + info.Key + "' in struct " + st.String() |
| return nil, errors.New(msg) |
| } |
| |
| fieldsList = append(fieldsList, info) |
| fieldsMap[info.Key] = info |
| } |
| |
| sinfo = &structInfo{fieldsMap, fieldsList, inlineMap} |
| |
| fieldMapMutex.Lock() |
| structMap[st] = sinfo |
| fieldMapMutex.Unlock() |
| return sinfo, nil |
| } |
| |
| func isZero(v reflect.Value) bool { |
| switch v.Kind() { |
| case reflect.String: |
| return len(v.String()) == 0 |
| case reflect.Interface, reflect.Ptr: |
| return v.IsNil() |
| case reflect.Slice: |
| return v.Len() == 0 |
| case reflect.Map: |
| return v.Len() == 0 |
| case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: |
| return v.Int() == 0 |
| case reflect.Float32, reflect.Float64: |
| return v.Float() == 0 |
| case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr: |
| return v.Uint() == 0 |
| case reflect.Bool: |
| return !v.Bool() |
| case reflect.Struct: |
| vt := v.Type() |
| for i := v.NumField() - 1; i >= 0; i-- { |
| if vt.Field(i).PkgPath != "" { |
| continue // Private field |
| } |
| if !isZero(v.Field(i)) { |
| return false |
| } |
| } |
| return true |
| } |
| return false |
| } |