move seed file to db package

Author: adamdevigili

api/pkg/db/init.go

@@ -93,7 +93,7 @@ func initPlayersTable(db *gorm.DB) {
 	log.Info("Populating players database with initial values..")
 
 	db.AutoMigrate(&models.Player{})
-	for _, p := range models.GenerateSeedPlayers() {
+	for _, p := range generateSeedPlayers() {
 		if err := InsertPlayer(db, p); err != nil {
 			log.Warn(err)
 		}
@@ -104,7 +104,7 @@ func initSportsTable(db *gorm.DB) {
 	log.Info("Populating sports database with initial values..")
 
 	db.AutoMigrate(&models.Sport{})
-	for _, s := range models.InitialSports {
+	for _, s := range initialSports {
 		if err := InsertSport(db, &s); err != nil {
 			log.Warn(err)
 		}

api/pkg/models/seed.go renamed to api/pkg/db/seed.go

@@ -1,45 +1,46 @@
-package models
+package db
 
 import (
 	"fmt"
 
+	"github.com/adamdevigili/skillbased/api/pkg/models"
 	"github.com/pioz/faker"
 	"github.com/rs/xid"
 )
 
 var (
-	ultimateFrisbee = Sport{
-		Base: Base{
+	ultimateFrisbee = models.Sport{
+		Base: models.Base{
 			Name: "Ultimate Frisbee",
 			ID:   xid.New(),
 		},
-		SkillWeights: SkillWeightMap{
+		SkillWeights: models.SkillWeightMap{
 			"handling": 0.9,
 			"speed":    0.8,
 			"stamina":  0.8,
 			"height":   0.4,
 		},
 	}
 
-	football = Sport{
-		Base: Base{
+	football = models.Sport{
+		Base: models.Base{
 			Name: "Football",
 			ID:   xid.New(),
 		},
-		SkillWeights: SkillWeightMap{
+		SkillWeights: models.SkillWeightMap{
 			"strength": 0.7,
 			"speed":    0.8,
 			"stamina":  0.8,
 			"agility":  0.5,
 		},
 	}
 
-	basketball = Sport{
-		Base: Base{
+	basketball = models.Sport{
+		Base: models.Base{
 			Name: "Basketball",
 			ID:   xid.New(),
 		},
-		SkillWeights: SkillWeightMap{
+		SkillWeights: models.SkillWeightMap{
 			"shooting": 0.9,
 			"speed":    0.6,
 			"stamina":  0.8,
@@ -48,7 +49,7 @@ var (
 		},
 	}
 
-	InitialSports = []Sport{
+	initialSports = []models.Sport{
 		ultimateFrisbee,
 		basketball,
 		football,
@@ -57,24 +58,24 @@ var (
 
 var playersToGenerate = 30
 
-func GenerateSeedPlayers() []*Player {
+func generateSeedPlayers() []*models.Player {
 	faker.SetSeed(623)
-	players := make([]*Player, playersToGenerate)
+	players := make([]*models.Player, playersToGenerate)
 
 	for i := range players {
 		fn, ln := faker.FirstName(), faker.LastName()
-		p := &Player{
+		p := &models.Player{
 			FirstName: fn,
 			LastName:  ln,
-			Base: Base{
+			Base: models.Base{
 				Name: fmt.Sprintf("%s %s", fn, ln),
 				ID:   xid.New(),
 			},
 			PowerScores: make(map[string]int),
 			IsSeed:      true,
 		}
 
-		for _, s := range skillsList {
+		for _, s := range models.SkillsList {
 			p.PowerScores[s] = faker.IntInRange(1, 10)
 		}
 

api/pkg/models/base.go

@@ -14,7 +14,7 @@ type Base struct {
 	UpdatedAt *time.Time `json:"updated_at,omitempty"`
 }
 
-var skillsList = []string{
+var SkillsList = []string{
 	"handling",
 	"power",
 	"speed",