fix return data in Connect function

This commit is contained in:
namhq1989 2022-08-26 15:17:25 +07:00
parent 8abf9ac5ed
commit 8da21b1024
1 changed files with 3 additions and 3 deletions

View File

@ -24,7 +24,7 @@ type Config struct {
} }
// Connect ... // Connect ...
func Connect(cfg Config, server string) *sql.DB { func Connect(cfg Config, server string) (*sql.DB, error) {
uri := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=%s", uri := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=%s",
cfg.Host, cfg.Port, cfg.User, cfg.Password, cfg.DBName, cfg.SSLMode) cfg.Host, cfg.Port, cfg.User, cfg.Password, cfg.DBName, cfg.SSLMode)
@ -41,7 +41,7 @@ func Connect(cfg Config, server string) *sql.DB {
Message: err.Error(), Message: err.Error(),
Data: cfg, Data: cfg,
}) })
panic(err) return nil, err
} }
// config // config
@ -66,5 +66,5 @@ func Connect(cfg Config, server string) *sql.DB {
fmt.Printf("⚡️[postgres]: connected to %s:%d \n", cfg.Host, cfg.Port) fmt.Printf("⚡️[postgres]: connected to %s:%d \n", cfg.Host, cfg.Port)
return db return db, nil
} }