45 lines
801 B
Go
45 lines
801 B
Go
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/joho/godotenv"
|
|
_ "github.com/lib/pq"
|
|
"github.com/vinovest/sqlx"
|
|
)
|
|
|
|
var db *sqlx.DB = nil
|
|
|
|
func init() {
|
|
if err := godotenv.Load(); err != nil {
|
|
log.Println("Error loading .env file. If you use docker, then you can ignore this.")
|
|
}
|
|
|
|
if db == nil {
|
|
connectPsql()
|
|
}
|
|
}
|
|
|
|
func getDSN() string {
|
|
user := os.Getenv("PSQL_USER")
|
|
password := os.Getenv("PSQL_PASS")
|
|
database := os.Getenv("PSQL_NAME")
|
|
host, exists := os.LookupEnv("PSQL_HOST")
|
|
if !exists {
|
|
host = "localhost"
|
|
}
|
|
return fmt.Sprintf("postgresql://%s:%s@%s/%s?sslmode=disable", user, password, host, database)
|
|
}
|
|
func connectPsql() {
|
|
var err error
|
|
db, err = sqlx.Connect("postgres", getDSN())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
func ClosePsql() error {
|
|
return db.Close()
|
|
}
|