Thomas Peetz
61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/viper"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
"log/slog"
|
|
)
|
|
|
|
func NewDB() (*gorm.DB, error) {
|
|
db_type := ServerConfig.DB.Type
|
|
slog.Info("using database", "type", db_type)
|
|
|
|
switch db_type {
|
|
case "postgres":
|
|
dsn := fmt.Sprintf(
|
|
"host=%s user=%s password=%s dbname=%s port=%d sslmode=disable",
|
|
viper.GetString("db.host"),
|
|
viper.GetString("db.user"),
|
|
viper.GetString("db.password"),
|
|
viper.GetString("db.name"),
|
|
viper.GetInt("db.port"),
|
|
)
|
|
|
|
return gorm.Open(postgres.Open(dsn), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Info),
|
|
})
|
|
case "sqlite":
|
|
return gorm.Open(sqlite.Open(viper.GetString("db.path")), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Info),
|
|
})
|
|
default:
|
|
return nil, fmt.Errorf("unknown db.type=%s", db_type)
|
|
}
|
|
// Configure GORM with custom settings
|
|
// The logger helps with debugging SQL queries during development
|
|
// if err != nil {
|
|
// log.Fatal("Failed to connect to database:", err)
|
|
// }
|
|
|
|
// // Get the underlying SQL database connection for pool configuration
|
|
// sqlDB, err := db.DB()
|
|
// if err != nil {
|
|
// log.Fatal("Failed to get database instance:", err)
|
|
// }
|
|
|
|
// // Configure connection pool settings
|
|
// // MaxIdleConns: number of connections kept open when idle
|
|
// sqlDB.SetMaxIdleConns(10)
|
|
// // MaxOpenConns: maximum total connections to the database
|
|
// sqlDB.SetMaxOpenConns(100)
|
|
// // ConnMaxLifetime: how long a connection can be reused
|
|
// sqlDB.SetConnMaxLifetime(time.Hour)
|
|
|
|
// slog.Info("Database connected successfully")
|
|
|
|
}
|