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

enhance PostgreSQL connection handling with connection pool settings

Browse Source
This commit is contained in:
pigwin
2025-01-04 17:07:41 +00:00
parent 6db5b12d7b
commit c12ec02270
1 changed files with 31 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
config/db.go
View File

@@ -9,36 +9,45 @@ 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
} }
fmt.Println("Connection to PostgreSQL opened successfully!") fmt.Println("Connection to PostgreSQL opened successfully!")
// Ping database to verify connection // Set the maximum number of open connections to 20
err = db.Ping() db.SetMaxOpenConns(20)
fmt.Println(err) // Set the maximum number of idle connections to 10
if err != nil { db.SetMaxIdleConns(10)
return nil, err
}
log.Println("Connected to PostgreSQL!") // Set the maximum connection lifetime to 1 hour
db.SetConnMaxLifetime(1 * time.Hour)
return db, nil // Ping database to verify connection
err = db.Ping()
fmt.Println(err)
if err != nil {
return nil, err
}
log.Println("Connected to PostgreSQL!")
return db, nil
} }
func DisconnectFromPostgreSQL(db *sql.DB) error { func DisconnectFromPostgreSQL(db *sql.DB) error {