diff --git a/config_example.yaml b/config_example.yaml index c00946e..a107465 100644 --- a/config_example.yaml +++ b/config_example.yaml @@ -1,6 +1,10 @@ logging: level: DEBUG +db: + driver: sqlite3 + connect_to: "file:csgowtfd.db?_fk=1&cache=shared" + parser: worker: 6 diff --git a/main.go b/main.go index e0f8fe0..1a0a571 100644 --- a/main.go +++ b/main.go @@ -374,7 +374,7 @@ func main() { Lock: new(sync.RWMutex), } - db.Client, err = ent.Open("sqlite3", "file:csgowtfd.db?_fk=1&cache=shared") + db.Client, err = ent.Open(conf.Db.Driver, conf.Db.ConnectTo) if err != nil { log.Panicf("Failed to open database %s: %v", "opencsgo.db", err) } diff --git a/utils/utils.go b/utils/utils.go index 99083cb..86771fd 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -26,6 +26,10 @@ type Conf struct { Logging struct { Level string } + Db struct { + Driver string + ConnectTo string `yaml:"connect_to"` + } Parser struct { Worker int }