github
/
statping
mirror of https://github.com/statping/statping
1
0
Fork 0
Code Issues Releases Wiki Activity

docker

pull/10/head
Hunter Long 2018-06-09 22:05:08 -07:00
parent c8907a3784
commit cdec79ae8f
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
checker.go
View File

@ -25,23 +25,27 @@ func (s *Service) CheckQueue() {
func (s *Service) Check() { func (s *Service) Check() {
t1 := time.Now() t1 := time.Now()
response, err := http.Get(s.Domain) client := http.Client{
Timeout: 30 * time.Second,
}
response, err := client.Get(s.Domain)
t2 := time.Now() t2 := time.Now()
s.Latency = t2.Sub(t1).Seconds() s.Latency = t2.Sub(t1).Seconds()
if err != nil { if err != nil {
s.Failure(response, fmt.Sprintf("HTTP Error %v", err)) s.Failure(fmt.Sprintf("HTTP Error %v", err))
return return
} }
defer response.Body.Close()
if s.Expected != "" { if s.Expected != "" {
contents, _ := ioutil.ReadAll(response.Body) contents, _ := ioutil.ReadAll(response.Body)
match, _ := regexp.MatchString(s.Expected, string(contents)) match, _ := regexp.MatchString(s.Expected, string(contents))
if !match { if !match {
s.Failure(response, fmt.Sprintf("HTTP Response Body did not match '%v'", s.Expected)) s.Failure(fmt.Sprintf("HTTP Response Body did not match '%v'", s.Expected))
return return
} }
} }
if s.ExpectedStatus != response.StatusCode { if s.ExpectedStatus != response.StatusCode {
s.Failure(response, fmt.Sprintf("HTTP Status Code %v did not match %v", response.StatusCode, s.ExpectedStatus)) s.Failure(fmt.Sprintf("HTTP Status Code %v did not match %v", response.StatusCode, s.ExpectedStatus))
return return
} }
s.Online = true s.Online = true
@ -49,11 +53,9 @@ func (s *Service) Check() {
} }
func (s *Service) Record(response *http.Response) { func (s *Service) Record(response *http.Response) {
defer response.Body.Close()
db.QueryRow("INSERT INTO hits(service,latency,created_at) VALUES($1,$2,NOW()) returning id;", s.Id, s.Latency).Scan() db.QueryRow("INSERT INTO hits(service,latency,created_at) VALUES($1,$2,NOW()) returning id;", s.Id, s.Latency).Scan()
} }
func (s *Service) Failure(response *http.Response, issue string) { func (s *Service) Failure(issue string) {
db.QueryRow("INSERT INTO failures(issue,service,created_at) VALUES($1,$2,NOW()) returning id;", issue, s.Id).Scan() db.QueryRow("INSERT INTO failures(issue,service,created_at) VALUES($1,$2,NOW()) returning id;", issue, s.Id).Scan()
s.Record(response)
} }