update code

internal/database/database.go

@@ -6,6 +6,7 @@ import (
 	"log"
 	"os"
 	"strconv"
+	"strings"
 	"time"
 
 	_ "github.com/jackc/pgx/v5/stdlib"
@@ -14,14 +15,14 @@ import (
 	"gorm.io/gorm"
 )
 
-type Service interface {
+type Services interface {
 	Health() map[string]string
-	Close() error
-	GetDB() *gorm.DB
+	GetDB(string) *gorm.DB
 }
 
 type service struct {
-	db *gorm.DB
+	simrsDB    *gorm.DB
+	satuDataDB *gorm.DB
 }
 
 var (
@@ -30,10 +31,17 @@ var (
 	passwordSimrs = os.Getenv("SIMRS_STAG_PASS")
 	dbNameSimrs   = os.Getenv("SIMRS_STAG_NAME")
 	portSimrs     = os.Getenv("SIMRS_STAG_PORT")
-	dbInstance    *service
+
+	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")
+
+	dbInstance *service
 )
 
-func New() Service {
+func New() Services {
 	// Reuse Connection
 	if dbInstance != nil {
 		return dbInstance
@@ -42,12 +50,25 @@ func New() Service {
 
 	SimrsDB, err := gorm.Open(postgres.Open(simrs), &gorm.Config{})
 	if err != nil {
-		log.Fatal("Failed to connect to SIM database: ", err)
+		log.Fatal("Failed to connect to database SIMRS: ", err)
 	} else {
-		log.Println("Successfully connected to the database")
+		log.Println("Successfully connected to the database SIMRS")
 	}
+	satuData := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=Asia/Jakarta", hostSatuData, userNameSatuData, passwordSatuData, dbNameSatuData, portSatuData)
+
+	satuDataDB, err := gorm.Open(postgres.Open(satuData), &gorm.Config{})
+	if err != nil {
+		log.Fatal("Failed to connect to database SatuData: ", err)
+	} else {
+		log.Println("Successfully connected to the database SatuData")
+	}
+
+	mongoURI := fmt.Sprintf("mongodb://%s:%s@%s:%s/%s?authSource=admin",
+		user, pass, host, port, database)
+
 	dbInstance = &service{
-		db: SimrsDB,
+		simrsDB:    SimrsDB,
+		satuDataDB: satuDataDB,
 	}
 	return dbInstance
 }
@@ -61,7 +82,7 @@ func (s *service) Health() map[string]string {
 	stats := make(map[string]string)
 
 	// Ping the database using GORM
-	db, err := s.db.DB() // Retrieve the underlying sql.DB instance from GORM
+	db, err := s.simrsDB.DB() // Retrieve the underlying sql.DB instance from GORM
 	if err != nil {
 		stats["status"] = "down"
 		stats["error"] = fmt.Sprintf("failed to get sql.DB from GORM: %v", err)
@@ -115,23 +136,13 @@ func (s *service) Health() map[string]string {
 // It logs a message indicating the disconnection from the specific database.
 // If the connection is successfully closed, it returns nil.
 // If an error occurs while closing the connection, it returns the error.
-func (s *service) Close() error {
-	db, err := s.db.DB() // Retrieve the underlying sql.DB instance from GORM
-	if err != nil {
-		log.Printf("Failed to retrieve sql.DB from GORM: %v", err)
-		return err
-	}
 
-	err = db.Close()
-	if err != nil {
-		log.Printf("Error closing the database connection: %v", err)
-		return err
+func (s *service) GetDB(database string) *gorm.DB {
+	if strings.ToLower(database) == "simrs" {
+		return s.simrsDB
+	} else if strings.ToLower(database) == "satudata" {
+		return s.satuDataDB
 	}
-
-	log.Printf("Disconnected from database successfully")
+	log.Println("Database tidak ditemukan")
 	return nil
 }
-
-func (s *service) GetDB() *gorm.DB {
-	return s.db
-}