105 lines
5.1 KiB
Go
105 lines
5.1 KiB
Go
package models
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ProjectProductTemplate struct {
|
|
ID uint `gorm:"primaryKey" json:"id"`
|
|
ProjectTypeCode string `gorm:"size:64;not null;index;uniqueIndex:idx_project_product_template" json:"projectTypeCode"`
|
|
ProductCode string `gorm:"size:64;not null;index;uniqueIndex:idx_project_product_template" json:"productCode"`
|
|
Quantity *int `json:"quantity"`
|
|
QuantityRule string `gorm:"size:32;not null;default:'exact'" json:"quantityRule"`
|
|
IsOptional bool `gorm:"not null;default:false" json:"isOptional"`
|
|
Notes string `gorm:"size:1024" json:"notes"`
|
|
Sort int `gorm:"not null;default:0;index" json:"sort"`
|
|
CreatedAt int64 `gorm:"not null" json:"created_at"`
|
|
UpdatedAt int64 `gorm:"not null" json:"updated_at"`
|
|
}
|
|
|
|
func (ProjectProductTemplate) TableName() string {
|
|
return "project_product_templates"
|
|
}
|
|
|
|
func (p *ProjectProductTemplate) BeforeCreate(tx *gorm.DB) (err error) {
|
|
now := time.Now().UnixMilli()
|
|
p.ProjectTypeCode = strings.TrimSpace(strings.ToLower(p.ProjectTypeCode))
|
|
p.ProductCode = normalizeProductCodeValue(p.ProductCode)
|
|
p.QuantityRule = normalizeQuantityRuleValue(p.QuantityRule)
|
|
p.Notes = strings.TrimSpace(p.Notes)
|
|
p.CreatedAt = now
|
|
p.UpdatedAt = now
|
|
return nil
|
|
}
|
|
|
|
func (p *ProjectProductTemplate) BeforeUpdate(tx *gorm.DB) (err error) {
|
|
p.ProjectTypeCode = strings.TrimSpace(strings.ToLower(p.ProjectTypeCode))
|
|
p.ProductCode = normalizeProductCodeValue(p.ProductCode)
|
|
p.QuantityRule = normalizeQuantityRuleValue(p.QuantityRule)
|
|
p.Notes = strings.TrimSpace(p.Notes)
|
|
p.UpdatedAt = time.Now().UnixMilli()
|
|
return nil
|
|
}
|
|
|
|
func normalizeQuantityRuleValue(rule string) string {
|
|
value := strings.TrimSpace(strings.ToLower(rule))
|
|
if value == "" {
|
|
return "exact"
|
|
}
|
|
return value
|
|
}
|
|
|
|
func intPtr(value int) *int {
|
|
return &value
|
|
}
|
|
|
|
func EnsureDefaultProjectProductTemplates(db *gorm.DB) error {
|
|
defaults := []ProjectProductTemplate{
|
|
{ProjectTypeCode: "heartrate", ProductCode: "charging_case_large", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: false, Notes: "可收纳20个手环", Sort: 10},
|
|
{ProjectTypeCode: "heartrate", ProductCode: "charging_case_small", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: true, Notes: "可选，可收纳10个手环", Sort: 20},
|
|
{ProjectTypeCode: "heartrate", ProductCode: "collection_band", Quantity: nil, QuantityRule: "flexible", IsOptional: false, Notes: "数量按班级规模或实际采购数量配置", Sort: 30},
|
|
{ProjectTypeCode: "heartrate", ProductCode: "control_tablet", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: false, Notes: "", Sort: 40},
|
|
{ProjectTypeCode: "heartrate", ProductCode: "collection_gateway", Quantity: intPtr(1), QuantityRule: "minimum", IsOptional: false, Notes: "至少1个", Sort: 50},
|
|
{ProjectTypeCode: "heartrate", ProductCode: "display_stand_screen", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: true, Notes: "可选，用于投屏展示", Sort: 60},
|
|
|
|
{ProjectTypeCode: "light", ProductCode: "charging_case_large", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: false, Notes: "可收纳20个手环", Sort: 110},
|
|
{ProjectTypeCode: "light", ProductCode: "charging_case_small", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: true, Notes: "可选，可收纳10个手环", Sort: 120},
|
|
{ProjectTypeCode: "light", ProductCode: "collection_band", Quantity: nil, QuantityRule: "flexible", IsOptional: false, Notes: "数量按班级规模或实际采购数量配置", Sort: 130},
|
|
{ProjectTypeCode: "light", ProductCode: "control_tablet", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: false, Notes: "", Sort: 140},
|
|
{ProjectTypeCode: "light", ProductCode: "collection_gateway", Quantity: intPtr(1), QuantityRule: "minimum", IsOptional: false, Notes: "至少1个", Sort: 150},
|
|
{ProjectTypeCode: "light", ProductCode: "display_stand_screen", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: true, Notes: "可选，用于投屏展示", Sort: 160},
|
|
{ProjectTypeCode: "light", ProductCode: "led_light_strip", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: false, Notes: "心肺耐力测试附加LED灯带", Sort: 170},
|
|
|
|
{ProjectTypeCode: "run50", ProductCode: "control_tablet", Quantity: intPtr(1), QuantityRule: "exact", IsOptional: false, Notes: "", Sort: 210},
|
|
{ProjectTypeCode: "run50", ProductCode: "collection_band", Quantity: nil, QuantityRule: "flexible", IsOptional: false, Notes: "数量按测试规模配置", Sort: 220},
|
|
}
|
|
|
|
for _, item := range defaults {
|
|
var existing ProjectProductTemplate
|
|
err := db.Where("project_type_code = ? AND product_code = ?", item.ProjectTypeCode, item.ProductCode).First(&existing).Error
|
|
switch {
|
|
case err == nil:
|
|
existing.Quantity = item.Quantity
|
|
existing.QuantityRule = item.QuantityRule
|
|
existing.IsOptional = item.IsOptional
|
|
existing.Notes = item.Notes
|
|
existing.Sort = item.Sort
|
|
if err := db.Save(&existing).Error; err != nil {
|
|
return err
|
|
}
|
|
case errors.Is(err, gorm.ErrRecordNotFound):
|
|
if err := db.Create(&item).Error; err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|