49 lines
889 B
Go
49 lines
889 B
Go
|
package postgresql
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
_ "github.com/lib/pq" // For postgres dialect
|
||
|
|
"github.com/logrusorgru/aurora"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
sqlxClient *sqlx.DB
|
||
|
|
)
|
||
|
|
|
||
|
|
// Connect to postgres database
|
||
|
|
func Connect(host, user, password, dbname, port, sslmode string) error {
|
||
|
|
// Connect string
|
||
|
|
dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=UTC",
|
||
|
|
host, user, password, dbname, port, sslmode,
|
||
|
|
)
|
||
|
|
|
||
|
|
db, err := sqlx.Connect("postgres", dsn)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalln(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println(aurora.Green("*** CONNECTED TO POSTGRESQL - SQLX: " + dsn))
|
||
|
|
|
||
|
|
// Config connection pool
|
||
|
|
sqlDB := db.DB
|
||
|
|
sqlDB.SetMaxOpenConns(100)
|
||
|
|
sqlDB.SetMaxIdleConns(20)
|
||
|
|
sqlDB.SetConnMaxLifetime(time.Minute * 5)
|
||
|
|
|
||
|
|
// Assign client
|
||
|
|
sqlxClient = db
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetSqlxInstance ...
|
||
|
|
func GetSqlxInstance() *sqlx.DB {
|
||
|
|
return sqlxClient
|
||
|
|
}
|
||
|
|
|