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
144
145
146
147
|
/*
** Zabbix
** Copyright (C) 2001-2021 Zabbix SIA
**
** This program is free software; you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation; either version 2 of the License, or
** (at your option) any later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software
** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**/
package ceph
import (
"context"
"crypto/tls"
"net/http"
"time"
"zabbix.com/pkg/zbxerr"
"zabbix.com/pkg/plugin"
)
const pluginName = "Ceph"
// Common params: [connString][,user][,apikey]
const commonParamsNum = 3
// Plugin inherits plugin.Base and store plugin-specific data.
type Plugin struct {
plugin.Base
options PluginOptions
client *http.Client
}
// impl is the pointer to the plugin implementation.
var impl Plugin
// whereToConnect builds a URI based on key's parameters and a configuration file.
func whereToConnect(params []string, sessions map[string]*Session, defaultURI string) (u *URI, err error) {
user := ""
if len(params) > 1 {
user = params[1]
}
apikey := ""
if len(params) > 2 {
apikey = params[2]
}
uri := defaultURI
// The first param can be either a URI or a session identifier
if len(params) > 0 && len(params[0]) > 0 {
if isLooksLikeURI(params[0]) {
// Use a URI defined as key's parameter
uri = params[0]
} else {
if _, ok := sessions[params[0]]; !ok {
return nil, zbxerr.ErrorUnknownSession
}
// Use a pre-defined session
uri = sessions[params[0]].URI
user = sessions[params[0]].User
apikey = sessions[params[0]].APIKey
}
}
return newURIWithCreds(uri, user, apikey)
}
// Export implements the Exporter interface.
func (p *Plugin) Export(key string, params []string, _ plugin.ContextProvider) (result interface{}, err error) {
uri, err := whereToConnect(params, p.options.Sessions, p.options.URI)
if err != nil {
return nil, zbxerr.New(err.Error())
}
if len(params) > commonParamsNum {
return nil, zbxerr.ErrorTooManyParameters
}
m := metrics[key]
responses := make(map[command][]byte)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
resCh := asyncRequest(ctx, cancel, p.client, uri.String(), m)
for range m.commands {
r := <-resCh
if r.err != nil {
if err == nil {
err = r.err
}
break
}
responses[command(r.cmd)] = r.data
}
if err != nil {
// Special logic of processing connection errors is used if keyPing is requested
// because it must return pingFailed if any error occurred.
if key == keyPing {
return pingFailed, nil
}
return nil, err
}
result, err = m.handle(responses)
if err != nil {
p.Errf(err.Error())
}
return result, err
}
// Start implements the Runner interface and performs initialization when plugin is activated.
func (p *Plugin) Start() {
p.client = &http.Client{
Timeout: time.Duration(p.options.Timeout) * time.Second,
}
p.client.Transport = &http.Transport{
DisableKeepAlives: false,
IdleConnTimeout: time.Duration(p.options.KeepAlive) * time.Second,
TLSClientConfig: &tls.Config{InsecureSkipVerify: p.options.InsecureSkipVerify},
}
}
// Stop implements the Runner interface and frees resources when plugin is deactivated.
func (p *Plugin) Stop() {
p.client.CloseIdleConnections()
p.client = nil
}
|
