forked from influxdata/influxdb
-
Notifications
You must be signed in to change notification settings - Fork 0
/
value.go
70 lines (60 loc) · 1.69 KB
/
value.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
package parser
// #include "query_types.h"
// #include <stdlib.h>
import "C"
import (
"bytes"
"fmt"
"regexp"
)
type ValueType int
const (
ValueRegex ValueType = C.VALUE_REGEX
ValueInt = C.VALUE_INT
ValueBool = C.VALUE_BOOLEAN
ValueFloat = C.VALUE_FLOAT
ValueString = C.VALUE_STRING
ValueIntoName = C.VALUE_INTO_NAME
ValueTableName = C.VALUE_TABLE_NAME
ValueSimpleName = C.VALUE_SIMPLE_NAME
ValueDuration = C.VALUE_DURATION
ValueWildcard = C.VALUE_WILDCARD
ValueFunctionCall = C.VALUE_FUNCTION_CALL
ValueExpression = C.VALUE_EXPRESSION
)
type Value struct {
Name string
Alias string
Type ValueType
Elems []*Value
compiledRegex *regexp.Regexp
IsInsensitive bool
}
func (self *Value) IsFunctionCall() bool {
return self.Type == ValueFunctionCall
}
func (self *Value) GetCompiledRegex() (*regexp.Regexp, bool) {
return self.compiledRegex, self.Type == ValueRegex
}
func (self *Value) GetString() string {
buffer := bytes.NewBufferString("")
switch self.Type {
case ValueExpression:
fmt.Fprintf(buffer, "%s %s %s", self.Elems[0].GetString(), self.Name, self.Elems[1].GetString())
case ValueFunctionCall:
fmt.Fprintf(buffer, "%s(%s)", self.Name, Values(self.Elems).GetString())
case ValueString:
fmt.Fprintf(buffer, "'%s'", self.Name)
case ValueRegex:
fmt.Fprintf(buffer, "/%s/", self.Name)
if self.IsInsensitive {
buffer.WriteString("i")
}
default:
buffer.WriteString(self.Name)
}
if self.Alias != "" {
fmt.Fprintf(buffer, " as %s", self.Alias)
}
return buffer.String()
}