From 58441b6420026dd89fe34eca8ac5f0d0a4b61eea Mon Sep 17 00:00:00 2001
From: liuxiaobo <1224730913@qq.com>
Date: Thu, 29 May 2025 13:15:18 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=B3=A8=E5=86=8Crpc?=
 =?UTF-8?q?=E6=B6=88=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/natsService.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/service/natsService.go b/service/natsService.go
index 8027d02..18bf0d8 100644
--- a/service/natsService.go
+++ b/service/natsService.go
@@ -132,6 +132,11 @@ func (s *NatsService) Send(topic string, msg *ipb.InternalMsg) error {
 	return s.nats.Publish(topic, data)
 }
 
+// 注册rpc响应方法，参数固定为func(*ipb.InternalMsg)
+func (s *NatsService) RegisterRpcMessage(cmd int32, delegate any) {
+	s.rpcProcessor.RegisterMessage(cmd, ipb.InternalMsg{}, delegate)
+}
+
 func (s *NatsService) Call(rpcTopic string, msg *ipb.InternalMsg, cb func(msg *ipb.InternalMsg)) error {
 	data, _ := proto.Marshal(ipb.MakeRpcMsg(msg.ServiceName, msg.ConnId, msg.UserId, msg.MsgId, msg.Msg))
 	err := s.nats.Publish(rpcTopic, data)