52 lines
908 B
Go
52 lines
908 B
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/mdiluz/rove/pkg/server"
|
|
"github.com/mdiluz/rove/pkg/version"
|
|
)
|
|
|
|
var ver = flag.Bool("version", false, "Display version number")
|
|
var port = flag.Int("port", 8080, "The port to host on")
|
|
var data = flag.String("data", "/tmp/", "Directory to store persistant data")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if *ver {
|
|
fmt.Println(version.Version)
|
|
os.Exit(0)
|
|
}
|
|
|
|
s := server.NewServer(
|
|
server.OptionPort(*port),
|
|
server.OptionPersistentData(*data))
|
|
|
|
fmt.Println("Initialising...")
|
|
if err := s.Initialise(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// Set up the close handler
|
|
c := make(chan os.Signal)
|
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
|
go func() {
|
|
<-c
|
|
fmt.Println("SIGTERM recieved, exiting...")
|
|
s.Close()
|
|
os.Exit(0)
|
|
}()
|
|
|
|
fmt.Println("Initialised")
|
|
|
|
s.Run()
|
|
|
|
if err := s.Close(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|