bergquist
f7ed24475c
simple solution for waiting for all go sub routines to finish before closing Grafana. We would use errGroup here as well but I dont like spreading context's all over the place. closes #10131
122 lines
3.0 KiB
Go
122 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"runtime"
|
|
"runtime/trace"
|
|
"strconv"
|
|
"syscall"
|
|
"time"
|
|
|
|
"net/http"
|
|
_ "net/http/pprof"
|
|
|
|
"github.com/grafana/grafana/pkg/log"
|
|
"github.com/grafana/grafana/pkg/metrics"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
|
|
_ "github.com/grafana/grafana/pkg/services/alerting/conditions"
|
|
_ "github.com/grafana/grafana/pkg/services/alerting/notifiers"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/cloudwatch"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/graphite"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/influxdb"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/mysql"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/opentsdb"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/postgres"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/prometheus"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/testdata"
|
|
)
|
|
|
|
var version = "4.6.0"
|
|
var commit = "NA"
|
|
var buildstamp string
|
|
var build_date string
|
|
|
|
var configFile = flag.String("config", "", "path to config file")
|
|
var homePath = flag.String("homepath", "", "path to grafana install/home path, defaults to working directory")
|
|
var pidFile = flag.String("pidfile", "", "path to pid file")
|
|
var exitChan = make(chan int)
|
|
|
|
func main() {
|
|
v := flag.Bool("v", false, "prints current version and exits")
|
|
profile := flag.Bool("profile", false, "Turn on pprof profiling")
|
|
profilePort := flag.Int("profile-port", 6060, "Define custom port for profiling")
|
|
flag.Parse()
|
|
if *v {
|
|
fmt.Printf("Version %s (commit: %s)\n", version, commit)
|
|
os.Exit(0)
|
|
}
|
|
|
|
if *profile {
|
|
runtime.SetBlockProfileRate(1)
|
|
go func() {
|
|
http.ListenAndServe(fmt.Sprintf("localhost:%d", *profilePort), nil)
|
|
}()
|
|
|
|
f, err := os.Create("trace.out")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
err = trace.Start(f)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer trace.Stop()
|
|
}
|
|
|
|
buildstampInt64, _ := strconv.ParseInt(buildstamp, 10, 64)
|
|
if buildstampInt64 == 0 {
|
|
buildstampInt64 = time.Now().Unix()
|
|
}
|
|
|
|
setting.BuildVersion = version
|
|
setting.BuildCommit = commit
|
|
setting.BuildStamp = buildstampInt64
|
|
|
|
metrics.M_Grafana_Version.WithLabelValues(version).Set(1)
|
|
shutdownCompleted := make(chan int)
|
|
server := NewGrafanaServer()
|
|
|
|
go listenToSystemSignals(server, shutdownCompleted)
|
|
|
|
go func() {
|
|
code := 0
|
|
if err := server.Start(); err != nil {
|
|
log.Error2("Startup failed", "error", err)
|
|
code = 1
|
|
}
|
|
|
|
exitChan <- code
|
|
}()
|
|
|
|
code := <-shutdownCompleted
|
|
log.Info2("Grafana shutdown completed.", "code", code)
|
|
log.Close()
|
|
os.Exit(code)
|
|
}
|
|
|
|
func listenToSystemSignals(server *GrafanaServerImpl, shutdownCompleted chan int) {
|
|
signalChan := make(chan os.Signal, 1)
|
|
ignoreChan := make(chan os.Signal, 1)
|
|
code := 0
|
|
|
|
signal.Notify(ignoreChan, syscall.SIGHUP)
|
|
signal.Notify(signalChan, os.Interrupt, os.Kill, syscall.SIGTERM)
|
|
|
|
select {
|
|
case sig := <-signalChan:
|
|
trace.Stop() // Stops trace if profiling has been enabled
|
|
server.Shutdown(0, fmt.Sprintf("system signal: %s", sig))
|
|
shutdownCompleted <- 0
|
|
case code = <-exitChan:
|
|
trace.Stop() // Stops trace if profiling has been enabled
|
|
server.Shutdown(code, "startup error")
|
|
shutdownCompleted <- code
|
|
}
|
|
}
|