Skip to content

Commit 21b4b35

Browse files
Add SeederEntry interface to dbo
1 parent c081786 commit 21b4b35

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

dbo/seeder.go

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,11 @@ import (
99

1010
type SeederHandler func(db *gorm.DB) (err error)
1111

12+
type DBSeeder interface {
13+
AddEntries(entries ...SeederEntry)
14+
Run(db *gorm.DB) (err error)
15+
}
16+
1217
type SeederEntry interface {
1318
Name() string
1419
Handler(db *gorm.DB) (err error)
@@ -68,6 +73,10 @@ func (x *dbSeeder) RunEntry(db *gorm.DB, entry SeederEntry) (err error) {
6873

6974
var Seeder dbSeeder
7075

76+
func NewSeeder() DBSeeder {
77+
return newSeeder(Seeder.opts)
78+
}
79+
7180
func newSeeder(opts *Options) *dbSeeder {
7281
s := &dbSeeder{opts: opts}
7382
initSeeders(s)

0 commit comments

Comments
 (0)