Refactor initialization to init func

This commit is contained in:
pluja 2023-11-13 07:14:52 +01:00
parent 3e4b3646b3
commit d3bfef3275

View File

@ -23,8 +23,9 @@ var (
listen = flag.String("listen", ":4488", "address to listen to") listen = flag.String("listen", ":4488", "address to listen to")
) )
func configSetup() { func init() {
// Config // Flags
flag.Parse()
config.Conf = config.Config{ config.Conf = config.Config{
Dev: *dev, Dev: *dev,
Cache: *nocache, Cache: *nocache,
@ -32,11 +33,6 @@ func configSetup() {
Scraper: *scraper, Scraper: *scraper,
ListenAddr: *listen, ListenAddr: *listen,
} }
}
func main() {
// Flags
flag.Parse()
configSetup()
// Flags // Flags
log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr}) log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
@ -76,18 +72,19 @@ func main() {
// Database init // Database init
log.Info().Msg("Initializing database.") log.Info().Msg("Initializing database.")
database.InitDb() database.InitDb()
defer database.Close()
// AI ToS Scraper init // AI ToS Scraper init
if config.Conf.Scraper { if config.Conf.Scraper {
log.Info().Msg("Initializing AI ToS scraper.") log.Info().Msg("Initializing AI ToS scraper.")
tos_scraper.InitTosScraperDaemon() tos_scraper.InitTosScraperDaemon()
} }
}
func main() {
// Server init // Server init
log.Info().Msg("Initializing server.") log.Info().Msg("Initializing server.")
defer database.Close()
server := server.NewServer(config.Conf.ListenAddr) server := server.NewServer(config.Conf.ListenAddr)
if err := server.Run(); err != nil { if err := server.Run(); err != nil {
log.Fatal().Err(err) log.Fatal().Err(err)
} }