Joren/HwidBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Refactor and modify db init

Browse Source
This commit is contained in:
Joren 2024-07-06 17:27:04 +02:00
parent 9afda27ab4
commit f24c836cbc
Signed by: Joren
GPG Key ID: 280E33DFBC0F1B55
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
main.go
View File

@ -1,13 +1,12 @@
package main
import (
"database/sql"
"fmt"
"log"
"os"
"os/signal"
"syscall"
"database/sql"
"github.com/BurntSushi/toml"
"github.com/bwmarrin/discordgo"
@ -38,16 +37,13 @@ type Database struct {
Password string `toml:"password"`
}
func loadConfig(filename string) (Config, error) {
var config Config
_, err := toml.DecodeFile(filename, &config)
return config, err
}
func connectDb(config Config) {
var err error
func connectDb(config Config) (*sql.DB, error) {
connectionString := fmt.Sprintf(
"host=%s port=%d user=%s password=%s dbname=%s sslmode=disable",
config.Database.Host,
@ -60,14 +56,17 @@ func connectDb(config Config) {
db, err := sql.Open("postgres", connectionString)
if err != nil {
log.Fatalf("Error connecting to the database: %v", err)
return nil, fmt.Errorf("error connecting to the database: %v", err)
}
defer db.Close()
err = db.Ping()
if err != nil {
log.Fatalf("Error pinging the database: %v", err)
db.Close()
return nil, fmt.Errorf("error pinging the database: %v", err)
}
log.Println("Successfully connected to the database.")
return db, nil
}
func init() {
@ -84,9 +83,11 @@ func init() {
return
}
connectDb(config)
db, err = connectDb(config)
if err != nil {
log.Println("Error initializing db connection:", err)
return
}
}
func main() {
@ -111,5 +112,9 @@ func main() {
log.Println("Gracefully shutting down.")
client.Close()
if db != nil {
db.Close()
}
}