From 13cc4bf78b9a40a7d92ed17147a532bed5e72187 Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Wed, 20 Nov 2024 16:04:00 +0800
Subject: [PATCH] 1

---
 server/wheelserver/service/award.go |  6 ++++++
 server/wheelserver/vo/common.go     | 14 ++++++++++++++
 2 files changed, 20 insertions(+)

diff --git a/server/wheelserver/service/award.go b/server/wheelserver/service/award.go
index b938db09..47b90f54 100644
--- a/server/wheelserver/service/award.go
+++ b/server/wheelserver/service/award.go
@@ -4,6 +4,7 @@ import (
 	"f5"
 	"main/vo"
 	"main/model"
+	"main/mt"
 )
 
 type award struct {
@@ -16,6 +17,10 @@ func (this *award) unInit() {
 }
 
 func (this *award) AddItem(accountId string, itemId int32, itemNum int32, baseVo *vo.BaseVo) {
+	itemMeta := mt.Table.Item.GetById(int64(itemId))
+	if itemMeta == nil {
+		return
+	}
 	nowTime := int32(f5.GetApp().GetRealSeconds())
 	bagItem := new(model.Bag)
 	if err, found := bagItem.Find(accountId, itemId); err == nil {
@@ -29,5 +34,6 @@ func (this *award) AddItem(accountId string, itemId int32, itemNum int32, baseVo
 			bagItem.ModifyTime = nowTime
 			bagItem.Create()
 		}
+		baseVo.GetOrCreateAward().AddItem(itemId, itemNum)
 	}
 }
diff --git a/server/wheelserver/vo/common.go b/server/wheelserver/vo/common.go
index 26843ae7..346e0b40 100644
--- a/server/wheelserver/vo/common.go
+++ b/server/wheelserver/vo/common.go
@@ -29,3 +29,17 @@ func (this *Award) AddItem(itemId int32, itemNum int32) {
 	p := new(AwardItem)
 	q5.AppendSlice(&this.Items, p)
 }
+
+func (this *BaseVo) GetOrCreateAward() *Award {
+	if this.Award == nil {
+		this.Award = new(Award)
+	}
+	return this.Award
+}
+
+func (this *BaseVo) GetOrCreateSideEffect() *SideEffect {
+	if this.SideEffect == nil {
+		this.SideEffect = new(SideEffect)
+	}
+	return this.SideEffect
+}