108 lines
1.8 KiB
Go
108 lines
1.8 KiB
Go
package check
|
|
|
|
import (
|
|
"alrm/check/ping"
|
|
"fmt"
|
|
"time"
|
|
"strings"
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
TK_NONE = iota
|
|
TK_COUNT
|
|
TK_TIMEOUT
|
|
TK_INTERVAL
|
|
)
|
|
|
|
type CheckPing struct {
|
|
Type string
|
|
Address string
|
|
Count int
|
|
Timeout time.Duration
|
|
Interval time.Duration
|
|
state int
|
|
}
|
|
|
|
func NewCheckPing(addr string) *CheckPing {
|
|
return &CheckPing{
|
|
Type: "ping", Address: addr,
|
|
Count: 1, Timeout: time.Second * 5,
|
|
Interval: time.Second,
|
|
}
|
|
}
|
|
|
|
func (c *CheckPing) Check(debuglvl int) error {
|
|
if debuglvl > 0 {
|
|
fmt.Printf("Pinging %s .. \n", c.Address)
|
|
}
|
|
|
|
p, err := ping.NewPinger(c.Address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
p.Count = c.Count
|
|
p.Timeout = c.Timeout
|
|
p.Interval = c.Interval
|
|
|
|
err = p.Run()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
stats := p.Statistics()
|
|
if len(stats.Rtts) < 1 {
|
|
return fmt.Errorf("ping failure")
|
|
}
|
|
|
|
if debuglvl > 0 {
|
|
for i, r := range stats.Rtts {
|
|
fmt.Printf("Ping %d: %s\n", i+1, r)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *CheckPing) Parse(tk string) (bool, error) {
|
|
var err error
|
|
switch c.state {
|
|
case TK_NONE:
|
|
switch strings.ToLower(tk){
|
|
case "count":
|
|
c.state = TK_COUNT
|
|
case "timeout":
|
|
c.state = TK_TIMEOUT
|
|
case "interval":
|
|
c.state = TK_INTERVAL
|
|
default:
|
|
return false, nil
|
|
}
|
|
|
|
case TK_COUNT:
|
|
c.Count, err = strconv.Atoi(tk)
|
|
if err != nil {
|
|
return false, fmt.Errorf("invalid count \"%s\"", tk)
|
|
}
|
|
c.state = TK_NONE
|
|
|
|
case TK_TIMEOUT:
|
|
c.Timeout, err = time.ParseDuration(tk)
|
|
if err != nil {
|
|
return false, fmt.Errorf("invalid timeout \"%s\"", tk)
|
|
}
|
|
c.state = TK_NONE
|
|
|
|
case TK_INTERVAL:
|
|
c.Interval, err = time.ParseDuration(tk)
|
|
if err != nil {
|
|
return false, fmt.Errorf("invalid interval \"%s\"", tk)
|
|
}
|
|
c.state = TK_NONE
|
|
|
|
default:
|
|
return false, fmt.Errorf("invalid state in check_ping")
|
|
}
|
|
return true, nil
|
|
}
|