43 lines
813 B
Go
43 lines
813 B
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
const (
|
|
host = "202.46.28.160"
|
|
port = 45432
|
|
user = "5803024008"
|
|
password = "pw5803024008"
|
|
dbname = "tgs01_5803024008"
|
|
)
|
|
|
|
// InitDB returns a database connection
|
|
func InitDB() (*sql.DB, error) {
|
|
// connection string
|
|
psqlconn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname)
|
|
|
|
// open database
|
|
db, err := sql.Open("postgres", psqlconn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// check db
|
|
err = db.Ping()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fmt.Println("Connected to database!")
|
|
return db, nil
|
|
}
|
|
|
|
func CheckError(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
} |