Fix DB connections in multiple services

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

View File

@@ -32,6 +32,9 @@ func main() {
if err != nil {
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
pb.RegisterChatsHandler(srv.Server(), &handler.Chats{DB: db, Time: time.Now})

View File

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

View File

@@ -30,6 +30,9 @@ func main() {
if err != nil {
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
pb.RegisterInvitesHandler(srv.Server(), &handler.Invites{DB: db})

View File

@@ -32,6 +32,9 @@ func main() {
if err != nil {
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
pb.RegisterStreamsHandler(srv.Server(), &handler.Streams{DB: db, Time: time.Now})

View File

@@ -32,6 +32,9 @@ func main() {
if err != nil {
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
pb.RegisterUsersHandler(srv.Server(), &handler.Users{DB: db, Time: time.Now})