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)
    }
}