func Poll() error { client, err := util.GetRancherClient() if err != nil { return err } if client == nil { return fmt.Errorf("Can not create RancherClient, No credentials found") } m := &Monitor{ client: client, reportedStatus: map[string]string{}, } for stat := range m.getStats() { m.processStat(stat) } return nil }
func main() { err := config.Parse() if err != nil { logrus.Fatal(err) } flag.Parse() defer glog.Flush() if config.Config.PidFile != "" { logrus.Infof("Writing pid %d to %s", os.Getpid(), config.Config.PidFile) if err := ioutil.WriteFile(config.Config.PidFile, []byte(strconv.Itoa(os.Getpid())), 0644); err != nil { logrus.Fatalf("Failed to write pid file %s: %v", config.Config.PidFile, err) } } if config.Config.LogFile != "" { if output, err := os.OpenFile(config.Config.LogFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666); err != nil { logrus.Fatalf("Failed to log to file %s: %v", config.Config.LogFile, err) } else { logrus.SetOutput(output) } } if config.Config.HaProxyMonitor { err := healthcheck.Poll() if err != nil { logrus.Fatal(err) } os.Exit(0) } processor := events.NewDockerEventsProcessor(config.Config.EventsPoolSize) err = processor.Process() if err != nil { logrus.Fatal(err) } rancherClient, err := util.GetRancherClient() if err != nil { logrus.Fatal(err) } tokenRequest := &rclient.HostApiProxyToken{ ReportedUuid: config.Config.HostUuid, } tokenResponse, err := getConnectionToken(0, tokenRequest, rancherClient) if err != nil { logrus.Fatal(err) } else if tokenResponse == nil { // nil error and blank token means the proxy is turned off. Just block forever so main function doesn't exit var block chan bool <-block } handlers := make(map[string]backend.Handler) handlers["/v1/logs/"] = &logs.LogsHandler{} handlers["/v1/stats/"] = &stats.StatsHandler{} handlers["/v1/hoststats/"] = &stats.HostStatsHandler{} handlers["/v1/containerstats/"] = &stats.ContainerStatsHandler{} handlers["/v1/exec/"] = &exec.ExecHandler{} handlers["/v1/dockersocket/"] = &dockersocketproxy.Handler{} backend.ConnectToProxy(tokenResponse.Url+"?token="+tokenResponse.Token, handlers) }