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.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})