From 83ce81edcfb6671a7e09116633cedcb1e8b92e2b Mon Sep 17 00:00:00 2001
From: Ben Toogood <bentoogood@gmail.com>
Date: Wed, 27 Jan 2021 13:30:50 +0000
Subject: [PATCH] Fix DB connections in multiple services

---
 chats/main.go   | 3 +++
 groups/main.go  | 2 --
 invites/main.go | 3 +++
 streams/main.go | 3 +++
 users/main.go   | 3 +++
 5 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/chats/main.go b/chats/main.go
index 65809d8..88173b9 100644
--- a/chats/main.go
+++ b/chats/main.go
@@ -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})
diff --git a/groups/main.go b/groups/main.go
index a4cbd28..7675716 100644
--- a/groups/main.go
+++ b/groups/main.go
@@ -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)
 	}
diff --git a/invites/main.go b/invites/main.go
index b219769..40c914e 100644
--- a/invites/main.go
+++ b/invites/main.go
@@ -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})
diff --git a/streams/main.go b/streams/main.go
index 4dfd700..121d896 100644
--- a/streams/main.go
+++ b/streams/main.go
@@ -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})
diff --git a/users/main.go b/users/main.go
index 94c43b0..da34440 100644
--- a/users/main.go
+++ b/users/main.go
@@ -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})