totole/services
1
0
Fork 0
mirror of https://github.com/kevin-DL/services.git synced 2026-01-18 21:55:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix DB connections in multiple services

Browse Source
This commit is contained in:
Ben Toogood
2021-01-27 13:30:50 +00:00
parent c43ff6a7e4
commit 83ce81edcf
5 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
chats/main.go
View File

@@ -32,6 +32,9 @@ func main() {
if err != nil { if err != nil {
logger.Fatalf("Error connecting to database: %v", err) logger.Fatalf("Error connecting to database: %v", err)
} }
if err := db.AutoMigrate(&handler.Chat{}, &handler.Message{}); err != nil {
logger.Fatalf("Error migrating database: %v", err)
}
// Register handler // Register handler
pb.RegisterChatsHandler(srv.Server(), &handler.Chats{DB: db, Time: time.Now}) pb.RegisterChatsHandler(srv.Server(), &handler.Chats{DB: db, Time: time.Now})

2
groups/main.go
View File

@@ -30,8 +30,6 @@ func main() {
if err != nil { if err != nil {
logger.Fatalf("Error connecting to database: %v", err) logger.Fatalf("Error connecting to database: %v", err)
} }
// Migrate the database
if err := db.AutoMigrate(&handler.Group{}, &handler.Membership{}); err != nil { if err := db.AutoMigrate(&handler.Group{}, &handler.Membership{}); err != nil {
logger.Fatalf("Error migrating database: %v", err) logger.Fatalf("Error migrating database: %v", err)
} }

3
invites/main.go
View File

@@ -30,6 +30,9 @@ func main() {
if err != nil { if err != nil {
logger.Fatalf("Error connecting to database: %v", err) logger.Fatalf("Error connecting to database: %v", err)
} }
if err := db.AutoMigrate(&handler.Invite{}); err != nil {
logger.Fatalf("Error migrating database: %v", err)
}
// Register handler // Register handler
pb.RegisterInvitesHandler(srv.Server(), &handler.Invites{DB: db}) pb.RegisterInvitesHandler(srv.Server(), &handler.Invites{DB: db})

3
streams/main.go
View File

@@ -32,6 +32,9 @@ func main() {
if err != nil { if err != nil {
logger.Fatalf("Error connecting to database: %v", err) logger.Fatalf("Error connecting to database: %v", err)
} }
if err := db.AutoMigrate(&handler.Conversation{}, &handler.Message{}); err != nil {
logger.Fatalf("Error migrating database: %v", err)
}
// Register handler // Register handler
pb.RegisterStreamsHandler(srv.Server(), &handler.Streams{DB: db, Time: time.Now}) pb.RegisterStreamsHandler(srv.Server(), &handler.Streams{DB: db, Time: time.Now})

3
users/main.go
View File

@@ -32,6 +32,9 @@ func main() {
if err != nil { if err != nil {
logger.Fatalf("Error connecting to database: %v", err) logger.Fatalf("Error connecting to database: %v", err)
} }
if err := db.AutoMigrate(&handler.User{}, &handler.Token{}); err != nil {
logger.Fatalf("Error migrating database: %v", err)
}
// Register handler // Register handler
pb.RegisterUsersHandler(srv.Server(), &handler.Users{DB: db, Time: time.Now}) pb.RegisterUsersHandler(srv.Server(), &handler.Users{DB: db, Time: time.Now})