connection.go 756 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. package db
  2. import (
  3. "log"
  4. "gorm.io/driver/mysql"
  5. "gorm.io/gorm"
  6. )
  7. // Config to connect mySQL
  8. var DSN = "root:password@tcp(mysql:3306)/mydbusers?charset=utf8mb4&parseTime=True&loc=Local"
  9. var DB *gorm.DB
  10. func DBConnection() {
  11. var err error
  12. DB, err = gorm.Open(mysql.Open(DSN), &gorm.Config{})
  13. // if err != nil {
  14. // log.Fatal(error)
  15. // } else {
  16. // log.Println("DB connected")
  17. // }
  18. if err != nil {
  19. log.Printf("Error al conectar a la base de datos: %v", err)
  20. return // Manejar el error adecuadamente
  21. }
  22. log.Println("DB connected")
  23. }
  24. // AutoMigrate realiza la migración de los modelos
  25. func AutoMigrate(models ...interface{}) {
  26. err := DB.AutoMigrate(models...)
  27. if err != nil {
  28. log.Fatal("Error al realizar la migration:", err)
  29. }
  30. }