68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
|
|
"git.snrd.de/Spaenny/steamServer"
|
|
"git.snrd.de/Spaenny/steamServer/postgres"
|
|
"git.snrd.de/Spaenny/steamServer/web"
|
|
"github.com/rumblefrog/go-a2s"
|
|
"github.com/robfig/cron/v3"
|
|
)
|
|
|
|
func main() {
|
|
dsn := "postgres://postgres:secret@postgres:5432/postgres?sslmode=disable"
|
|
|
|
store, err := postgres.NewStore(dsn)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
ss, err := store.Servers()
|
|
if err != nil {
|
|
log.Fatal("There is an issue with the databse: %w", err)
|
|
}
|
|
|
|
for _, s := range ss {
|
|
serverString := s.ServerIP
|
|
c := cron.New()
|
|
c.AddFunc("*/1 * * * *", func() { updateServers(serverString, store) })
|
|
c.Start()
|
|
}
|
|
|
|
|
|
h := web.NewHandler(store)
|
|
http.ListenAndServe(":3000", h)
|
|
}
|
|
|
|
func updateServers(host string, store steamServer.Store) {
|
|
client, err := a2s.NewClient(host)
|
|
|
|
if err != nil {
|
|
log.Fatal("Creation of client failed: %w", err)
|
|
return
|
|
}
|
|
|
|
defer client.Close()
|
|
|
|
info, err := client.QueryInfo()
|
|
|
|
if err != nil {
|
|
log.Printf("Querying failed: %s", err)
|
|
return
|
|
}
|
|
|
|
log.Printf("%s - %s\n", host, info.Name)
|
|
|
|
server := steamServer.Server {
|
|
AppID: info.ID,
|
|
Name: info.Name,
|
|
Map: info.Map,
|
|
Players: info.Players,
|
|
MaxPlayers: info.MaxPlayers,
|
|
}
|
|
|
|
store.UpdateServerByQuery(&server, host)
|
|
}
|