-
Notifications
You must be signed in to change notification settings - Fork 154
/
dict.go
143 lines (123 loc) · 3.43 KB
/
dict.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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
package dict
import (
"github.com/influxdata/flux/codes"
"github.com/influxdata/flux/internal/errors"
"github.com/influxdata/flux/internal/function"
"github.com/influxdata/flux/interpreter"
"github.com/influxdata/flux/semantic"
"github.com/influxdata/flux/values"
)
const pkgpath = "dict"
// FromList will convert a list of values into a Dictionary.
func FromList(args interpreter.Arguments) (values.Value, error) {
pairs, err := args.GetRequiredArray("pairs", semantic.Object)
if err != nil {
return nil, err
}
// Retrieve the array element type to determine the
// types for the Dictionary.
elemType, err := pairs.Type().ElemType()
if err != nil {
return nil, err
}
// Retrieve the properties in a sorted way so that the
// properties are in a defined order. Since the symbol
// key will always be before value in order, then key
// will be the first element and value the second.
props, err := elemType.SortedProperties()
if err != nil {
return nil, err
}
// Exactly two properties.
if len(props) != 2 {
return nil, errors.New(codes.Internal, "expected exactly 2 properties for the array element type")
}
// First property should be "key".
if props[0].Name() != "key" {
return nil, errors.New(codes.Internal, "missing key property from array element")
}
keyType, err := props[0].TypeOf()
if err != nil {
return nil, err
}
// Second property should be "value".
if props[1].Name() != "value" {
return nil, errors.New(codes.Internal, "missing value property from array element")
}
valueType, err := props[1].TypeOf()
if err != nil {
return nil, err
}
// Construct the dictionary from each element in the array.
dictType := semantic.NewDictType(keyType, valueType)
builder := values.NewDictBuilder(dictType)
// Track any errors that happen when building the dictionary.
pairs.Array().Range(func(i int, v values.Value) {
if err != nil {
return
}
o := v.Object()
key, _ := o.Get("key")
value, _ := o.Get("value")
err = builder.Insert(key, value)
})
if err != nil {
return nil, err
}
return builder.Dict(), nil
}
// Get will retrieve a value from a Dictionary.
func Get(args interpreter.Arguments) (values.Value, error) {
from, err := args.GetRequiredDictionary("dict")
if err != nil {
return nil, err
}
key, err := args.GetRequired("key")
if err != nil {
return nil, err
}
def, err := args.GetRequired("default")
if err != nil {
return nil, err
}
return from.Get(key, def), nil
}
// Insert will insert a value into a Dictionary and
// return the new Dictionary. It will not modify
// the original Dictionary.
func Insert(args interpreter.Arguments) (values.Value, error) {
dict, err := args.GetRequiredDictionary("dict")
if err != nil {
return nil, err
}
key, err := args.GetRequired("key")
if err != nil {
return nil, err
}
value, err := args.GetRequired("value")
if err != nil {
return nil, err
}
return dict.Insert(key, value)
}
// Remove will remove a value from a Dictionary and
// return the new Dictionary. It will not modify
// the original Dictionary.
func Remove(args interpreter.Arguments) (values.Value, error) {
dict, err := args.GetRequiredDictionary("dict")
if err != nil {
return nil, err
}
key, err := args.GetRequired("key")
if err != nil {
return nil, err
}
return dict.Remove(key), nil
}
func init() {
b := function.ForPackage(pkgpath)
b.Register("fromList", FromList)
b.Register("get", Get)
b.Register("insert", Insert)
b.Register("remove", Remove)
}