pigwin/ti1
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: improve connection string formatting and enhance logging in ConnectToPostgreSQL function

Browse Source
This commit is contained in:
pigwin
2025-01-07 18:54:49 +00:00
parent 97a6506a65
commit a2c1766dd1
1 changed files with 25 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
config/db.go
View File

@@ -10,41 +10,41 @@ import (
) )
func ConnectToPostgreSQL() (*sql.DB, error) { func ConnectToPostgreSQL() (*sql.DB, error) {
fmt.Println("Connecting to PostgreSQL...") fmt.Println("Connecting to PostgreSQL...")
config, err := LoadConfig("config/conf.json") config, err := LoadConfig("config/conf.json")
if err != nil { if err != nil {
return nil, err 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", 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) config.Database.Host, config.Database.Port, config.Database.User, config.Database.Password, config.Database.DBName, config.Database.SSLMode)
// Open connection to database // Open connection to database
db, err := sql.Open("postgres", connStr) db, err := sql.Open("postgres", connStr)
if err != nil { if err != nil {
return nil, err return nil, err
} }
// Set connection pool settings // Set connection pool settings
db.SetMaxOpenConns(25) // Maximum number of open connections to the database db.SetMaxOpenConns(25) // Maximum number of open connections to the database
db.SetMaxIdleConns(25) // Maximum number of connections in the idle connection pool 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 db.SetConnMaxLifetime(1 * time.Hour) // Maximum amount of time a connection may be reused
fmt.Println("Connection to PostgreSQL opened successfully :D") fmt.Println("Connection to PostgreSQL opened successfully :D")
// Ping database to verify connection // Ping database to verify connection
err = db.Ping() err = db.Ping()
fmt.Println(err) fmt.Println(err)
if err != nil { if err != nil {
return nil, err 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 { func DisconnectFromPostgreSQL(db *sql.DB) error {