From f32bf350abaa8d18a165bb02cbd345a8eac3a7de Mon Sep 17 00:00:00 2001
From: liwei <liwei1dao@gmail.com>
Date: Mon, 14 Nov 2022 16:01:05 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E8=A3=85=E5=A4=87=E9=99=84?=
 =?UTF-8?q?=E9=AD=94=E4=BB=A3=E7=A0=81=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/equipment/api_ench.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/modules/equipment/api_ench.go b/modules/equipment/api_ench.go
index 428a328ab..f8f7571fd 100644
--- a/modules/equipment/api_ench.go
+++ b/modules/equipment/api_ench.go
@@ -25,6 +25,7 @@ func (this *apiComp) Ench(session comm.IUserSession, req *pb.EquipmentEnchReq) (
 		conf       *cfg.GameEquipEnchantingData
 		hero       *pb.DBHero
 		equipments []*pb.DB_Equipment
+		need       []*cfg.Gameatn
 		AttrValue  int32
 		err        error
 	)
@@ -35,6 +36,17 @@ func (this *apiComp) Ench(session comm.IUserSession, req *pb.EquipmentEnchReq) (
 		code = pb.ErrorCode_ConfigNoFound
 		return
 	}
+	need = make([]*cfg.Gameatn, 0)
+	need = append(need, conf.Need...)
+	need = append(need, &cfg.Gameatn{
+		A: comm.ItemType,
+		T: req.Itemid,
+		N: 1,
+	})
+	if code = this.module.ConsumeRes(session, need, true); code != pb.ErrorCode_Success {
+		return
+	}
+
 	if equip, err = this.module.modelEquipment.QueryUserEquipmentsById(session.GetUserId(), req.Eid); err != nil {
 		this.module.Errorf("Equip reader uid:%s equipment:%s err:%v", session.GetUserId(), req.Eid, err)
 		code = pb.ErrorCode_SystemError
@@ -57,6 +69,7 @@ func (this *apiComp) Ench(session comm.IUserSession, req *pb.EquipmentEnchReq) (
 		return
 	}
 	if equip.HeroId != "" {
+		equipments = make([]*pb.DB_Equipment, 8)
 		if hero, code = this.module.ModuleHero.GetHeroByObjID(session.GetUserId(), equip.HeroId); code != pb.ErrorCode_Success {
 			this.module.Errorf("Upgrade code:%d", code)
 			return