📚 gorm-seeder - Awesome Go Library for Database
Simple database seeder for Gorm ORM.
🏷️ Database
📂 Data stores with expiring records, in-memory distributed data stores, or in-memory subsets of file-based databases.
⭐ 13 stars
Detailed Description of gorm-seeder
Simple Gorm seeder
Description
Simple Gorm seeder package
Download
go get -u github.com/kachit/gorm-seeder
Usage
package main
import (
"fmt"
"github.com/kachit/gorm-seeder"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"time"
)
//Init model
type User struct {
Id uint64 `json:"id" gorm:"primaryKey"`
Name string `json:"name"`
Email string `json:"email"`
Password string `json:"password"`
CreatedAt time.Time `json:"created_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) (err error) {
u.CreatedAt = time.Now().UTC()
return
}
// TableName overrides
func (User) TableName() string {
return "users"
}
//Create new seeder instance of gorm_seeder.SeederInterface
type UsersSeeder struct {
gorm_seeder.SeederAbstract
}
func NewUsersSeeder(cfg gorm_seeder.SeederConfiguration) UsersSeeder {
return UsersSeeder{gorm_seeder.NewSeederAbstract(cfg)}
}
//Implement Seed method
func (s *UsersSeeder) Seed(db *gorm.DB) error {
var users []User
for i := 0; i < s.Configuration.Rows; i++ {
indexStr := fmt.Sprint(i)
user := User{
Name: "Name LastName" + indexStr,
Email: "foo" + indexStr + "@bar.gov",
Password: "password-hash-string" + indexStr,
}
users = append(users, user)
}
return db.CreateInBatches(users, s.Configuration.Rows).Error
}
//Implement Clear method
func (s *UsersSeeder) Clear(db *gorm.DB) error {
entity := User{}
return s.SeederAbstract.Delete(db, entity.TableName())
}
func main(){
//Init DB connection
db, _ := gorm.Open(postgres.New(postgres.Config{
DSN: "DSN connection string",
}))
//Build seeders stack
usersSeeder := NewUsersSeeder(gorm_seeder.SeederConfiguration{Rows: 10})
seedersStack := gorm_seeder.NewSeedersStack(db)
seedersStack.AddSeeder(&usersSeeder)
//Apply seed
err := seedersStack.Seed()
fmt.Println(err)
//Apply clear
err = seedersStack.Clear()
fmt.Println(err)
}