29 lines
548 B
Go
29 lines
548 B
Go
package database
|
|
|
|
import (
|
|
"os"
|
|
|
|
"go-microservices/services/user-service/internal/models"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func Init() (*gorm.DB, error) {
|
|
dsn := os.Getenv("DATABASE_URL")
|
|
if dsn == "" {
|
|
dsn = "host=localhost user=postgres password=postgres dbname=postgres port=5432 sslmode=disable TimeZone=UTC"
|
|
}
|
|
|
|
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := db.AutoMigrate(&models.User{}, &models.Client{}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db, nil
|
|
}
|