Skip to content

Commit 0f4a8c0

Browse files
committed
Implement database connection and error handling and also add database migrations for Event and UserPreference models
1 parent d56090c commit 0f4a8c0

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

database/database.go

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,27 @@
11
package database
22

33
import (
4+
"fmt"
5+
6+
"github.com/Jetlum/WalletAlertService/models"
47
"gorm.io/driver/postgres"
58
"gorm.io/gorm"
69
)
710

8-
// Exported DB variable
911
var DB *gorm.DB
1012

1113
func InitDB(dsn string) error {
1214
var err error
1315
DB, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
1416
if err != nil {
15-
return err
17+
return fmt.Errorf("failed to connect to database: %w", err)
1618
}
19+
20+
// Run migrations
21+
err = DB.AutoMigrate(&models.Event{}, &models.UserPreference{})
22+
if err != nil {
23+
return fmt.Errorf("failed to run migrations: %w", err)
24+
}
25+
1726
return nil
1827
}

0 commit comments

Comments
 (0)