forked from ledgerwatch/erigon
/
parsing.go
111 lines (96 loc) · 2.07 KB
/
parsing.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
package metrics
import (
"fmt"
"regexp"
"strings"
"github.com/prometheus/client_golang/prometheus"
)
func parseMetric(s string) (string, prometheus.Labels, error) {
if len(s) == 0 {
return "", nil, fmt.Errorf("metric cannot be empty")
}
n := strings.IndexByte(s, '{')
if n < 0 {
if err := validateIdent(s); err != nil {
return "", nil, err
}
return s, nil, nil
}
ident := s[:n]
s = s[n+1:]
if err := validateIdent(ident); err != nil {
return "", nil, err
}
if len(s) == 0 || s[len(s)-1] != '}' {
return "", nil, fmt.Errorf("missing closing curly brace at the end of %q", ident)
}
tags, err := parseTags(s[:len(s)-1])
if err != nil {
return "", nil, err
}
return ident, tags, nil
}
func parseTags(s string) (prometheus.Labels, error) {
if len(s) == 0 {
return nil, nil
}
var labels prometheus.Labels
for {
n := strings.IndexByte(s, '=')
if n < 0 {
return nil, fmt.Errorf("missing `=` after %q", s)
}
ident := s[:n]
s = s[n+1:]
if err := validateIdent(ident); err != nil {
return nil, err
}
if len(s) == 0 || s[0] != '"' {
return nil, fmt.Errorf("missing starting `\"` for %q value; tail=%q", ident, s)
}
s = s[1:]
value := ""
for {
n = strings.IndexByte(s, '"')
if n < 0 {
return nil, fmt.Errorf("missing trailing `\"` for %q value; tail=%q", ident, s)
}
m := n
for m > 0 && s[m-1] == '\\' {
m--
}
if (n-m)%2 == 1 {
value = value + s[:n]
s = s[n+1:]
continue
}
value = value + s[:n]
if labels == nil {
labels = prometheus.Labels{}
}
labels[ident] = value
s = s[n+1:]
if len(s) == 0 {
return labels, nil
}
if !strings.HasPrefix(s, ",") {
return nil, fmt.Errorf("missing `,` after %q value; tail=%q", ident, s)
}
s = skipSpace(s[1:])
break
}
}
}
func skipSpace(s string) string {
for len(s) > 0 && s[0] == ' ' {
s = s[1:]
}
return s
}
func validateIdent(s string) error {
if !identRegexp.MatchString(s) {
return fmt.Errorf("invalid identifier %q", s)
}
return nil
}
var identRegexp = regexp.MustCompile("^[a-zA-Z_:.][a-zA-Z0-9_:.]*$")