Skip to content

Commit

Permalink
Merge pull request #24 from RECYTHNG/develop
Browse files Browse the repository at this point in the history 
feat(seed): add back dummy data
  • Loading branch information
@sawalreverr
sawalreverr authored Jun 26, 2024
2 parents 4c2d3f4 + c0d0f2b commit 9cb0b0e
Show file tree
Hide file tree
Showing 6 changed files with 156 additions and 150 deletions.
71 changes: 36 additions & 35 deletions internal/database/seedArticle.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package database

import (
"fmt"
"log"
"math/rand"
"time"

Expand All @@ -10,41 +11,41 @@ import (
)

func (m *mysqlDatabase) InitArticle() {
// if err := m.GetDB().Migrator().DropTable(&art.ArticleCategories{}); err != nil {
// return
// }
// if err := m.GetDB().Migrator().DropTable(&art.ArticleSection{}); err != nil {
// return
// }
// if err := m.GetDB().Migrator().DropTable(&art.Article{}); err != nil {
// return
// }

// if err := m.GetDB().AutoMigrate(&art.ArticleCategories{}); err != nil {
// return
// }
// if err := m.GetDB().AutoMigrate(&art.ArticleSection{}); err != nil {
// return
// }
// if err := m.GetDB().AutoMigrate(&art.Article{}); err != nil {
// return
// }

// articles, articleSections, articleCategories := generateArticle()

// for _, article := range articles {
// m.GetDB().FirstOrCreate(&article, article)
// }

// for _, articleSection := range articleSections {
// m.GetDB().FirstOrCreate(&articleSection, articleSection)
// }

// for _, articleCategory := range articleCategories {
// m.GetDB().FirstOrCreate(&articleCategory, articleCategory)
// }

// log.Println("Article data added!")
if err := m.GetDB().Migrator().DropTable(&art.ArticleCategories{}); err != nil {
return
}
if err := m.GetDB().Migrator().DropTable(&art.ArticleSection{}); err != nil {
return
}
if err := m.GetDB().Migrator().DropTable(&art.Article{}); err != nil {
return
}

if err := m.GetDB().AutoMigrate(&art.ArticleCategories{}); err != nil {
return
}
if err := m.GetDB().AutoMigrate(&art.ArticleSection{}); err != nil {
return
}
if err := m.GetDB().AutoMigrate(&art.Article{}); err != nil {
return
}

articles, articleSections, articleCategories := generateArticle()

for _, article := range articles {
m.GetDB().FirstOrCreate(&article, article)
}

for _, articleSection := range articleSections {
m.GetDB().FirstOrCreate(&articleSection, articleSection)
}

for _, articleCategory := range articleCategories {
m.GetDB().FirstOrCreate(&articleCategory, articleCategory)
}

log.Println("Article data added!")
}

var wasteCategories = []struct {
Expand Down
41 changes: 21 additions & 20 deletions internal/database/seedComment.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package database

import (
"fmt"
"log"
"math/rand"
"time"

Expand All @@ -11,31 +12,31 @@ import (
)

func (m *mysqlDatabase) InitComment() {
// if err := m.GetDB().Migrator().DropTable(&art.ArticleComment{}); err != nil {
// return
// }
// if err := m.GetDB().Migrator().DropTable(&vid.Comment{}); err != nil {
// return
// }
if err := m.GetDB().Migrator().DropTable(&art.ArticleComment{}); err != nil {
return
}
if err := m.GetDB().Migrator().DropTable(&vid.Comment{}); err != nil {
return
}

// if err := m.GetDB().AutoMigrate(&art.ArticleComment{}); err != nil {
// return
// }
// if err := m.GetDB().AutoMigrate(&vid.Comment{}); err != nil {
// return
// }
if err := m.GetDB().AutoMigrate(&art.ArticleComment{}); err != nil {
return
}
if err := m.GetDB().AutoMigrate(&vid.Comment{}); err != nil {
return
}

// articleComments, videoComments := generateComment()
articleComments, videoComments := generateComment()

// for _, articleComment := range articleComments {
// m.GetDB().FirstOrCreate(&articleComment, articleComment)
// }
for _, articleComment := range articleComments {
m.GetDB().FirstOrCreate(&articleComment, articleComment)
}

// for _, videoComment := range videoComments {
// m.GetDB().FirstOrCreate(&videoComment, videoComment)
// }
for _, videoComment := range videoComments {
m.GetDB().FirstOrCreate(&videoComment, videoComment)
}

// log.Println("Comments data added!")
log.Println("Comments data added!")
}

func randomUserID() string {
Expand Down
66 changes: 33 additions & 33 deletions internal/database/seedReport.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,39 +11,39 @@ import (
)

func (m *mysqlDatabase) InitReport() {
// if err := m.GetDB().Migrator().DropTable(&rpt.ReportImage{}); err != nil {
// return
// }
// if err := m.GetDB().Migrator().DropTable(&rpt.ReportWasteMaterial{}); err != nil {
// return
// }
// if err := m.GetDB().Migrator().DropTable(&rpt.Report{}); err != nil {
// return
// }

// if err := m.GetDB().AutoMigrate(&rpt.ReportImage{}); err != nil {
// return
// }
// if err := m.GetDB().AutoMigrate(&rpt.ReportWasteMaterial{}); err != nil {
// return
// }
// if err := m.GetDB().AutoMigrate(&rpt.Report{}); err != nil {
// return
// }

// reports, reportWasteMaterials, reportImages := generateReports()

// for _, report := range reports {
// m.GetDB().FirstOrCreate(&report, report)
// }

// for _, reportWasteMaterial := range reportWasteMaterials {
// m.GetDB().FirstOrCreate(&reportWasteMaterial, reportWasteMaterial)
// }

// for _, reportImage := range reportImages {
// m.GetDB().FirstOrCreate(&reportImage, reportImage)
// }
if err := m.GetDB().Migrator().DropTable(&rpt.ReportImage{}); err != nil {
return
}
if err := m.GetDB().Migrator().DropTable(&rpt.ReportWasteMaterial{}); err != nil {
return
}
if err := m.GetDB().Migrator().DropTable(&rpt.Report{}); err != nil {
return
}

if err := m.GetDB().AutoMigrate(&rpt.ReportImage{}); err != nil {
return
}
if err := m.GetDB().AutoMigrate(&rpt.ReportWasteMaterial{}); err != nil {
return
}
if err := m.GetDB().AutoMigrate(&rpt.Report{}); err != nil {
return
}

reports, reportWasteMaterials, reportImages := generateReports()

for _, report := range reports {
m.GetDB().FirstOrCreate(&report, report)
}

for _, reportWasteMaterial := range reportWasteMaterials {
m.GetDB().FirstOrCreate(&reportWasteMaterial, reportWasteMaterial)
}

for _, reportImage := range reportImages {
m.GetDB().FirstOrCreate(&reportImage, reportImage)
}
}

var wasteMaterials = []struct {
Expand Down
51 changes: 26 additions & 25 deletions internal/database/seedTask.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package database

import (
"fmt"
"log"
"math/rand"
"time"

Expand All @@ -10,31 +11,31 @@ import (
)

func (m *mysqlDatabase) InitTasks() {
// if err := m.GetDB().Migrator().DropTable(&taskEntity.TaskStep{}); err != nil {
// return
// }
// if err := m.GetDB().Migrator().DropTable(&taskEntity.TaskChallenge{}); err != nil {
// return
// }

// if err := m.GetDB().AutoMigrate(&taskEntity.TaskStep{}); err != nil {
// return
// }
// if err := m.GetDB().AutoMigrate(&taskEntity.TaskChallenge{}); err != nil {
// return
// }

// taskChallenges, taskSteps := generateTask()

// for _, taskChallenge := range taskChallenges {
// m.GetDB().FirstOrCreate(&taskChallenge, taskChallenge)
// }

// for _, taskStep := range taskSteps {
// m.GetDB().FirstOrCreate(&taskStep, taskStep)
// }

// log.Println("Task Challenge data added!")
if err := m.GetDB().Migrator().DropTable(&taskEntity.TaskStep{}); err != nil {
return
}
if err := m.GetDB().Migrator().DropTable(&taskEntity.TaskChallenge{}); err != nil {
return
}

if err := m.GetDB().AutoMigrate(&taskEntity.TaskStep{}); err != nil {
return
}
if err := m.GetDB().AutoMigrate(&taskEntity.TaskChallenge{}); err != nil {
return
}

taskChallenges, taskSteps := generateTask()

for _, taskChallenge := range taskChallenges {
m.GetDB().FirstOrCreate(&taskChallenge, taskChallenge)
}

for _, taskStep := range taskSteps {
m.GetDB().FirstOrCreate(&taskStep, taskStep)
}

log.Println("Task Challenge data added!")
}

func generateTask() ([]taskEntity.TaskChallenge, []taskEntity.TaskStep) {
Expand Down
26 changes: 14 additions & 12 deletions internal/database/seedUser.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,12 @@ package database

import (
"fmt"
"log"
"math/rand"
"time"

"github.com/brianvoe/gofakeit/v6"
"github.com/sawalreverr/recything/internal/helper"
userEntity "github.com/sawalreverr/recything/internal/user"
)

Expand Down Expand Up @@ -84,20 +86,20 @@ func generateUser(password string) []userEntity.User {
}

func (m *mysqlDatabase) InitUser() {
// if err := m.GetDB().Migrator().DropTable(&userEntity.User{}); err != nil {
// return
// }
if err := m.GetDB().Migrator().DropTable(&userEntity.User{}); err != nil {
return
}

// if err := m.GetDB().AutoMigrate(&userEntity.User{}); err != nil {
// return
// }
if err := m.GetDB().AutoMigrate(&userEntity.User{}); err != nil {
return
}

// hashed, _ := helper.GenerateHash("password@123")
// users := generateUser(hashed)
hashed, _ := helper.GenerateHash("password@123")
users := generateUser(hashed)

// for _, user := range users {
// m.GetDB().FirstOrCreate(&user, user)
// }
for _, user := range users {
m.GetDB().FirstOrCreate(&user, user)
}

// log.Println("User data added!")
log.Println("User data added!")
}
51 changes: 26 additions & 25 deletions internal/database/seedVideo.go
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package database

import (
"log"
"math/rand"
"time"

Expand All @@ -9,31 +10,31 @@ import (
)

func (m *mysqlDatabase) InitVideos() {
// if err := m.GetDB().Migrator().DropTable(&videoEntity.VideoCategory{}); err != nil {
// return
// }
// if err := m.GetDB().Migrator().DropTable(&videoEntity.Video{}); err != nil {
// return
// }

// if err := m.GetDB().AutoMigrate(&videoEntity.VideoCategory{}); err != nil {
// return
// }
// if err := m.GetDB().AutoMigrate(&videoEntity.Video{}); err != nil {
// return
// }

// videos, videoCategories := generateVideo()

// for _, video := range videos {
// m.GetDB().FirstOrCreate(&video, video)
// }

// for _, videoCategory := range videoCategories {
// m.GetDB().FirstOrCreate(&videoCategory, videoCategory)
// }

// log.Println("Video data added!")
if err := m.GetDB().Migrator().DropTable(&videoEntity.VideoCategory{}); err != nil {
return
}
if err := m.GetDB().Migrator().DropTable(&videoEntity.Video{}); err != nil {
return
}

if err := m.GetDB().AutoMigrate(&videoEntity.VideoCategory{}); err != nil {
return
}
if err := m.GetDB().AutoMigrate(&videoEntity.Video{}); err != nil {
return
}

videos, videoCategories := generateVideo()

for _, video := range videos {
m.GetDB().FirstOrCreate(&video, video)
}

for _, videoCategory := range videoCategories {
m.GetDB().FirstOrCreate(&videoCategory, videoCategory)
}

log.Println("Video data added!")
}

func generateVideo() ([]videoEntity.Video, []videoEntity.VideoCategory) {
Expand Down

0 comments on commit 9cb0b0e

Please sign in to comment.