104 lines
1.9 KiB
Go
104 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"math/rand"
|
|
"os"
|
|
"os/signal"
|
|
"strconv"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/mdiluz/rove/cmd/rove-server/internal"
|
|
"github.com/mdiluz/rove/pkg/persistence"
|
|
"github.com/mdiluz/rove/pkg/version"
|
|
)
|
|
|
|
var ver = flag.Bool("version", false, "Display version number")
|
|
|
|
// Path for persistent storage
|
|
var data = os.Getenv("DATA_PATH")
|
|
|
|
// The tick rate of the server in seconds
|
|
var tick = os.Getenv("TICK_RATE")
|
|
|
|
// InnerMain is our main function so tests can run it
|
|
func InnerMain() {
|
|
// Ensure we've seeded rand
|
|
rand.Seed(time.Now().UTC().UnixNano())
|
|
|
|
flag.Parse()
|
|
|
|
// Print the version if requested
|
|
if *ver {
|
|
log.Println(version.Version)
|
|
return
|
|
}
|
|
|
|
// Address to host the server on
|
|
var iport int
|
|
var port = os.Getenv("PORT")
|
|
if len(port) == 0 {
|
|
iport = 9090
|
|
} else {
|
|
var err error
|
|
iport, err = strconv.Atoi(port)
|
|
if err != nil {
|
|
log.Fatal("$PORT not valid int")
|
|
}
|
|
}
|
|
log.Printf("Initialising version %s...\n", version.Version)
|
|
|
|
// Set the persistence path
|
|
if len(data) == 0 {
|
|
log.Fatal("DATA_PATH not set")
|
|
} else if err := persistence.SetPath(data); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Convert the tick rate
|
|
tickRate := 1
|
|
if len(tick) > 0 {
|
|
var err error
|
|
tickRate, err = strconv.Atoi(tick)
|
|
if err != nil {
|
|
log.Fatalf("TICK_RATE not set to valid int: %s", err)
|
|
}
|
|
}
|
|
|
|
// Create the server data
|
|
s := internal.NewServer(
|
|
internal.OptionAddress(fmt.Sprintf(":%d", iport)),
|
|
internal.OptionPersistentData(),
|
|
internal.OptionTick(tickRate))
|
|
|
|
// Initialise the server
|
|
if err := s.Initialise(true); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// Set up the close handler
|
|
c := make(chan os.Signal)
|
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
|
go func() {
|
|
<-c
|
|
log.Println("Quit requested, exiting...")
|
|
if err := s.Stop(); err != nil {
|
|
panic(err)
|
|
}
|
|
}()
|
|
|
|
// Run the server
|
|
s.Run()
|
|
|
|
// Close the server
|
|
if err := s.Close(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
InnerMain()
|
|
}
|