diff --git a/config/db.go b/config/db.go index 762c46a..51b8ef1 100644 --- a/config/db.go +++ b/config/db.go @@ -10,41 +10,41 @@ import ( ) func ConnectToPostgreSQL() (*sql.DB, error) { - fmt.Println("Connecting to PostgreSQL...") - config, err := LoadConfig("config/conf.json") - if err != nil { - return nil, err - } + fmt.Println("Connecting to PostgreSQL...") + config, err := LoadConfig("config/conf.json") + if err != nil { + return nil, err + } - fmt.Println("Configuration loaded successfully!") + fmt.Println("Configuration loaded successfully!") - connStr := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=%s", - config.Database.Host, config.Database.Port, config.Database.User, config.Database.Password, config.Database.DBName, config.Database.SSLMode) + connStr := fmt.Sprintf("host=%s port=%s user='%s' password='%s' dbname='%s' sslmode=%s", + config.Database.Host, config.Database.Port, config.Database.User, config.Database.Password, config.Database.DBName, config.Database.SSLMode) - // Open connection to database - db, err := sql.Open("postgres", connStr) - if err != nil { - return nil, err - } + // Open connection to database + db, err := sql.Open("postgres", connStr) + if err != nil { + return nil, err + } - // Set connection pool settings - db.SetMaxOpenConns(25) // Maximum number of open connections to the database - db.SetMaxIdleConns(25) // Maximum number of connections in the idle connection pool - db.SetConnMaxLifetime(1 * time.Hour) // Maximum amount of time a connection may be reused + // Set connection pool settings + db.SetMaxOpenConns(25) // Maximum number of open connections to the database + db.SetMaxIdleConns(25) // Maximum number of connections in the idle connection pool + db.SetConnMaxLifetime(1 * time.Hour) // Maximum amount of time a connection may be reused fmt.Println("Connection to PostgreSQL opened successfully :D") - // Ping database to verify connection - err = db.Ping() + // Ping database to verify connection + err = db.Ping() - fmt.Println(err) - if err != nil { - return nil, err - } + fmt.Println(err) + if err != nil { + return nil, err + } - log.Println("Connected to PostgreSQL!") + log.Println("Connected to PostgreSQL!") - return db, nil + return db, nil } func DisconnectFromPostgreSQL(db *sql.DB) error {