Add SeederEntry interface to dbo

Author: christianezeani

dbo/instance.go

@@ -52,6 +52,10 @@ func (s *dbInstance) AddSeeders(handlers ...SeederHandler) {
 	s.seeders.Add(handlers...)
 }
 
+func (s *dbInstance) AddSeederEntries(entries ...SeederEntry) {
+	s.seeders.AddEntries(entries...)
+}
+
 func (s *dbInstance) Migrate(seed ...bool) (err error) {
 	if err = s.migrations.Run(s.db); err != nil {
 		return

dbo/seeder.go

@@ -1,20 +1,43 @@
 package dbo
 
-import "gorm.io/gorm"
+import (
+	"strings"
+
+	"github.com/Cyberpull/gokit/fmt"
+	"gorm.io/gorm"
+)
 
 type SeederHandler func(db *gorm.DB) (err error)
 
+type SeederEntry interface {
+	Name() string
+	Handler(db *gorm.DB) (err error)
+}
+
 type dbSeeder struct {
 	opts     *Options
+	entries  []SeederEntry
 	handlers []SeederHandler
 }
 
-func (s *dbSeeder) Add(handlers ...SeederHandler) {
-	s.handlers = append(s.handlers, handlers...)
+func (x *dbSeeder) Add(handlers ...SeederHandler) {
+	x.handlers = append(x.handlers, handlers...)
+}
+
+func (x *dbSeeder) AddEntries(entries ...SeederEntry) {
+	x.entries = append(x.entries, entries...)
 }
 
-func (s *dbSeeder) Run(db *gorm.DB) (err error) {
-	for _, handler := range s.handlers {
+func (x *dbSeeder) Run(db *gorm.DB) (err error) {
+	for _, entry := range x.entries {
+		tx := NewSession(db)
+
+		if err = x.RunEntry(tx, entry); err != nil {
+			return
+		}
+	}
+
+	for _, handler := range x.handlers {
 		tx := NewSession(db)
 
 		if err = handler(tx); err != nil {
@@ -25,6 +48,22 @@ func (s *dbSeeder) Run(db *gorm.DB) (err error) {
 	return
 }
 
+func (x *dbSeeder) RunEntry(db *gorm.DB, entry SeederEntry) (err error) {
+	name := strings.TrimSpace(entry.Name())
+
+	fmt.Magenta.Printf("Seeding '%v'... ", name)
+
+	err = entry.Handler(db)
+
+	if err != nil {
+		return
+	}
+
+	fmt.Magenta.Println(" Completed!")
+
+	return
+}
+
 // ===================
 
 var Seeder dbSeeder
@@ -35,8 +74,9 @@ func newSeeder(opts *Options) *dbSeeder {
 	return s
 }
 
-func initSeeders(s *dbSeeder) {
-	s.handlers = make([]SeederHandler, 0)
+func initSeeders(x *dbSeeder) {
+	x.entries = []SeederEntry{}
+	x.handlers = []SeederHandler{}
 }
 
 func init() {