mirror of
https://github.com/astaxie/beego.git
synced 2024-11-18 06:00:55 +00:00
106 lines
2.1 KiB
Go
106 lines
2.1 KiB
Go
|
// io.Reader --> map[string]interface{} or JSON string
|
||
|
// nothing magic - just implements generic Go case
|
||
|
|
||
|
package x2j
|
||
|
|
||
|
import (
|
||
|
"encoding/json"
|
||
|
"encoding/xml"
|
||
|
"io"
|
||
|
)
|
||
|
|
||
|
// ToTree() - parse a XML io.Reader to a tree of Nodes
|
||
|
func ToTree(rdr io.Reader) (*Node, error) {
|
||
|
p := xml.NewDecoder(rdr)
|
||
|
p.CharsetReader = X2jCharsetReader
|
||
|
n, perr := xmlToTree("", nil, p)
|
||
|
if perr != nil {
|
||
|
return nil, perr
|
||
|
}
|
||
|
|
||
|
return n, nil
|
||
|
}
|
||
|
|
||
|
// ToMap() - parse a XML io.Reader to a map[string]interface{}
|
||
|
func ToMap(rdr io.Reader, recast ...bool) (map[string]interface{}, error) {
|
||
|
var r bool
|
||
|
if len(recast) == 1 {
|
||
|
r = recast[0]
|
||
|
}
|
||
|
n, err := ToTree(rdr)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
m := make(map[string]interface{})
|
||
|
m[n.key] = n.treeToMap(r)
|
||
|
|
||
|
return m, nil
|
||
|
}
|
||
|
|
||
|
// ToJson() - parse a XML io.Reader to a JSON string
|
||
|
func ToJson(rdr io.Reader, recast ...bool) (string, error) {
|
||
|
var r bool
|
||
|
if len(recast) == 1 {
|
||
|
r = recast[0]
|
||
|
}
|
||
|
m, merr := ToMap(rdr, r)
|
||
|
if m == nil || merr != nil {
|
||
|
return "", merr
|
||
|
}
|
||
|
|
||
|
b, berr := json.Marshal(m)
|
||
|
if berr != nil {
|
||
|
return "", berr
|
||
|
}
|
||
|
|
||
|
return string(b), nil
|
||
|
}
|
||
|
|
||
|
// ToJsonIndent - the pretty form of ReaderToJson
|
||
|
func ToJsonIndent(rdr io.Reader, recast ...bool) (string, error) {
|
||
|
var r bool
|
||
|
if len(recast) == 1 {
|
||
|
r = recast[0]
|
||
|
}
|
||
|
m, merr := ToMap(rdr, r)
|
||
|
if m == nil || merr != nil {
|
||
|
return "", merr
|
||
|
}
|
||
|
|
||
|
b, berr := json.MarshalIndent(m, "", " ")
|
||
|
if berr != nil {
|
||
|
return "", berr
|
||
|
}
|
||
|
|
||
|
// NOTE: don't have to worry about safe JSON marshaling with json.Marshal, since '<' and '>" are reservedin XML.
|
||
|
return string(b), nil
|
||
|
}
|
||
|
|
||
|
|
||
|
// ReaderValuesFromTagPath - io.Reader version of ValuesFromTagPath()
|
||
|
func ReaderValuesFromTagPath(rdr io.Reader, path string, getAttrs ...bool) ([]interface{}, error) {
|
||
|
var a bool
|
||
|
if len(getAttrs) == 1 {
|
||
|
a = getAttrs[0]
|
||
|
}
|
||
|
m, err := ToMap(rdr)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return ValuesFromKeyPath(m, path, a), nil
|
||
|
}
|
||
|
|
||
|
// ReaderValuesForTag - io.Reader version of ValuesForTag()
|
||
|
func ReaderValuesForTag(rdr io.Reader, tag string) ([]interface{}, error) {
|
||
|
m, err := ToMap(rdr)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return ValuesForKey(m, tag), nil
|
||
|
}
|
||
|
|
||
|
|