コード例 #1
0
ファイル: scheduler.go プロジェクト: raintank/raintank-probe
func GetCheck(checkType m.CheckType, settings map[string]interface{}) (RaintankProbeCheck, error) {
	switch checkType {
	case m.PING_CHECK:
		return checks.NewRaintankPingProbe(settings)
	case m.DNS_CHECK:
		return checks.NewRaintankDnsProbe(settings)
	case m.HTTP_CHECK:
		return checks.NewRaintankHTTPProbe(settings)
	case m.HTTPS_CHECK:
		return checks.NewRaintankHTTPSProbe(settings)
	default:
		return nil, fmt.Errorf("unknown check type %s ", checkType)
	}

}
コード例 #2
0
ファイル: main.go プロジェクト: AriaEdo/raintank-probe
func GetCheck(checkType string, body []byte) (RaintankProbeCheck, error) {
	switch checkType {
	case "ping":
		return checks.NewRaintankPingProbe(body)
	case "dns":
		return checks.NewRaintankDnsProbe(body)
	case "http":
		return checks.NewRaintankHTTPProbe(body)
	case "https":
		return checks.NewRaintankHTTPSProbe(body)
	default:
		return nil, fmt.Errorf("unknown check type. " + checkType)
	}

}
コード例 #3
0
ファイル: health.go プロジェクト: raintank/raintank-probe
// Ping scheduler.HealthHosts to determin if this probe is healthy and should
// execute checks.  If all of the HealthHosts are experiencing issues, then
// there is likely something wrong with this probe so it should stop executing
// checks until things recover.
//
func (s *Scheduler) CheckHealth() {
	chks := make([]*checks.RaintankProbePing, len(s.HealthHosts))
	for i, host := range s.HealthHosts {
		settings := make(map[string]interface{})
		settings["timeout"] = 1.0
		settings["hostname"] = host
		chk, err := checks.NewRaintankPingProbe(settings)
		if err != nil {
			log.Fatal(4, "unable to create health check. %s", err)
		}
		chks[i] = chk
	}

	lastState := 1

	ticker := time.NewTicker(time.Second)
	var wg sync.WaitGroup
	for range ticker.C {
		resultsCh := make(chan int, len(chks))
		for i := range chks {
			check := chks[i]
			wg.Add(1)
			go func(ch chan int, chk *checks.RaintankProbePing) {
				defer wg.Done()
				results, err := chk.Run()
				if err != nil {
					ch <- 3
					return
				}
				if results.ErrorMsg() != "" {
					log.Warn("Health check to %s failed. %s", chk.Hostname, results.ErrorMsg())
					ch <- 1
					return
				}
				ch <- 0
			}(resultsCh, check)
		}
		wg.Wait()
		close(resultsCh)
		score := 0
		for r := range resultsCh {
			if r == 3 {
				// fatal error, trying to run the check.
				score = len(chks)
			} else {
				score += r
			}
		}
		newState := 0
		// if more the 50% of healthHosts are down, then we consider ourselves down.
		if float64(score) > float64(len(chks)/2.0) {
			newState = 1
		}

		if newState != lastState {
			if newState == 1 {
				// we are now unhealthy.
				s.Lock()
				log.Warn("This probe is in an unhealthy state. Stopping execution of checks.")
				s.Healthy = false
				for _, instance := range s.Checks {
					instance.Stop()
				}
				s.Unlock()
			} else {
				//we are now healthy.
				s.Lock()
				log.Warn("This probe is now healthy again. Resuming execution of checks.")
				s.Healthy = true
				for _, instance := range s.Checks {
					log.Debug("starting %s check for %s", instance.Check.Type, instance.Check.Slug)
					go instance.Run()
				}
				s.Unlock()
			}
			lastState = newState
		}

	}
}