1
0
Fork 0
mirror of https://github.com/postmannen/ctrl.git synced 2025-03-15 10:57:42 +00:00
ctrl/cmd/steward/main.go
2021-08-09 09:18:30 +02:00

59 lines
1.1 KiB
Go

package main
import (
"fmt"
"log"
"net/http"
"os"
"os/signal"
"time"
_ "net/http/pprof"
"github.com/RaaLabs/steward"
)
func main() {
c := steward.NewConfiguration()
err := c.CheckFlags()
if err != nil {
log.Printf("%v\n", err)
return
}
// Start profiling if profiling port is specified
if c.ProfilingPort != "" {
go func() {
http.ListenAndServe("localhost:"+c.ProfilingPort, nil)
}()
}
s, err := steward.NewServer(c)
if err != nil {
log.Printf("%v\n", err)
os.Exit(1)
}
// Start up the server
go s.Start()
// Wait for ctrl+c to stop the server.
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, os.Interrupt)
// Block and wait for CTRL+C
sig := <-sigCh
fmt.Printf("Got exit signal, terminating all processes, %v\n", sig)
// Adding a safety function here so we can make sure that all processes
// are stopped after a given time if the context cancelation hangs.
go func() {
time.Sleep(time.Second * 10)
log.Printf("error: doing a non graceful shutdown of all processes..\n")
os.Exit(1)
}()
// Stop all processes.
s.Stop()
}