-
Notifications
You must be signed in to change notification settings - Fork 11
/
config.go
105 lines (89 loc) · 2.31 KB
/
config.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
// SPDX-License-Identifier: MIT
// Package config 处理配置文件
package config
import (
"encoding/xml"
"errors"
"fmt"
"io"
)
// Config 用于表示配置文件中的数据
type Config struct {
parent *Config
Attrs map[string]string `yaml:"attrs" json:"attrs"` // 参数列表
Items map[string]*Config `yaml:"items" json:"items"` // 若是容器,则还有子项
}
// Sanitize 检测语法错误及基本的内容错误
func (cfg *Config) Sanitize() error {
if len(cfg.Attrs) > 0 {
return errors.New("根元素不能存在任何属性")
}
if len(cfg.Items) == 0 {
return errors.New("items 子项不能为空")
}
for name, item := range cfg.Items {
if len(item.Items) == 0 {
return fmt.Errorf("子项 %s 不能为空", name+".items")
}
}
return nil
}
// UnmarshalXML xml.Unmarshaler 接口实现
func (cfg *Config) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
node := cfg
for t, err := d.Token(); ; t, err = d.Token() {
if err != nil {
if err != io.EOF {
return err
}
return nil
}
switch token := t.(type) {
case xml.StartElement:
c := &Config{
parent: node,
Attrs: make(map[string]string, len(token.Attr)),
}
for _, v := range token.Attr {
c.Attrs[v.Name.Local] = v.Value
}
if node.Items == nil {
node.Items = make(map[string]*Config)
}
if _, found := node.Items[token.Name.Local]; found {
return fmt.Errorf("重复的元素名:%v", token.Name)
}
node.Items[token.Name.Local] = c
node = c
case xml.EndElement:
if node.parent != nil {
node = node.parent
}
} // end switch
} // end for
}
// MarshalXML xml.Unmarshaler 接口实现
func (cfg *Config) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
return cfg.marshalXML(e, xml.StartElement{Name: xml.Name{Local: "logs"}})
}
func (cfg *Config) marshalXML(e *xml.Encoder, start xml.StartElement) error {
if err := e.EncodeToken(start); err != nil {
return err
}
for k, v := range cfg.Items {
s := xml.StartElement{
Name: xml.Name{Local: k},
Attr: make([]xml.Attr, 0, len(v.Attrs)),
}
for name, val := range v.Attrs {
s.Attr = append(s.Attr, xml.Attr{
Name: xml.Name{Local: name},
Value: val,
})
}
if err := v.marshalXML(e, s); err != nil {
return err
}
}
return e.EncodeToken(xml.EndElement{Name: start.Name})
}