Adiciona possibilidade de uso com MongoDB

.env.sample

@@ -2,7 +2,8 @@
 DATABASE_URL=postgres://minhareceita:minhareceita@postgres:5432/minhareceita?sslmode=disable
 
 # Tests
-TEST_DATABASE_URL=postgres://minhareceita:minhareceita@postgres_test:5555/minhareceita?sslmode=disable
+TEST_POSTGRES_URL=postgres://minhareceita:minhareceita@postgres_test:5555/minhareceita?sslmode=disable
+TEST_MONGODB_URL=mongodb://minhareceita:minhareceita@mongo_test:27017/minhareceita?authSource=admin
 
 # Mirror
 ENDPOINT_URL=https://nyc3.digitaloceanspaces.com

cmd/api.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"fmt"
 	"os"
+	"strings"
 
 	"github.com/cuducos/minha-receita/api"
 	"github.com/cuducos/minha-receita/db"
@@ -53,18 +54,30 @@ var apiCmd = &cobra.Command{
 				return err
 			}
 		}
-		pg, err := db.NewPostgreSQL(u, postgresSchema, nr)
-		if err != nil {
-			return err
-		}
-		defer pg.Close()
 		if port == "" {
 			port = os.Getenv("PORT")
 		}
 		if port == "" {
 			port = defaultPort
 		}
-		api.Serve(&pg, port, nr)
+		if strings.HasPrefix(u, "mongodb://") {
+			mdb, _ := db.NewMongoDB(u)
+			if err != nil {
+				return err
+			}
+			defer mdb.Close()
+			api.Serve(&mdb, port, nr)
+		} else if strings.HasPrefix(u, "postgres://") || strings.HasPrefix(u, "postgresql://") {
+			pg, err := db.NewPostgreSQL(u, postgresSchema, nr)
+			if err != nil {
+				return err
+			}
+			defer pg.Close()
+			api.Serve(&pg, port, nr)
+		} else {
+			return fmt.Errorf("unsupported database URI, must start with postgres:// or mongodb://")
+		}
+
 		return nil
 	},
 }

cmd/cmd.go

@@ -4,10 +4,10 @@
 import (
 	"fmt"
 	"os"
-
-	"github.com/spf13/cobra"
+	"strings"
 
 	"github.com/cuducos/minha-receita/db"
+	"github.com/spf13/cobra"
 )
 
 const (
@@ -47,7 +47,7 @@
 	}
 	u := os.Getenv("DATABASE_URL")
 	if u == "" {
-		return "", fmt.Errorf("could not find a database URI, set the DATABASE_URL environment variable with the credentials for a PostgreSQL database")
+		return "", fmt.Errorf("could not find a database URI, set the DATABASE_URL environment variable with the credentials for a PostgreSQL or MongoDB database")
 	}
 	return u, nil
 }
@@ -60,18 +60,34 @@
 
 var createCmd = &cobra.Command{
 	Use:   "create",
-	Short: "Creates the required tables in PostgreSQL",
+	Short: "Creates the required tables in the database",
 	RunE: func(_ *cobra.Command, _ []string) error {
 		u, err := loadDatabaseURI()
 		if err != nil {
 			return err
 		}
-		pg, err := db.NewPostgreSQL(u, postgresSchema, nil)
-		if err != nil {
+		uri := os.Getenv("DATABASE_URL")
+		if strings.HasPrefix(uri, "mongodb://") {
+			mdb, err := db.NewMongoDB(uri)
+			if err != nil {
+				return err
+			}
+			err = mdb.CreateCollection()
+			if err != nil {
+				return err
+			}
+			defer mdb.Close()
 			return err
+		} else if strings.HasPrefix(uri, "postgres://") || strings.HasPrefix(uri, "postgresql://") {
+			pg, err := db.NewPostgreSQL(u, postgresSchema, nil)
+			if err != nil {
+				return err
+			}
+			defer pg.Close()
+			return pg.CreateTable()
+		} else {
+			return fmt.Errorf("A URL não contém 'mongodb' nem 'postgres'")
 		}
-		defer pg.Close()
-		return pg.CreateTable()
 	},
 }
 
@@ -83,12 +99,27 @@
 		if err != nil {
 			return err
 		}
-		pg, err := db.NewPostgreSQL(u, postgresSchema, nil)
-		if err != nil {
+		uri := os.Getenv("DATABASE_URL")
+		if strings.HasPrefix(uri, "mongodb://") {
+			mdb, err := db.NewMongoDB(uri)
+			if err != nil {
+				return err
+			}
+			err = mdb.DropCollection()
+			if err != nil {
+				return err
+			}
 			return err
+		} else if strings.HasPrefix(uri, "postgres://") || strings.HasPrefix(uri, "postgresql://") {
+			pg, err := db.NewPostgreSQL(u, postgresSchema, nil)
+			if err != nil {
+				return err
+			}
+			defer pg.Close()
+			return pg.DropTable()
+		} else {
+			return fmt.Errorf("A URL não contém 'mongodb' nem 'postgres'")
 		}
-		defer pg.Close()
-		return pg.DropTable()
 	},
 }
 
@@ -98,7 +129,7 @@
 }
 
 func addDatabase(c *cobra.Command) *cobra.Command {
-	c.Flags().StringVarP(&databaseURI, "database-uri", "u", "", "PostgreSQL URI (default DATABASE_URL environment variable)")
+	c.Flags().StringVarP(&databaseURI, "database-uri", "u", "", "Database URI (default DATABASE_URL environment variable)")
 	c.Flags().StringVarP(&postgresSchema, "postgres-schema", "s", "public", "PostgreSQL schema")
 	return c
 }

cmd/transform.go

@@ -1,6 +1,9 @@
 package cmd
 
 import (
+	"fmt"
+	"strings"
+
 	"github.com/cuducos/minha-receita/db"
 	"github.com/cuducos/minha-receita/transform"
 	"github.com/spf13/cobra"
@@ -35,21 +38,42 @@
 		if err != nil {
 			return err
 		}
-		pg, err := db.NewPostgreSQL(u, postgresSchema, nil)
-		if err != nil {
-			return err
-		}
-		defer pg.Close()
 
-		if cleanUp {
-			if err := pg.DropTable(); err != nil {
+		if strings.HasPrefix(u, "mongodb://") {
+			mdb, err := db.NewMongoDB(u)
+			if err != nil {
 				return err
 			}
-			if err := pg.CreateTable(); err != nil {
+			if cleanUp {
+				err = mdb.DropCollection()
+				if err != nil {
+					return err
+				}
+				err = mdb.CreateCollection()
+				if err != nil {
+					return err
+				}
+			}
+			return transform.Transform(dir, &mdb, maxParallelDBQueries, batchSize, !noPrivacy)
+		} else if strings.HasPrefix(u, "postgres://") || strings.HasPrefix(u, "postgresql://") {
+			pg, err := db.NewPostgreSQL(u, postgresSchema, nil)
+			if err != nil {
 				return err
 			}
+			defer pg.Close()
+			if cleanUp {
+				if err := pg.DropTable(); err != nil {
+					return err
+				}
+				if err := pg.CreateTable(); err != nil {
+					return err
+				}
+			}
+			return transform.Transform(dir, &pg, maxParallelDBQueries, batchSize, !noPrivacy)
+		} else {
+			return fmt.Errorf("A URL não contém 'mongodb' nem 'postgres'")
 		}
-		return transform.Transform(dir, &pg, maxParallelDBQueries, batchSize, !noPrivacy)
+
 	},
 }