container_network/crawler/health/ssh/SSHHealthCrawler.go
2018-07-05 17:33:10 +09:00

65 lines
1.3 KiB
Go

package ssh
import (
"fmt"
"strings"
"time"
cnsms "git.loafle.net/commons/service_matcher-go/ssh"
ocmsc "git.loafle.net/overflow/commons-go/model/sensorconfig"
"git.loafle.net/overflow/container_network/crawler/health"
"git.loafle.net/overflow/crawler-go"
)
type SSHHealthCrawler struct {
health.SocketHealthCrawler
}
func (c *SSHHealthCrawler) Key() string {
return "SSH_HEALTH"
}
func (c *SSHHealthCrawler) String() string {
return "SSH Health Crawler"
}
func (c *SSHHealthCrawler) Auth(auth map[string]interface{}) error {
return nil
}
func (c *SSHHealthCrawler) Get(config *ocmsc.SensorConfig) (map[string]string, error) {
result := make(map[string]string, 0)
conn, err := c.GetConnection(config)
if err != nil {
return nil, err
}
defer conn.Close()
for _, mci := range config.MetaCollectionItems {
switch mci.Key {
case "ssh.response_time":
start := time.Now().UTC()
if err := c.CheckHealth(config, conn); err != nil {
return nil, err
}
elapsed := time.Since(start)
if !strings.Contains(elapsed.String(), "ms") {
return nil, fmt.Errorf("Not valid data : %s", elapsed)
}
result[mci.Key] = strings.Split(elapsed.String(), "ms")[0]
break
default:
break
}
}
return result, nil
}
func NewCrawler() crawler.Crawler {
c := &SSHHealthCrawler{}
c.SetMatcher(cnsms.NewMatcher())
return c
}