Add Product entity, migration, seeder, and JSON data for multi-tenant support
This commit is contained in:
parent
cc66c9c251
commit
dae2c0daa9
|
|
@ -0,0 +1,16 @@
|
|||
package entities
|
||||
|
||||
import (
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
type Product struct {
|
||||
ID uuid.UUID `gorm:"type:uuid;primary_key;default:uuid_generate_v4()" json:"id"`
|
||||
Name string `gorm:"type:varchar(100);not null" json:"name"`
|
||||
Price float64 `gorm:"type:numeric;not null" json:"price"`
|
||||
|
||||
TenantID uuid.UUID `gorm:"type:uuid;null;index" json:"tenant_id"`
|
||||
Tenant Tenant `gorm:"foreignKey:TenantID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;" json:"tenant"`
|
||||
|
||||
Timestamp
|
||||
}
|
||||
|
|
@ -8,5 +8,7 @@ type Tenant struct {
|
|||
ID uuid.UUID `gorm:"type:uuid;primary_key;default:uuid_generate_v4()" json:"id"`
|
||||
Name string `gorm:"type:varchar(100);not null" json:"name"`
|
||||
|
||||
Products []Product `gorm:"foreignKey:TenantID" json:"products"`
|
||||
Users []User `gorm:"foreignKey:TenantID" json:"users"`
|
||||
Timestamp
|
||||
}
|
||||
|
|
|
|||
|
|
@ -10,9 +10,24 @@ func Migrate(db *gorm.DB) error {
|
|||
&entities.Tenant{},
|
||||
&entities.User{},
|
||||
&entities.RefreshToken{},
|
||||
&entities.Product{},
|
||||
); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func MigrateFresh(db *gorm.DB) error {
|
||||
// Drop tables
|
||||
if err := db.Migrator().DropTable(
|
||||
&entities.Tenant{},
|
||||
&entities.User{},
|
||||
&entities.RefreshToken{},
|
||||
&entities.Product{},
|
||||
); err != nil {
|
||||
return err
|
||||
}
|
||||
// Migrate ulang
|
||||
return Migrate(db)
|
||||
}
|
||||
|
|
|
|||
|
|
@ -9,6 +9,7 @@ func Seeder(db *gorm.DB) error {
|
|||
seeders := []func(*gorm.DB) error{
|
||||
seeds.ListTenantSeeder,
|
||||
seeds.ListUserSeeder,
|
||||
seeds.ListProductSeeder,
|
||||
}
|
||||
|
||||
for _, seeder := range seeders {
|
||||
|
|
|
|||
|
|
@ -0,0 +1,42 @@
|
|||
[
|
||||
{
|
||||
"name": "MacBook Pro",
|
||||
"price": 10000,
|
||||
"tenant_name": "Tenant A"
|
||||
},
|
||||
{
|
||||
"name": "iPhone 13",
|
||||
"price": 5000,
|
||||
"tenant_name": "Tenant B"
|
||||
},
|
||||
{
|
||||
"name": "Laptop Pro",
|
||||
"price": 13000.99,
|
||||
"tenant_name": "Tenant A"
|
||||
},
|
||||
{
|
||||
"name": "iPhone 11",
|
||||
"price": 7000.99,
|
||||
"tenant_name": "Tenant B"
|
||||
},
|
||||
{
|
||||
"name": "MacBook Pro M1",
|
||||
"price": 10000.15,
|
||||
"tenant_name": "Tenant A"
|
||||
},
|
||||
{
|
||||
"name": "iPhone 10",
|
||||
"price": 5000.99,
|
||||
"tenant_name": "Tenant B"
|
||||
},
|
||||
{
|
||||
"name": "MacBook Pro M2",
|
||||
"price": 90000.99,
|
||||
"tenant_name": "Tenant A"
|
||||
},
|
||||
{
|
||||
"name": "iPhone 14",
|
||||
"price": 20000.15,
|
||||
"tenant_name": "Tenant B"
|
||||
}
|
||||
]
|
||||
|
|
@ -0,0 +1,109 @@
|
|||
package seeds
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"io"
|
||||
"os"
|
||||
|
||||
"github.com/Caknoooo/go-gin-clean-starter/database/entities"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type ProductSeed struct {
|
||||
Name string `json:"name"`
|
||||
Price float64 `json:"price"`
|
||||
TenantName string `json:"tenant_name"`
|
||||
}
|
||||
|
||||
func ListProductSeeder(db *gorm.DB) error {
|
||||
jsonFile, err := os.Open("./database/seeders/json/products.json")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
jsonData, err := io.ReadAll(jsonFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var listProduct []ProductSeed
|
||||
if err := json.Unmarshal(jsonData, &listProduct); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
hasTable := db.Migrator().HasTable(&entities.Product{})
|
||||
if !hasTable {
|
||||
if err := db.Migrator().CreateTable(&entities.Product{}); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
for _, data := range listProduct {
|
||||
var tenant entities.Tenant
|
||||
if err := db.Where("name = ?", data.TenantName).First(&tenant).Error; err != nil {
|
||||
return err // tenant tidak ditemukan
|
||||
}
|
||||
|
||||
var product entities.Product
|
||||
err := db.Where(&entities.Product{Name: data.Name}).First(&product).Error
|
||||
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return err
|
||||
}
|
||||
|
||||
isData := db.Find(&product, "name = ?", data.Name).RowsAffected
|
||||
if isData == 0 {
|
||||
newProduct := entities.Product{
|
||||
Name: data.Name,
|
||||
Price: data.Price,
|
||||
TenantID: tenant.ID,
|
||||
}
|
||||
if err := db.Create(&newProduct).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// func ListUserSeeder(db *gorm.DB) error {
|
||||
// jsonFile, err := os.Open("./database/seeders/json/users.json")
|
||||
// if err != nil {
|
||||
// return err
|
||||
// }
|
||||
|
||||
// jsonData, err := io.ReadAll(jsonFile)
|
||||
// if err != nil {
|
||||
// return err
|
||||
// }
|
||||
|
||||
// var listUser []entities.User
|
||||
// if err := json.Unmarshal(jsonData, &listUser); err != nil {
|
||||
// return err
|
||||
// }
|
||||
|
||||
// hasTable := db.Migrator().HasTable(&entities.User{})
|
||||
// if !hasTable {
|
||||
// if err := db.Migrator().CreateTable(&entities.User{}); err != nil {
|
||||
// return err
|
||||
// }
|
||||
// }
|
||||
|
||||
// for _, data := range listUser {
|
||||
// var user entities.User
|
||||
// err := db.Where(&entities.User{Email: data.Email}).First(&user).Error
|
||||
// if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
// return err
|
||||
// }
|
||||
|
||||
// isData := db.Find(&user, "email = ?", data.Email).RowsAffected
|
||||
// if isData == 0 {
|
||||
// if err := db.Create(&data).Error; err != nil {
|
||||
// return err
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
// return nil
|
||||
// }
|
||||
Loading…
Reference in New Issue