noury/mirror-nourybot
1
0
Fork 0
mirror of https://github.com/lyx0/nourybot.git synced 2024-11-13 19:49:55 +01:00
Code Issues Projects Releases Packages Wiki Activity

add env flag for database connection

Browse source
This commit is contained in:
lyx0 2023-03-03 21:16:32 +00:00
parent 0ff77d63b3
commit e2a4e32dae
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
cmd/bot/main.go
View file

@ -3,6 +3,7 @@ package main
import (
"context"
"database/sql"
"flag"
"log"
"os"
"time"
@ -41,6 +42,13 @@ type Application struct {
Scheduler *cron.Cron
}
var envFlag string
func init() {
flag.StringVar(&envFlag, "env", "dev", "database connection to use: (dev/prod)")
flag.Parse()
}
func main() {
var cfg config
@ -62,11 +70,13 @@ func main() {
cfg.commandPrefix = os.Getenv("TWITCH_COMMAND_PREFIX")
tc := twitch.NewClient(cfg.twitchUsername, cfg.twitchOauth)
// Will be used someday Copesen
cfg.environment = "Development"
switch envFlag {
case "dev":
cfg.db.dsn = os.Getenv("LOCAL_DSN")
case "prod":
cfg.db.dsn = os.Getenv("SUPABASE_DSN")
}
// Database config variables
cfg.db.dsn = os.Getenv("DB_DSN")
cfg.db.maxOpenConns = 25
cfg.db.maxIdleConns = 25
cfg.db.maxIdleTime = "15m"