Re-implement seeding

Author: victorsteven

api/models/User.go

@@ -19,7 +19,6 @@ type User struct {
 	Password  string    `gorm:"size:100;not null;" json:"password"`
 	CreatedAt time.Time `gorm:"default:current_timestamp()" json:"created_at"`
 	UpdatedAt time.Time `gorm:"default:current_timestamp()" json:"updated_at"`
-	// Posts     []Post    `gorm:"foreignkey:AuthorID" json:"posts"`
 }
 
 func (u *User) BeforeSave() error {

api/seed/seeder.go

@@ -3,7 +3,7 @@ package seed
 import (
 	"log"
 
-	"github.com/victorsteven/fullstack/api/database"
+	"github.com/jinzhu/gorm"
 	"github.com/victorsteven/fullstack/api/models"
 )
 
@@ -14,28 +14,26 @@ var users = []models.User{
 		Password: "password",
 	},
 	models.User{
-		Nickname: "Steven victor",
-		Email:    "steven@gmail.com",
+		Nickname: "Martin Luther",
+		Email:    "luther@gmail.com",
 		Password: "password",
 	},
 }
 
 var posts = []models.Post{
 	models.Post{
-		Title:   "Title",
-		Content: "Hello world",
+		Title:   "Title 1",
+		Content: "Hello world 1",
+	},
+	models.Post{
+		Title:   "Title 2",
+		Content: "Hello world 2",
 	},
 }
 
-func Load() {
-	db, err := database.Connect()
+func Load(db *gorm.DB) {
 
-	if err != nil {
-		log.Fatalf("cannot connect to the database: %v", err)
-	}
-	defer db.Close()
-
-	err = db.Debug().DropTableIfExists(&models.Post{}, &models.User{}).Error
+	err := db.Debug().DropTableIfExists(&models.Post{}, &models.User{}).Error
 	if err != nil {
 		log.Fatalf("cannot drop table: %v", err)
 	}
@@ -54,7 +52,6 @@ func Load() {
 		if err != nil {
 			log.Fatalf("cannot seed users table: %v", err)
 		}
-
 		posts[i].AuthorID = users[i].ID
 
 		err = db.Debug().Model(&models.Post{}).Create(&posts[i]).Error

api/server.go

@@ -1,49 +1,20 @@
 package api
 
 import (
-	"fmt"
-	"log"
-	"os"
-
-	// "github.com/victorsteven/fullstack/api/router"
-	"github.com/joho/godotenv"
 	"github.com/victorsteven/fullstack/api/controllers"
+	"github.com/victorsteven/fullstack/api/seed"
+	"github.com/victorsteven/fullstack/config"
 )
 
 var server = controllers.Server{}
 
-// func refreshTable() error {
-// 	err := server.DB.Debug().DropTableIfExists(&models.User{}).Error
-// 	if err != nil {
-// 		return err
-// 	}
-// 	err = server.DB.Debug().AutoMigrate(&models.User{}).Error
-// 	if err != nil {
-// 		return err
-// 	}
-// 	log.Printf("Successfully refreshed table")
-// 	return nil
-// }
-
 func Run() {
 
-	// config.Load()
-	// seed.Load()
-	// fmt.Println("this is after seeding fresh")
-	fmt.Printf("running... at port %d\n\n", 8080)
-	// listen(8080)
-
-	var err error
-	err = godotenv.Load()
-	if err != nil {
-		log.Fatalf("Error getting env %v", err)
-	} else {
-		fmt.Println("We are getting the env values")
-	}
+	config.Load()
 
-	server.Initialize(os.Getenv("DB_DRIVER"), os.Getenv("DB_USER"), os.Getenv("DB_PASSWORD"), os.Getenv("DB_NAME"))
+	server.Initialize(config.DBDRIVER, config.DBUSER, config.DBPASSWORD, config.DBNAME)
 
-	// refreshTable()
+	seed.Load(server.DB)
 
 	server.Run(":8080")
 

config/config.go

@@ -10,32 +10,36 @@ import (
 )
 
 var (
-	PORT      = 0
-	SECRETKEY []byte
-	DBURL     = ""
-	DBDRIVER  = ""
+	PORT       = 0
+	SECRETKEY  []byte
+	DBURL      = ""
+	DBDRIVER   = ""
+	DBUSER     = ""
+	DBPASSWORD = ""
+	DBNAME     = ""
 )
 
 func Load() {
+
 	var err error
 	err = godotenv.Load()
 	if err != nil {
 		log.Fatalf("Error getting env %v", err)
+	} else {
+		fmt.Println("We are getting the env values")
 	}
 	PORT, err = strconv.Atoi(os.Getenv("API_PORT"))
-
 	if err != nil {
 		PORT = 8000
 	}
-
-	DBHOST := os.Getenv("DATABASE_HOST")
+	DBHOST := os.Getenv("DB_HOST")
 	if DBHOST == "" {
 		DBHOST = "127.0.0.1"
 	}
 	DBDRIVER = os.Getenv("DB_DRIVER")
-
-	DBURL = fmt.Sprintf("%s:%s@tcp(%s:3306)/%s?charset=utf8&parseTime=True&loc=Local", os.Getenv("DB_USER"), os.Getenv("DB_PASSWORD"), DBHOST, os.Getenv("DB_NAME"))
-
+	DBUSER = os.Getenv("DB_USER")
+	DBPASSWORD = os.Getenv("DB_PASSWORD")
+	DBNAME = os.Getenv("DB_NAME")
 	SECRETKEY = []byte(os.Getenv("API_SECRET"))
 
 }