dmitriygnatenko-v2/internal/services/db/db.go

package db

import (
	"database/sql"
	"time"

	"git.dmitriygnatenko.ru/dima/dmitriygnatenko-v2/internal/interfaces"
)

func Init(env interfaces.Env) (*sql.DB, error) {
	dataSource := "user=" + env.GetDBUser() +
		" password=" + env.GetDBPassword() +
		" dbname=" + env.GetDBName() +
		" host=" + env.GetDBHost() +
		" port=" + env.GetDBPort() +
		" sslmode=disable"

	db, err := sql.Open("postgres", dataSource)
	if err != nil {
		return nil, err
	}

	if env.GetDBMaxOpenConns() > 0 {
		db.SetMaxOpenConns(env.GetDBMaxOpenConns())
	}

	if env.GetDBMaxIdleConns() > 0 {
		db.SetMaxIdleConns(env.GetDBMaxIdleConns())
	}

	if env.GetDBMaxConnLifetime() > 0 {
		db.SetConnMaxLifetime(time.Second * time.Duration(env.GetDBMaxConnLifetime()))
	}

	if env.GetDBMaxIdleConnLifetime() > 0 {
		db.SetConnMaxIdleTime(time.Second * time.Duration(env.GetDBMaxIdleConnLifetime()))
	}

	if err = db.Ping(); err != nil {
		return nil, err
	}

	return db, nil
}