func CleanupDB() {
cleaningMutex.Lock()
defer cleaningMutex.Unlock()
o.Do(func() {
os.Setenv("DEPLOYMENT_ENV", "test")
config.SetupConstants()
database.Init()
authority.RegisterTypes()
migrations.Do()
stores.SetupStores()
})
tables := []string{
"admin_log_entries",
"banned_players_lobbies",
"chat_messages",
"http_sessions",
"lobbies",
"lobby_slots",
"player_bans",
"player_settings",
"player_stats",
"players",
"server_records",
"spectators_players_lobbies",
"requirements",
}
for _, table := range tables {
database.DB.Exec("TRUNCATE TABLE " + table + " RESTART IDENTITY")
}
}
func main() {
helpers.InitLogger()
config.SetupConstants()
if config.Constants.ProfilerEnable {
address := "localhost:" + config.Constants.ProfilerPort
go func() {
graceful.Run(address, 1*time.Second, nil)
}()
helpers.Logger.Debug("Running Profiler at %s", address)
}
pid := &pid.Instance{}
if pid.Create() == nil {
defer pid.Remove()
}
authority.RegisterTypes()
helpers.InitAuthorization()
database.Init()
migrations.Do()
stores.SetupStores()
models.PaulingConnect()
models.FumbleConnect()
models.InitializeLobbySettings("./lobbySettingsData.json")
StartPaulingListener()
chelpers.InitDB()
if config.Constants.SteamIDWhitelist != "" {
go chelpers.WhitelistListener()
}
// lobby := models.NewLobby("cp_badlands", 10, "a", "a", 1)
helpers.Logger.Debug("Starting the server")
// init http server
// init socket.io server
server := wsevent.NewServer()
nologin := wsevent.NewServer()
broadcaster.Init(server, nologin)
socket.ServerInit(server, nologin)
routes.SetupHTTPRoutes(server, nologin)
if val := os.Getenv("DEPLOYMENT_ENV"); strings.ToLower(val) != "production" {
// init static FileServer
// TODO be careful to set this to correct location when deploying
http.HandleFunc("/static/", func(w http.ResponseWriter, r *http.Request) {
http.ServeFile(w, r, r.URL.Path[1:])
})
}
corsHandler := cors.New(cors.Options{
AllowedOrigins: config.Constants.AllowedCorsOrigins,
AllowCredentials: true,
}).Handler(http.DefaultServeMux)
// start the server
helpers.Logger.Debug("Serving at %s", config.Constants.Domain)
graceful.Run(":"+config.Constants.Port, 1*time.Second, corsHandler)
}
func TestCleanup() {
if os.Getenv("DEPLOYMENT_ENV") == "" {
os.Setenv("DEPLOYMENT_ENV", "test")
defer os.Unsetenv("DEPLOYMENT_ENV")
}
config.SetupConstants()
database.Init()
database.DB.Exec("DROP SCHEMA public CASCADE;")
database.DB.Exec("CREATE SCHEMA public;")
Do()
}
func CleanupDB() {
cleaningMutex.Lock()
defer cleaningMutex.Unlock()
if os.Getenv("DEPLOYMENT_ENV") == "" {
os.Setenv("DEPLOYMENT_ENV", "test")
defer os.Unsetenv("DEPLOYMENT_ENV")
}
config.SetupConstants()
database.Init()
database.DB.Exec("DROP SCHEMA public CASCADE;")
database.DB.Exec("CREATE SCHEMA public;")
migrations.Do()
stores.SetupStores()
}
func main() {
authority.RegisterTypes()
helpers.InitLogger()
helpers.InitAuthorization()
config.SetupConstants()
database.Init()
migrations.Do()
stores.SetupStores()
models.PaulingConnect()
go models.ReadyTimeoutListener()
StartListener()
chelpers.CheckLogger()
if config.Constants.SteamIDWhitelist != "" {
chelpers.InitSteamIDWhitelist(config.Constants.SteamIDWhitelist)
}
// lobby := models.NewLobby("cp_badlands", 10, "a", "a", 1)
helpers.Logger.Debug("Starting the server")
// init http server
routes.SetupHTTPRoutes()
// init socket.io server
socketServer, err := socketio.NewServer(nil)
if err != nil {
helpers.Logger.Fatal(err.Error())
}
broadcaster.Init(socketServer)
defer broadcaster.Stop()
routes.SetupSocketRoutes(socketServer)
http.Handle("/socket.io/", socketServer)
// init static FileServer
// TODO be careful to set this to correct location when deploying
http.HandleFunc("/static/", func(w http.ResponseWriter, r *http.Request) {
http.ServeFile(w, r, r.URL.Path[1:])
})
corsHandler := cors.New(cors.Options{
AllowedOrigins: config.Constants.AllowedCorsOrigins,
AllowCredentials: true,
}).Handler(http.DefaultServeMux)
// start the server
helpers.Logger.Debug("Serving at localhost:" + config.Constants.Port + "...")
graceful.Run(":"+config.Constants.Port, 10*time.Second, corsHandler)
}
func CleanupDB() {
cleaningMutex.Lock()
defer cleaningMutex.Unlock()
o.Do(func() {
ci := os.Getenv("CI")
config.Constants.DbAddr = "127.0.0.1:5432"
config.Constants.SteamDevAPIKey = ""
if ci == "true" {
config.Constants.DbUsername = "******"
config.Constants.DbDatabase = "travis_ci_test"
config.Constants.DbPassword = ""
} else {
config.Constants.DbDatabase = "TESTtf2stadium"
config.Constants.DbUsername = "******"
config.Constants.DbPassword = "******"
}
database.Init()
migrations.Do()
})
tables := []string{
"admin_log_entries",
"banned_players_lobbies",
"chat_messages",
"lobbies",
"lobby_slots",
"player_bans",
"player_stats",
"players",
"reports",
"requirements",
"server_records",
"spectators_players_lobbies",
"stored_servers",
}
for _, table := range tables {
database.DB.Exec("TRUNCATE TABLE " + table + " RESTART IDENTITY")
}
}
func main() {
helpers.InitLogger()
config.SetupConstants()
database.Init()
migrations.Do()
stores.SetupStores()
models.InitServerConfigs()
// lobby := models.NewLobby("cp_badlands", 10, "a", "a", 1)
helpers.Logger.Debug("Starting the server")
r := mux.NewRouter()
// init http server
routes.SetupHTTPRoutes(r)
http.Handle("/", r)
// init socket.io server
socketServer, err := socketio.NewServer(nil)
if err != nil {
helpers.Logger.Fatal(err.Error())
}
socket.InitBroadcaster(socketServer)
routes.SetupSocketRoutes(socketServer)
r.Handle("/socket.io/", socketServer)
// init static FileServer
// TODO be careful to set this to correct location when deploying
r.PathPrefix("/static/").Handler(http.StripPrefix("/static/",
http.FileServer(http.Dir(config.Constants.StaticFileLocation))))
corsHandler := cors.New(cors.Options{
AllowedOrigins: config.Constants.AllowedCorsOrigins,
AllowCredentials: true,
}).Handler(r)
// start the server
helpers.Logger.Debug("Serving at localhost:" + config.Constants.Port + "...")
graceful.Run(":"+config.Constants.Port, 10*time.Second, corsHandler)
}
func main() {
helpers.InitLogger()
flag.Parse()
if *flagGen {
key := securecookie.GenerateRandomKey(64)
if len(key) == 0 {
logrus.Fatal("Couldn't generate random key")
}
base64Key := base64.StdEncoding.EncodeToString(key)
fmt.Println(base64Key)
return
}
config.SetupConstants()
go rpc.StartRPC()
if config.Constants.ProfilerEnable {
address := "localhost:" + config.Constants.ProfilerPort
go func() {
graceful.Run(address, 1*time.Second, nil)
}()
logrus.Info("Running Profiler at %s", address)
}
pid := &pid.Instance{}
if pid.Create() == nil {
defer pid.Remove()
}
authority.RegisterTypes()
helpers.InitAuthorization()
database.Init()
migrations.Do()
stores.SetupStores()
models.InitializeLobbySettings("./lobbySettingsData.json")
models.ConnectRPC()
models.DeleteUnusedServerRecords()
go models.Ping()
chelpers.InitGeoIPDB()
if config.Constants.SteamIDWhitelist != "" {
go chelpers.WhitelistListener()
}
// lobby := models.NewLobby("cp_badlands", 10, "a", "a", 1)
mux := http.NewServeMux()
routes.SetupHTTP(mux)
socket.RegisterHandlers()
if val := os.Getenv("DEPLOYMENT_ENV"); strings.ToLower(val) != "production" {
// init static FileServer
// TODO be careful to set this to correct location when deploying
mux.HandleFunc("/static/", func(w http.ResponseWriter, r *http.Request) {
http.ServeFile(w, r, r.URL.Path[1:])
})
}
corsHandler := cors.New(cors.Options{
AllowedOrigins: config.AllowedCorsOrigins,
AllowCredentials: true,
}).Handler(context.ClearHandler(mux))
// start the server
logrus.Info("Serving on", config.HTTPAddress())
graceful.Run(config.Constants.Address, 1*time.Second, corsHandler)
}
func main() {
flag.Parse()
if *flagGen {
key := make([]byte, 64)
_, err := rand.Read(key)
if err != nil {
logrus.Fatal(err)
}
base64Key := base64.StdEncoding.EncodeToString(key)
fmt.Println(base64Key)
return
}
if *docPrint {
config.PrintConfigDoc()
os.Exit(0)
}
if helpers.Raven != nil {
hook, err := logrus_sentry.NewWithClientSentryHook(helpers.Raven, []logrus.Level{
logrus.PanicLevel,
logrus.FatalLevel,
logrus.ErrorLevel,
logrus.WarnLevel,
})
if err == nil {
logrus.AddHook(hook)
} else {
logrus.Fatal(err)
}
}
logrus.Debug("Commit: ", version.GitCommit)
logrus.Debug("Branch: ", version.GitBranch)
logrus.Debug("Build date: ", version.BuildDate)
controllers.InitTemplates()
//models.ReadServers()
err := os.Mkdir(config.Constants.DemosFolder, 0755)
if err != nil && !os.IsExist(err) {
logrus.Fatal(err)
}
if config.Constants.ProfilerAddr != "" {
go http.ListenAndServe(config.Constants.ProfilerAddr, nil)
logrus.Info("Running Profiler at ", config.Constants.ProfilerAddr)
}
database.Init()
database.DB.DB().SetMaxOpenConns(*dbMaxopen)
migrations.Do()
helpers.ConnectAMQP()
event.StartListening()
helpers.InitGeoIPDB()
err = lobbySettings.LoadLobbySettingsFromFile("assets/lobbySettingsData.json")
if err != nil {
logrus.Fatal(err)
}
lobby.CreateLocks()
rpc.ConnectRPC(helpers.AMQPConn)
lobby.RestoreServemeChecks()
//go models.TFTVStreamStatusUpdater()
if config.Constants.SteamIDWhitelist != "" {
go chelpers.WhitelistListener()
}
mux := http.NewServeMux()
routes.SetupHTTP(mux)
socket.RegisterHandlers()
corsHandler := cors.New(cors.Options{
AllowedOrigins: config.Constants.AllowedOrigins,
AllowedMethods: []string{"GET", "POST", "DELETE", "OPTIONS"},
AllowCredentials: true,
}).Handler(mux)
sig := make(chan os.Signal, 1)
signal.Notify(sig, os.Interrupt, os.Kill, syscall.SIGTERM, syscall.SIGKILL)
go func() {
<-sig
shutdown()
os.Exit(0)
}()
logrus.Info("Serving on ", config.Constants.ListenAddress)
logrus.Fatal(http.ListenAndServe(config.Constants.ListenAddress, corsHandler))
}