~nskaggs/+junk/xenial-test

« back to all changes in this revision

Viewing changes to src/github.com/juju/govmomi/govc/host/esxcli/response.go

  • Committer: Nicholas Skaggs
  • Date: 2016-10-24 20:56:05 UTC
  • Revision ID: nicholas.skaggs@canonical.com-20161024205605-z8lta0uvuhtxwzwl
Initi with beta15

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
package esxcli
 
2
 
 
3
import (
 
4
        "io"
 
5
 
 
6
        "github.com/juju/govmomi/vim25/xml"
 
7
)
 
8
 
 
9
type Values map[string][]string
 
10
 
 
11
type Response struct {
 
12
        Info   *CommandInfoMethod
 
13
        Values []Values
 
14
}
 
15
 
 
16
func (v Values) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
 
17
        for {
 
18
                t, err := d.Token()
 
19
                if err != nil {
 
20
                        if err == io.EOF {
 
21
                                return nil
 
22
                        }
 
23
                        return err
 
24
                }
 
25
 
 
26
                if s, ok := t.(xml.StartElement); ok {
 
27
                        t, err = d.Token()
 
28
                        if err != nil {
 
29
                                return err
 
30
                        }
 
31
 
 
32
                        key := s.Name.Local
 
33
                        var val string
 
34
                        if c, ok := t.(xml.CharData); ok {
 
35
                                val = string(c)
 
36
                        }
 
37
                        v[key] = append(v[key], val)
 
38
                }
 
39
        }
 
40
}
 
41
 
 
42
func (r *Response) Type(start xml.StartElement) string {
 
43
        for _, a := range start.Attr {
 
44
                if a.Name.Local == "type" {
 
45
                        return a.Value
 
46
                }
 
47
        }
 
48
        return ""
 
49
}
 
50
 
 
51
func (r *Response) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
 
52
        stype := r.Type(start)
 
53
 
 
54
        if stype != "ArrayOfDataObject" {
 
55
                v := Values{}
 
56
                if err := d.DecodeElement(&v, &start); err != nil {
 
57
                        return err
 
58
                }
 
59
                r.Values = append(r.Values, v)
 
60
                return nil
 
61
        }
 
62
 
 
63
        for {
 
64
                t, err := d.Token()
 
65
                if err != nil {
 
66
                        if err == io.EOF {
 
67
                                return nil
 
68
                        }
 
69
                        return err
 
70
                }
 
71
 
 
72
                if s, ok := t.(xml.StartElement); ok {
 
73
                        if s.Name.Local == "DataObject" {
 
74
                                v := Values{}
 
75
                                if err := d.DecodeElement(&v, &s); err != nil {
 
76
                                        return err
 
77
                                }
 
78
                                r.Values = append(r.Values, v)
 
79
                        }
 
80
                }
 
81
        }
 
82
}