コード例 #1
0
func TrapKillSignals() {
	c := make(chan os.Signal, 1)
	signal.Notify(c, syscall.SIGINT)
	signal.Notify(c, syscall.SIGTERM)
	signal.Notify(c, syscall.SIGKILL)
	go func() {
		for range c {
			log.Println("Caught a fatal signal. Terminating.")
			app.CallExitFuncs()
			os.Exit(1)
		}
	}()
}
コード例 #2
0
func main() {
	flag.Parse()

	// Make sure exit functions get called. This program has no normal exit -- it either dies from panics
	// or from catching fatal signals. The former case is handled by defer. The latter case is handled
	// by TrapKillSignals.
	defer app.CallExitFuncs()
	TrapKillSignals()

	if !*stderr {
		// Set up log file.
		logname := path.Join(os.Getenv("HOME"), "server.log")
		err := os.Rename(logname, fmt.Sprintf(
			"%s.%s.%d", logname, time.Now().Format("2006-01-02"), time.Now().Unix()))
		if err != nil && !os.IsNotExist(err) {
			log.Fatalln(err)
		}
		logfile, err := os.Create(logname)
		checkInitError(err)
		app.RunAtExit(func() { logfile.Close() })
		log.Println("Subsequent log messages will go to disk, not stderr.")
		log.SetOutput(logfile)
	} else {
		log.Println("Logging to stderr.")
	}
	log.Println("Starting server.")

	// Static files.
	pwd, err := os.Getwd()
	checkInitError(err)
	http.Handle("/", http.FileServer(http.Dir(path.Join(pwd, "static-files"))))

	// Special handler for static graphs.
	http.Handle("/graph/", http.StripPrefix("/graph/", http.FileServer(http.Dir(app.TempGraphDir))))

	// Visualization builder.
	http.HandleFunc("/visualize", visualizeHandler)

	// Loading page.
	template.Must(template.ParseFiles("templates/loading.html")) // Sanity check.
	http.HandleFunc("/loading", loadingHandler)

	// Dashboard page.
	template.Must(template.ParseFiles("templates/dashboard.html")) // Sanity check.
	http.HandleFunc("/dashboard", dashboardHandler)

	// Contact info signup.
	smtpPassword = os.Getenv("EMAIL_CHARTS_SMTP_PASSWORD")
	if smtpPassword == "" {
		log.Println("Must set EMAIL_CHARTS_SMTP_PASSWORD")
		return
	}
	http.HandleFunc("/staytuned", signupHandler)
	ok := func(w http.ResponseWriter, _ *http.Request) {
		w.Write([]byte("ok"))
	}
	http.HandleFunc("/_ah/start", ok)
	http.HandleFunc("/_ah/health", ok)

	log.Printf(http.ListenAndServe(*httpAddr, Log(http.DefaultServeMux)).Error())
}