varnitgoyal95/houston-be
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TP-0000 | SlackBot Changes (#16)

Browse Source 
* TP-0000 | SlackBot Changes

* TP-000 | Confluene Page column

* TP-0000 | comments

* TP-0000 | properties fix

* TP-0000 | func refactor

* TP-0000 | minor changes

Co-authored-by: ravi chandora <ravi.chandora@navi.com>
This commit is contained in:
Abhijeet Gupta
2023-04-15 19:24:02 +05:30
committed by GitHub Enterprise
parent 753228eb77
commit 96ce2fff26
29 changed files with 501 additions and 325 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
pkg/postgres/config.go
View File

@@ -2,6 +2,7 @@ package postgres
import (
"os"
"time"
"go.uber.org/zap"
"gorm.io/driver/postgres"
@@ -13,26 +14,37 @@ type Client struct {
gormClient *gorm.DB
}
func NewGormClient(dsn string, logger *zap.Logger) *gorm.DB {
// todo: set the connection configs
func NewGormClient(dsn string, connMaxIdleTime string, connMaxLifetime string, setMaxIdleConns int, setMaxOpenConns int, logger *zap.Logger) *gorm.DB {
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
logger.Error("database connection failed", zap.Error(err))
os.Exit(1)
}
sqlDB, err := db.DB()
if err != nil {
logger.Fatal(err.Error())
}
connMaxIdleDuration, err := time.ParseDuration(connMaxIdleTime)
if err != nil {
logger.Fatal(err.Error())
}
connMaxLifetimeDuration, err := time.ParseDuration(connMaxLifetime)
if err != nil {
logger.Fatal(err.Error())
}
sqlDB.SetConnMaxIdleTime(time.Duration(connMaxIdleDuration.Seconds()))
sqlDB.SetConnMaxLifetime(time.Duration(connMaxLifetimeDuration.Seconds()))
sqlDB.SetMaxIdleConns(setMaxIdleConns)
sqlDB.SetMaxOpenConns(setMaxOpenConns)
err = sqlDB.Ping()
if err != nil {
logger.Fatal(err.Error())
}
logger.Info("database connection established")
return db
}
func PQConnection(logger *zap.Logger) *gorm.DB {
dsn := ""
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
logger.Error("database connection failed", zap.Error(err))
os.Exit(1)
}
logger.Info("database connection successful")
return db
}