fix: CallClientStream result has error, result.Result() is nil to *tr…

…i.Response will cause a panic (#2732)
apache · Sep 26, 2024 · f03922a · f03922a
1 parent 4329b00
commit f03922a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/protocol/triple/server.go b/protocol/triple/server.go
@@ -270,7 +270,12 @@ func (s *Server) handleServiceWithInfo(interfaceName string, invoker protocol.In
 					ctx = context.WithValue(ctx, constant.AttachmentKey, attachments)
 					invo := invocation.NewRPCInvocation(m.Name, args, attachments)
 					res := invoker.Invoke(ctx, invo)
-					return res.Result().(*tri.Response), res.Error()
+					if triResp, ok := res.Result().(*tri.Response); ok {
+						return triResp, res.Error()
+					}
+					// please refer to proxy/proxy_factory/ProxyInvoker.Invoke
+					triResp := tri.NewResponse([]interface{}{res.Result()})
+					return triResp, res.Error()
 				},
 				opts...,
 			)