-
Notifications
You must be signed in to change notification settings - Fork 5
/
main.go
88 lines (79 loc) · 2.73 KB
/
main.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
package main
import (
"fmt"
"math"
"github.com/facebookincubator/ntp/ntpcheck/checker"
"github.com/sensu-community/sensu-plugin-sdk/sensu"
"github.com/sensu/sensu-go/types"
)
// Config represents the check plugin config.
type Config struct {
sensu.PluginConfig
Warning float64
Critical float64
}
var (
plugin = Config{
PluginConfig: sensu.PluginConfig{
Name: "check-ntp",
Short: "Check NTP offset and provide metrics",
Keyspace: "sensu.io/plugins/check-ntp/config",
},
}
options = []*sensu.PluginConfigOption{
{
Path: "critical",
Argument: "critical",
Shorthand: "c",
Default: float64(100),
Usage: "Critical threshold for offset in ms",
Value: &plugin.Critical,
},
{
Path: "warning",
Argument: "warning",
Shorthand: "w",
Default: float64(10),
Usage: "Warning threshold for offset in ms",
Value: &plugin.Warning,
},
}
)
func main() {
check := sensu.NewGoCheck(&plugin.PluginConfig, options, checkArgs, executeCheck, false)
check.Execute()
}
func checkArgs(event *types.Event) (int, error) {
if plugin.Critical == 0 {
return sensu.CheckStateWarning, fmt.Errorf("--critical is required")
}
if plugin.Warning == 0 {
return sensu.CheckStateWarning, fmt.Errorf("--warning is required")
}
if plugin.Warning > plugin.Critical {
return sensu.CheckStateWarning, fmt.Errorf("--warning cannot be greater than --critical")
}
return sensu.CheckStateOK, nil
}
func executeCheck(event *types.Event) (int, error) {
result, err := checker.RunCheck("")
if err != nil {
fmt.Printf("%s CRITICAL: failed to run check, error: %v\n", plugin.PluginConfig.Name, err)
return sensu.CheckStateCritical, nil
}
stats, err := checker.NewNTPStats(result)
if err != nil {
fmt.Printf("%s CRITICAL: failed to extract NTP statistics, error: %v\n", plugin.PluginConfig.Name, err)
return sensu.CheckStateCritical, nil
}
perfData := fmt.Sprintf("clk_jitter=%f, clk_wander=%f, frequency=%f, mintc=%d, offset=%f, stratum=%d, sys_jitter=%f, tc=%d", stats.PeerJitter, result.SysVars.ClkWander, stats.Frequency, result.SysVars.MinTC, stats.PeerOffset, stats.PeerStratum, result.SysVars.SysJitter, result.SysVars.TC)
if math.Abs(stats.PeerOffset) > plugin.Critical {
fmt.Printf("%s CRITICAL: offset %.3f exceeds threshold | %s\n", plugin.PluginConfig.Name, stats.PeerOffset, perfData)
return sensu.CheckStateCritical, nil
} else if math.Abs(stats.PeerOffset) > plugin.Warning {
fmt.Printf("%s WARNING: offset %.3f exceeds threshold | %s\n", plugin.PluginConfig.Name, stats.PeerOffset, perfData)
return sensu.CheckStateWarning, nil
}
fmt.Printf("%s OK: offset %.3f within thresholds | %s\n", plugin.PluginConfig.Name, stats.PeerOffset, perfData)
return sensu.CheckStateOK, nil
}