feat: extend staging seed data

This commit is contained in:
2026-01-26 09:04:29 +08:00
parent 9607b914c9
commit 619e17fc15
3 changed files with 252 additions and 13 deletions

View File

@@ -67,7 +67,6 @@ func Serve(cmd *cobra.Command, args []string) error {
}
// Buyer
buyer := &models.User{
Username: "test",
Phone: "13800138000",
@@ -82,8 +81,22 @@ func Serve(cmd *cobra.Command, args []string) error {
buyer, _ = models.UserQuery.WithContext(ctx).Where(models.UserQuery.Phone.Eq("13800138000")).First()
}
// 2. Tenant
// Superadmin
superAdmin := &models.User{
Username: "superadmin",
Phone: "13800009999",
Nickname: "平台管理员",
Avatar: "https://api.dicebear.com/7.x/avataaars/svg?seed=Admin",
Balance: 0,
Status: consts.UserStatusVerified,
Roles: types.Array[consts.Role]{consts.RoleSuperAdmin},
}
if err := models.UserQuery.WithContext(ctx).Create(superAdmin); err != nil {
fmt.Printf("Create superadmin failed: %v\n", err)
superAdmin, _ = models.UserQuery.WithContext(ctx).Where(models.UserQuery.Username.Eq("superadmin")).First()
}
// 2. Tenant
tenant := &models.Tenant{
UserID: creator.ID,
Name: "梅派艺术工作室",
@@ -96,6 +109,17 @@ func Serve(cmd *cobra.Command, args []string) error {
tenant, _ = models.TenantQuery.WithContext(ctx).Where(models.TenantQuery.UserID.Eq(creator.ID)).First()
}
// Tenant membership (buyer joins as member)
member := &models.TenantUser{
TenantID: tenant.ID,
UserID: buyer.ID,
Role: types.Array[consts.TenantUserRole]{consts.TenantUserRoleMember},
Status: consts.UserStatusVerified,
}
if err := models.TenantUserQuery.WithContext(ctx).Create(member); err != nil {
fmt.Printf("Create tenant member failed: %v\n", err)
}
// 3. Contents
titles := []string{
"《锁麟囊》春秋亭 (程砚秋)", "昆曲《牡丹亭》游园惊梦", "越剧《红楼梦》葬花",
@@ -109,6 +133,7 @@ func Serve(cmd *cobra.Command, args []string) error {
"https://images.unsplash.com/photo-1469571486292-0ba58a3f068b?ixlib=rb-1.2.1&auto=format&fit=crop&w=400&q=60",
}
var seededContents []*models.Content
for i, title := range titles {
price := int64((i % 3) * 1000) // 0, 10.00, 20.00
if i == 3 {
@@ -126,7 +151,9 @@ func Serve(cmd *cobra.Command, args []string) error {
Views: int32(rand.Intn(10000)),
Likes: int32(rand.Intn(1000)),
}
models.ContentQuery.WithContext(ctx).Create(c)
if err := models.ContentQuery.WithContext(ctx).Create(c); err == nil {
seededContents = append(seededContents, c)
}
// Price
models.ContentPriceQuery.WithContext(ctx).Create(&models.ContentPrice{
TenantID: tenant.ID,
@@ -163,30 +190,142 @@ func Serve(cmd *cobra.Command, args []string) error {
cp1 := &models.Coupon{
TenantID: tenant.ID,
Title: "新人立减券",
Type: "fix_amount",
Type: consts.CouponTypeFixAmount,
Value: 500, // 5.00
MinOrderAmount: 1000,
TotalQuantity: 100,
StartAt: time.Now().Add(-24 * time.Hour),
EndAt: time.Now().Add(30 * 24 * time.Hour),
}
models.CouponQuery.WithContext(ctx).Create(cp1)
if err := models.CouponQuery.WithContext(ctx).Create(cp1); err != nil {
fmt.Printf("Create coupon failed: %v\n", err)
}
// Give to buyer
models.UserCouponQuery.WithContext(ctx).Create(&models.UserCoupon{
UserID: buyer.ID,
CouponID: cp1.ID,
Status: "unused",
Status: consts.UserCouponStatusUnused,
})
// 5. Notifications
models.NotificationQuery.WithContext(ctx).Create(&models.Notification{
UserID: buyer.ID,
Type: "system",
Title: "欢迎注册",
Content: "欢迎来到曲韵平台!",
IsRead: false,
// 5. Orders & library access (first content)
if len(seededContents) > 0 {
content := seededContents[0]
order := &models.Order{
TenantID: tenant.ID,
UserID: buyer.ID,
Type: consts.OrderTypeContentPurchase,
Status: consts.OrderStatusPaid,
Currency: consts.CurrencyCNY,
AmountOriginal: 990,
AmountDiscount: 0,
AmountPaid: 990,
IdempotencyKey: uuid.NewString(),
PaidAt: time.Now().Add(-2 * time.Hour),
}
if err := models.OrderQuery.WithContext(ctx).Create(order); err != nil {
fmt.Printf("Create order failed: %v\n", err)
} else {
models.OrderItemQuery.WithContext(ctx).Create(&models.OrderItem{
TenantID: tenant.ID,
UserID: buyer.ID,
OrderID: order.ID,
ContentID: content.ID,
ContentUserID: creator.ID,
AmountPaid: order.AmountPaid,
})
models.ContentAccessQuery.WithContext(ctx).Create(&models.ContentAccess{
TenantID: tenant.ID,
UserID: buyer.ID,
ContentID: content.ID,
OrderID: order.ID,
Status: consts.ContentAccessStatusActive,
})
}
}
// 6. Creator join request & invite
models.TenantJoinRequestQuery.WithContext(ctx).Create(&models.TenantJoinRequest{
TenantID: tenant.ID,
UserID: buyer.ID,
Status: "pending",
Reason: "申请加入租户用于创作",
})
models.TenantInviteQuery.WithContext(ctx).Create(&models.TenantInvite{
TenantID: tenant.ID,
UserID: creator.ID,
Code: "invite" + cast.ToString(rand.Intn(100000)),
Status: "active",
MaxUses: 5,
UsedCount: 0,
Remark: "staging seed invite",
ExpiresAt: time.Now().Add(7 * 24 * time.Hour),
})
// 7. Notifications & templates
models.NotificationQuery.WithContext(ctx).Create(&models.Notification{
UserID: buyer.ID,
TenantID: tenant.ID,
Type: string(consts.NotificationTypeSystem),
Title: "欢迎注册",
Content: "欢迎来到曲韵平台!",
IsRead: false,
})
models.NotificationTemplateQuery.WithContext(ctx).Create(&models.NotificationTemplate{
TenantID: 0,
Name: "订单支付通知",
Type: consts.NotificationTypeOrder,
Title: "订单支付成功",
Content: "您的订单已支付成功。",
IsActive: true,
})
models.NotificationTemplateQuery.WithContext(ctx).Create(&models.NotificationTemplate{
TenantID: 0,
Name: "内容审核通知",
Type: consts.NotificationTypeAudit,
Title: "内容审核通过",
Content: "您提交的内容已通过审核。",
IsActive: true,
})
// 8. System config
models.SystemConfigQuery.WithContext(ctx).Create(&models.SystemConfig{
ConfigKey: "site_name",
Value: types.JSON([]byte(`{"value":"曲韵平台"}`)),
Description: "站点名称",
})
// 9. Audit log
models.AuditLogQuery.WithContext(ctx).Create(&models.AuditLog{
TenantID: tenant.ID,
OperatorID: superAdmin.ID,
Action: "seed",
TargetID: fmt.Sprintf("tenant:%d", tenant.ID),
Detail: "staging seed data",
})
// 10. Content report
if len(seededContents) > 0 {
models.ContentReportQuery.WithContext(ctx).Create(&models.ContentReport{
TenantID: tenant.ID,
ContentID: seededContents[0].ID,
ReporterID: buyer.ID,
Reason: "spam",
Detail: "疑似广告内容",
Status: "pending",
})
}
// 11. Comments
if len(seededContents) > 0 {
models.CommentQuery.WithContext(ctx).Create(&models.Comment{
TenantID: tenant.ID,
UserID: buyer.ID,
ContentID: seededContents[0].ID,
Content: "好喜欢这段演出!",
Likes: 1,
})
}
fmt.Println("Seed done.")
return nil