mirror of
https://github.com/prometheus/node_exporter.git
synced 2024-11-15 18:14:17 -08:00
33f99c4fc1
Uses godep to vendor dependencies. Godeps is not necessary during build, golang's new vendor support is used instead.
165 lines
3.3 KiB
Go
165 lines
3.3 KiB
Go
package xmlrpc
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/xml"
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type encodeFunc func(reflect.Value) ([]byte, error)
|
|
|
|
func marshal(v interface{}) ([]byte, error) {
|
|
if v == nil {
|
|
return []byte{}, nil
|
|
}
|
|
|
|
val := reflect.ValueOf(v)
|
|
return encodeValue(val)
|
|
}
|
|
|
|
func encodeValue(val reflect.Value) ([]byte, error) {
|
|
var b []byte
|
|
var err error
|
|
|
|
if val.Kind() == reflect.Ptr || val.Kind() == reflect.Interface {
|
|
if val.IsNil() {
|
|
return []byte("<value/>"), nil
|
|
}
|
|
|
|
val = val.Elem()
|
|
}
|
|
|
|
switch val.Kind() {
|
|
case reflect.Struct:
|
|
switch val.Interface().(type) {
|
|
case time.Time:
|
|
t := val.Interface().(time.Time)
|
|
b = []byte(fmt.Sprintf("<dateTime.iso8601>%s</dateTime.iso8601>", t.Format(iso8601)))
|
|
default:
|
|
b, err = encodeStruct(val)
|
|
}
|
|
case reflect.Map:
|
|
b, err = encodeMap(val)
|
|
case reflect.Slice:
|
|
b, err = encodeSlice(val)
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
b = []byte(fmt.Sprintf("<int>%s</int>", strconv.FormatInt(val.Int(), 10)))
|
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
b = []byte(fmt.Sprintf("<i4>%s</i4>", strconv.FormatUint(val.Uint(), 10)))
|
|
case reflect.Float32, reflect.Float64:
|
|
b = []byte(fmt.Sprintf("<double>%s</double>",
|
|
strconv.FormatFloat(val.Float(), 'g', -1, val.Type().Bits())))
|
|
case reflect.Bool:
|
|
if val.Bool() {
|
|
b = []byte("<boolean>1</boolean>")
|
|
} else {
|
|
b = []byte("<boolean>0</boolean>")
|
|
}
|
|
case reflect.String:
|
|
var buf bytes.Buffer
|
|
|
|
xml.Escape(&buf, []byte(val.String()))
|
|
|
|
if _, ok := val.Interface().(Base64); ok {
|
|
b = []byte(fmt.Sprintf("<base64>%s</base64>", buf.String()))
|
|
} else {
|
|
b = []byte(fmt.Sprintf("<string>%s</string>", buf.String()))
|
|
}
|
|
default:
|
|
return nil, fmt.Errorf("xmlrpc encode error: unsupported type")
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return []byte(fmt.Sprintf("<value>%s</value>", string(b))), nil
|
|
}
|
|
|
|
func encodeStruct(val reflect.Value) ([]byte, error) {
|
|
var b bytes.Buffer
|
|
|
|
b.WriteString("<struct>")
|
|
|
|
t := val.Type()
|
|
for i := 0; i < t.NumField(); i++ {
|
|
b.WriteString("<member>")
|
|
f := t.Field(i)
|
|
|
|
name := f.Tag.Get("xmlrpc")
|
|
if name == "" {
|
|
name = f.Name
|
|
}
|
|
b.WriteString(fmt.Sprintf("<name>%s</name>", name))
|
|
|
|
p, err := encodeValue(val.FieldByName(f.Name))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
b.Write(p)
|
|
|
|
b.WriteString("</member>")
|
|
}
|
|
|
|
b.WriteString("</struct>")
|
|
|
|
return b.Bytes(), nil
|
|
}
|
|
|
|
func encodeMap(val reflect.Value) ([]byte, error) {
|
|
var t = val.Type()
|
|
|
|
if t.Key().Kind() != reflect.String {
|
|
return nil, fmt.Errorf("xmlrpc encode error: only maps with string keys are supported")
|
|
}
|
|
|
|
var b bytes.Buffer
|
|
|
|
b.WriteString("<struct>")
|
|
|
|
keys := val.MapKeys()
|
|
|
|
for i := 0; i < val.Len(); i++ {
|
|
key := keys[i]
|
|
kval := val.MapIndex(key)
|
|
|
|
b.WriteString("<member>")
|
|
b.WriteString(fmt.Sprintf("<name>%s</name>", key.String()))
|
|
|
|
p, err := encodeValue(kval)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
b.Write(p)
|
|
b.WriteString("</member>")
|
|
}
|
|
|
|
b.WriteString("</struct>")
|
|
|
|
return b.Bytes(), nil
|
|
}
|
|
|
|
func encodeSlice(val reflect.Value) ([]byte, error) {
|
|
var b bytes.Buffer
|
|
|
|
b.WriteString("<array><data>")
|
|
|
|
for i := 0; i < val.Len(); i++ {
|
|
p, err := encodeValue(val.Index(i))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
b.Write(p)
|
|
}
|
|
|
|
b.WriteString("</data></array>")
|
|
|
|
return b.Bytes(), nil
|
|
}
|