kris/Go-Back-Skeleton
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

split DB connection string into separate env vars

Browse source
This commit is contained in:
dhax 2020-01-30 14:59:00 +01:00
parent c6b8aff2a9
commit 069d655d9c
4 changed files with 19 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

17
database/postgres.go
View file

@ -11,13 +11,20 @@ import (
// DBConn returns a postgres connection pool.
func DBConn() (*pg.DB, error) {
viper.SetDefault("db_network", "tcp")
viper.SetDefault("db_addr", "localhost:5432")
viper.SetDefault("db_user", "postgres")
viper.SetDefault("db_password", "postgres")
viper.SetDefault("db_database", "gobase")
opts, err := pg.ParseURL(viper.GetString("database_url"))
if err != nil {
return nil, err
}
db := pg.Connect(&pg.Options{
Network: viper.GetString("db_network"),
Addr: viper.GetString("db_addr"),
User: viper.GetString("db_user"),
Password: viper.GetString("db_password"),
Database: viper.GetString("db_database"),
})
db := pg.Connect(opts)
if err := checkConn(db); err != nil {
return nil, err
}