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) {
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 {