From ee9b19b98c8d4e361982acee534b17e3d8d4ba1c Mon Sep 17 00:00:00 2001
From: hahwu <31872165+hahwu@users.noreply.github.com>
Date: Thu, 19 Mar 2026 09:41:40 +0800
Subject: [PATCH] =?UTF-8?q?=E9=82=80=E8=AF=B7=E5=A5=BD=E5=8F=8B,=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E5=B7=B2=E6=88=90=E5=8A=9F=E5=88=A4=E6=96=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/server/game/message_handler.go   | 5 ++++-
 src/server/game/mod/invite/invite.go | 6 +++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/server/game/message_handler.go b/src/server/game/message_handler.go
index df0c54b2..1779a079 100644
--- a/src/server/game/message_handler.go
+++ b/src/server/game/message_handler.go
@@ -150,7 +150,10 @@ func (p *Player) handle(m *msg.Msg) error {
 		p.PlayMod.save()
 	case msg.HANDLE_TYPE_INVITE_FRIEND: // 邀请好友
 		InviteMod := p.PlayMod.getInviteMod()
-		InviteMod.AddInvite(m.From)
+		err := InviteMod.AddInvite(m.From)
+		if err != nil {
+			return err
+		}
 		p.AddLog(m.From, friend.LOG_TYPE_FRIEND_INVITE, "", m.SendT)
 		p.PlayMod.save()
 		p.PushClientRes(InviteMod.NotifySuccess())
diff --git a/src/server/game/mod/invite/invite.go b/src/server/game/mod/invite/invite.go
index 4db9e05a..1cc26505 100644
--- a/src/server/game/mod/invite/invite.go
+++ b/src/server/game/mod/invite/invite.go
@@ -19,8 +19,12 @@ func (i *InviteMod) InitData() {
 	}
 }
 
-func (i *InviteMod) AddInvite(id int) {
+func (i *InviteMod) AddInvite(id int) error {
+	if _, exists := i.InviteList[id]; exists {
+		return fmt.Errorf("invite already exists")
+	}
 	i.InviteList[id] = struct{}{}
+	return nil
 }
 
 func (i *InviteMod) GetReward(Index int) ([]*item.Item, error) {