package res_m import ( "git.hilo.cn/hilo-common/domain" "git.hilo.cn/hilo-common/resource/mysql" "gorm.io/gorm" "hilo-user/_const/enum/res_e" "hilo-user/myerr" "hilo-user/myerr/bizerr" ) type ResProperty struct { mysql.Entity *domain.Model `gorm:"-"` Name mysql.Str PicUrl mysql.Str EffectUrl mysql.Str } type ResPropertyDiamond struct { mysql.Entity *domain.Model `gorm:"-"` ResPropertyId mysql.ID DiamondNum mysql.Num Second mysql.Num Status mysql.UserYesNo } type ResPropertyAvatar struct { mysql.Entity *domain.Model `gorm:"-"` ResPropertyId mysql.ID Type res_e.ResPropertyAvatarType SendUserId mysql.ID ReceiverUserId mysql.ID Area int } func InitResPropertyDiamond(model *domain.Model, resPropertyId mysql.ID, diamondNum mysql.Num, second mysql.Num) *ResPropertyDiamond { return &ResPropertyDiamond{ Model: model, ResPropertyId: resPropertyId, DiamondNum: diamondNum, Second: second, Status: mysql.NOUSER, } } //id获取头饰,不存在则抛异常 func GetPropertyById(model *domain.Model, id mysql.ID) (*ResProperty, error) { resProperty := ResProperty{} if err := model.Db.Model(&ResProperty{}).First(&resProperty, id).Error; err != nil { return nil, myerr.WrapErr(err) } else { resProperty.Model = model return &resProperty, nil } } func GetResPropertyDiamond(model *domain.Model, resPropertyDiamondId mysql.ID) (*ResPropertyDiamond, error) { resPropertyDiamond := ResPropertyDiamond{} if err := model.Db.Model(&ResPropertyDiamond{}).First(&resPropertyDiamond, resPropertyDiamondId).Error; err != nil { return nil, myerr.WrapErr(err) } if resPropertyDiamond.Status == mysql.NOUSER { return nil, bizerr.ResPropertyDiamondNoUse } return &resPropertyDiamond, nil } func GetResPropertyDiamondByPropertyIdOrNil(model *domain.Model, resPropertyId mysql.ID) (*ResPropertyDiamond, error) { resPropertyDiamond := ResPropertyDiamond{} if err := model.Db.Model(&ResPropertyDiamond{}).Where(&ResPropertyDiamond{ ResPropertyId: resPropertyId, }).First(&resPropertyDiamond).Error; err != nil { if err == gorm.ErrRecordNotFound { return nil, nil } else { return nil, myerr.WrapErr(err) } } resPropertyDiamond.Model = model return &resPropertyDiamond, nil } //设置成未使用 func (resPropertyDiamond *ResPropertyDiamond) SetUser() *ResPropertyDiamond { resPropertyDiamond.Status = mysql.USER return resPropertyDiamond } //设置成未使用 func (resPropertyDiamond *ResPropertyDiamond) SetNoUser() *ResPropertyDiamond { resPropertyDiamond.Status = mysql.NOUSER return resPropertyDiamond } func (resPropertyDiamond *ResPropertyDiamond) SetDiamondNum(diamondNum uint32) *ResPropertyDiamond { resPropertyDiamond.DiamondNum = diamondNum return resPropertyDiamond } func (resPropertyDiamond *ResPropertyDiamond) SetSecond(second uint32) *ResPropertyDiamond { resPropertyDiamond.Second = second return resPropertyDiamond } func AddProperty(model *domain.Model, p *ResProperty) error { return model.Db.Create(p).Error } func EditProperty(model *domain.Model, propertyId uint64, name, picUrl, effectUrl string) error { activityConfig := ResProperty{Entity: mysql.Entity{ID: propertyId}} err := model.Db.First(&activityConfig).Error if err != nil { return err } activityConfig.Name = name activityConfig.PicUrl = picUrl activityConfig.EffectUrl = effectUrl return model.Db.Save(&activityConfig).Error } func (p *ResProperty) Get(db *gorm.DB) error { return db.Where(p).First(p).Error } func (p *ResProperty) GetAll(db *gorm.DB) (map[uint64]ResProperty, error) { rows := make([]ResProperty, 0) if err := db.Where(p).Find(&rows).Error; err != nil { return nil, err } result := make(map[uint64]ResProperty, 0) for _, i := range rows { result[i.ID] = i } return result, nil } func (p *ResPropertyAvatar) GetAll(db *gorm.DB) (map[uint64]ResPropertyAvatar, error) { rows := make([]ResPropertyAvatar, 0) if err := db.Where(p).Find(&rows).Error; err != nil { return nil, myerr.WrapErr(err) } result := make(map[uint64]ResPropertyAvatar, 0) for _, i := range rows { result[i.ResPropertyId] = i } return result, nil }