From 46e558d6fdf8a35f6d23d2a635ad2a13a11a5735 Mon Sep 17 00:00:00 2001 From: acgist <289547414@qq.com> Date: Tue, 10 Oct 2023 07:58:03 +0800 Subject: [PATCH] =?UTF-8?q?[*]=20=E6=97=A5=E5=B8=B8=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../signal/protocol/control/ControlWakeupProtocol.java | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/taoyao-signal-server/taoyao-signal/src/main/java/com/acgist/taoyao/signal/protocol/control/ControlWakeupProtocol.java b/taoyao-signal-server/taoyao-signal/src/main/java/com/acgist/taoyao/signal/protocol/control/ControlWakeupProtocol.java index da5bea7..5d2a376 100644 --- a/taoyao-signal-server/taoyao-signal/src/main/java/com/acgist/taoyao/signal/protocol/control/ControlWakeupProtocol.java +++ b/taoyao-signal-server/taoyao-signal/src/main/java/com/acgist/taoyao/signal/protocol/control/ControlWakeupProtocol.java @@ -5,6 +5,8 @@ import java.util.Map; import com.acgist.taoyao.boot.annotation.Description; import com.acgist.taoyao.boot.annotation.Protocol; import com.acgist.taoyao.boot.model.Message; +import com.acgist.taoyao.boot.model.MessageCode; +import com.acgist.taoyao.boot.model.MessageCodeException; import com.acgist.taoyao.signal.client.Client; import com.acgist.taoyao.signal.client.ClientType; import com.acgist.taoyao.signal.protocol.ProtocolControlAdapter; @@ -38,6 +40,9 @@ public class ControlWakeupProtocol extends ProtocolControlAdapter implements ICo @Override public void execute(String clientId, ClientType clientType, Client client, Client targetClient, Message message, Map body) { + if(client.equals(targetClient)) { + throw MessageCodeException.of(MessageCode.CODE_3400, "不能自己唤醒自己"); + } client.push(targetClient.request(message)); }