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)
}