50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var DB *gorm.DB
|
|
var SatuDataDB *gorm.DB
|
|
var err error
|
|
|
|
func ConnectDB() {
|
|
// host := os.Getenv("DB_HOST")
|
|
// userName := os.Getenv("DB_USERNAME")
|
|
// password := os.Getenv("DB_PASSWORD")
|
|
// dbName := os.Getenv("DB_NAME")
|
|
// port := os.Getenv("DB_PORT")
|
|
|
|
hostSatuData := os.Getenv("SATUDATA_HOST")
|
|
userNameSatuData := os.Getenv("SATUDATA_USERNAME")
|
|
passwordSatuData := os.Getenv("SATUDATA_PASSWORD")
|
|
dbNameSatuData := os.Getenv("SATUDATA_NAME")
|
|
portSatuData := os.Getenv("SATUDATA_PORT")
|
|
|
|
// dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=Asia/Jakarta", host, userName, password, dbName, port)
|
|
// if dsn == "" {
|
|
// log.Fatal("POSTGRES_DSN environment variable not set")
|
|
// }
|
|
|
|
satuData := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=Asia/Jakarta", hostSatuData, userNameSatuData, passwordSatuData, dbNameSatuData, portSatuData)
|
|
|
|
// DB, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
|
|
// if err != nil {
|
|
// log.Fatal("Failed to connect to database: ", err)
|
|
// } else {
|
|
// log.Println("Successfully connected to the database")
|
|
// }
|
|
|
|
SatuDataDB, err = gorm.Open(postgres.Open(satuData), &gorm.Config{})
|
|
if err != nil {
|
|
log.Fatal("Failed to connect to Satu Data database: ", err)
|
|
} else {
|
|
log.Println("Successfully connected to the database")
|
|
}
|
|
}
|