refactor(seeder): add a way to figure out how to use the specific bin…

…d vars
danvergara · May 8, 2021 · 9edad16 · 9edad16
1 parent 56284c8
commit 9edad16
Show file tree

Hide file tree

Showing 5 changed files with 39 additions and 7 deletions.
diff --git a/cmd/seeder/main.go b/cmd/seeder/main.go
@@ -33,7 +33,7 @@ func handleArgs() {
 			if err != nil {
 				log.Fatalf("Error opening DB: %v", err)
 			}
-			seeds.Execute(db, args[1:]...)
+			seeds.Execute(db, cfg.Driver(), args[1:]...)
 			os.Exit(0)
 		}
 	}

diff --git a/db/seeds/customer.go b/db/seeds/customer.go
@@ -9,8 +9,18 @@ import (
 // CustomerSeed seeds the database with customers.
 func (s Seed) CustomerSeed() {
 	for i := 0; i < 100; i++ {
+		var err error
+
 		// execute query.
-		_, err := s.db.Exec(`INSERT INTO customers(name, email) VALUES ($1, $2)`, faker.Name(), faker.Email())
+		switch s.driver {
+		case "postgres":
+			_, err = s.db.Exec(`INSERT INTO customers(name, email) VALUES ($1, $2)`, faker.Name(), faker.Email())
+		case "mysql":
+			_, err = s.db.Exec(`INSERT INTO customers(name, email) VALUES (?, ?)`, faker.Name(), faker.Email())
+		default:
+			log.Println("unsupported driver")
+		}
+
 		if err != nil {
 			log.Fatalf("error seeding customers: %v", err)
 		}

diff --git a/db/seeds/product.go b/db/seeds/product.go
@@ -10,8 +10,18 @@ import (
 // ProductSeed seeds product data.
 func (s Seed) ProductSeed() {
 	for i := 0; i < 100; i++ {
+		var err error
+
 		// execute query.
-		_, err := s.db.Exec(`INSERT INTO products(name, price) VALUES ($1, $2)`, faker.Word(), rand.Float32())
+		switch s.driver {
+		case "postgres":
+			_, err = s.db.Exec(`INSERT INTO products(name, price) VALUES ($1, $2)`, faker.Word(), rand.Float32())
+		case "mysql":
+			_, err = s.db.Exec(`INSERT INTO products(name, price) VALUES (?, ?)`, faker.Word(), rand.Float32())
+		default:
+			log.Println("unsupported driver")
+		}
+
 		if err != nil {
 			log.Fatalf("error seeding products: %v", err)
 		}

diff --git a/db/seeds/seeder.go b/db/seeds/seeder.go
@@ -14,13 +14,15 @@ import (
 
 // Seed type.
 type Seed struct {
-	db *sqlx.DB
+	db     *sqlx.DB
+	driver string
 }
 
 // Execute will executes the given seeder method.
-func Execute(db *sqlx.DB, seedMethodNames ...string) {
+func Execute(db *sqlx.DB, driver string, seedMethodNames ...string) {
 	s := Seed{
-		db: db,
+		db:     db,
+		driver: driver,
 	}
 
 	seedType := reflect.TypeOf(s)

diff --git a/db/seeds/user.go b/db/seeds/user.go
@@ -9,8 +9,18 @@ import (
 // UserSeed seeds the database with users.
 func (s Seed) UserSeed() {
 	for i := 0; i < 100; i++ {
+		var err error
+
 		// execute query.
-		_, err := s.db.Exec(`INSERT INTO users(username) VALUES ($1)`, faker.Name())
+		switch s.driver {
+		case "postgres":
+			_, err = s.db.Exec(`INSERT INTO users(username) VALUES ($1)`, faker.Name())
+		case "mysql":
+			_, err = s.db.Exec(`INSERT INTO users(username) VALUES (?)`, faker.Name())
+		default:
+			log.Println("unsupported driver")
+		}
+
 		if err != nil {
 			log.Fatalf("error seeding users: %v", err)
 		}