45 lines
887 B
Go
45 lines
887 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"github.com/golang/glog"
|
|
c "github.com/mayuresh82/gocast/config"
|
|
"github.com/mayuresh82/gocast/controller"
|
|
"github.com/mayuresh82/gocast/server"
|
|
log "github.com/sirupsen/logrus"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
)
|
|
|
|
var (
|
|
config = flag.String("config", "", "Path to config file")
|
|
)
|
|
|
|
func main() {
|
|
if glog.V(4) {
|
|
log.SetLevel(log.DebugLevel)
|
|
}
|
|
flag.Parse()
|
|
conf := c.GetConfig(*config)
|
|
mon := controller.NewMonitor(conf)
|
|
srv := server.NewServer(conf.Agent.ListenAddr, mon)
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
// catch interrupt
|
|
signalChan := make(chan os.Signal, 1)
|
|
signal.Notify(signalChan, os.Interrupt, syscall.SIGHUP, syscall.SIGTERM)
|
|
go func() {
|
|
for {
|
|
sig := <-signalChan
|
|
if sig == os.Interrupt || sig == syscall.SIGTERM {
|
|
mon.CloseAll()
|
|
cancel()
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
srv.Serve(ctx)
|
|
}
|