/
stats.go
91 lines (84 loc) · 2.3 KB
/
stats.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
// Copyright 2019 HAProxy Technologies
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
package runtime
import (
"fmt"
"strings"
"github.com/mitchellh/mapstructure"
"github.com/haproxytech/client-native/v5/models"
)
// GetStats fetches HAProxy stats from runtime API
func (s *SingleRuntime) GetStats() *models.NativeStatsCollection {
rAPI := ""
if s.worker != 0 {
rAPI = fmt.Sprintf("%s@%v", s.socketPath, s.worker)
} else {
rAPI = s.socketPath
}
result := &models.NativeStatsCollection{RuntimeAPI: rAPI}
rawdata, err := s.ExecuteWithResponse("show stat")
if err != nil {
result.Error = err.Error()
return result
}
lines := strings.Split(rawdata[2:], "\n")
stats := []*models.NativeStat{}
keys := strings.Split(lines[0], ",")
for i := 1; i < len(lines); i++ {
data := map[string]string{}
line := strings.Split(lines[i], ",")
if len(line) < len(keys) {
continue
}
oneLineData := &models.NativeStat{}
for index, key := range keys {
if len(line[index]) > 0 {
data[key] = line[index]
}
if key == "type" {
switch line[index] {
case "0", "1":
oneLineData.Name = line[0]
oneLineData.Type = strings.ToLower(line[1])
case "2":
oneLineData.Name = line[1]
oneLineData.Type = "server"
oneLineData.BackendName = line[0]
case "3":
// we ignore listener
default:
// add logging when available fmt.Printf("unexpected stat type: %s", line[32])
}
}
}
var st models.NativeStatStats
decoder, err := mapstructure.NewDecoder(&mapstructure.DecoderConfig{
Result: &st,
WeaklyTypedInput: true,
TagName: "json",
})
if err != nil {
continue
}
err = decoder.Decode(data)
if err != nil {
continue
}
oneLineData.Stats = &st
stats = append(stats, oneLineData)
}
result.Stats = stats
return result
}