forked from open-falcon/falcon-plus
-
Notifications
You must be signed in to change notification settings - Fork 0
/
sdk.go
78 lines (62 loc) · 1.78 KB
/
sdk.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
package sdk
import (
"encoding/json"
"fmt"
cmodel "github.com/open-falcon/falcon-plus/common/model"
"github.com/open-falcon/falcon-plus/common/sdk/requests"
"github.com/open-falcon/falcon-plus/modules/aggregator/g"
f "github.com/open-falcon/falcon-plus/modules/api/app/model/falcon_portal"
"github.com/toolkits/net/httplib"
"time"
)
func HostnamesByID(group_id int64) ([]string, error) {
uri := fmt.Sprintf("%s/api/v1/hostgroup/%d", g.Config().Api.PlusApi, group_id)
req, err := requests.CurlPlus(uri, "GET", "aggregator", g.Config().Api.PlusApiToken,
map[string]string{}, map[string]string{})
if err != nil {
return []string{}, err
}
type RESP struct {
HostGroup f.HostGroup `json:"hostgroup"`
Hosts []f.Host `json:"hosts"`
}
resp := &RESP{}
err = req.ToJson(&resp)
if err != nil {
return []string{}, err
}
hosts := []string{}
for _, x := range resp.Hosts {
hosts = append(hosts, x.Hostname)
}
return hosts, nil
}
func QueryLastPoints(endpoints, counters []string) (resp []*cmodel.GraphLastResp, err error) {
cfg := g.Config()
uri := fmt.Sprintf("%s/api/v1/graph/lastpoint", cfg.Api.PlusApi)
var req *httplib.BeegoHttpRequest
headers := map[string]string{"Content-type": "application/json"}
req, err = requests.CurlPlus(uri, "POST", "aggregator", cfg.Api.PlusApiToken,
headers, map[string]string{})
if err != nil {
return
}
req.SetTimeout(time.Duration(cfg.Api.ConnectTimeout)*time.Millisecond,
time.Duration(cfg.Api.RequestTimeout)*time.Millisecond)
body := []*cmodel.GraphLastParam{}
for _, e := range endpoints {
for _, c := range counters {
body = append(body, &cmodel.GraphLastParam{e, c})
}
}
b, err := json.Marshal(body)
if err != nil {
return
}
req.Body(b)
err = req.ToJson(&resp)
if err != nil {
return
}
return resp, nil
}