added maxpacketloss variable for ping check

check/check_ping.go

@@ -3,9 +3,9 @@ package check
 import (
 	"alrm/check/ping"
 	"fmt"
-	"time"
-	"strings"
 	"strconv"
+	"strings"
+	"time"
 )
 
 const (
@@ -13,22 +13,24 @@ const (
 	TK_COUNT
 	TK_TIMEOUT
 	TK_INTERVAL
+	TK_MAXPACKETLOSS
 )
 
 type CheckPing struct {
-	Type       string
-	Address    string
-	Count      int
-	Timeout    time.Duration
-	Interval   time.Duration
-	state      int
+	Type          string
+	Address       string
+	Count         int
+	Timeout       time.Duration
+	Interval      time.Duration
+	MaxPacketLoss float64
+	state         int
 }
 
 func NewCheckPing(addr string) *CheckPing {
 	return &CheckPing{
 		Type: "ping", Address: addr,
 		Count: 1, Timeout: time.Second * 5,
-		Interval: time.Second,
+		Interval: time.Second, MaxPacketLoss: 0,
 	}
 }
 
@@ -52,56 +54,70 @@ func (c *CheckPing) Check(debuglvl int) error {
 	}
 
 	stats := p.Statistics()
-	if len(stats.Rtts) < 1 {
-		return fmt.Errorf("ping failure")
-	}
 
-	if debuglvl > 0 {
+	if debuglvl > 1 {
 		for i, r := range stats.Rtts {
 			fmt.Printf("Ping %d: %s\n", i+1, r)
 		}
 	}
+
+	if debuglvl > 0 {
+		fmt.Printf("Packet loss: %.f%%\n", stats.PacketLoss)
+	}
+
+	if stats.PacketLoss > c.MaxPacketLoss {
+		return fmt.Errorf("ping packet loss exceeds max")
+	}
 	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
-
+	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
+		case "maxpacketloss":
+			c.state = TK_MAXPACKETLOSS
 		default:
-			return false, fmt.Errorf("invalid state in check_ping")
+			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
+
+	case TK_MAXPACKETLOSS:
+		c.MaxPacketLoss, err = strconv.ParseFloat(tk, 64)
+		if err != nil {
+			return false, fmt.Errorf("invalid minpacketloss \"%s\"", tk)
+		}
+		c.state = TK_NONE
+
+	default:
+		return false, fmt.Errorf("invalid state in check_ping")
 	}
 	return true, nil
 }