@@ -87,7 +87,7 @@ public void CleanupInvocationsByInstance(string instanceId)
8787 {
8888 if ( invocation . RouterInstanceId == instanceId )
8989 {
90- var message = new CompletionMessage ( invocationId , $ "Connection '{ invocation . ConnectionId } ' is disconnected.", null , false ) ;
90+ var message = CompletionMessage . WithError ( invocationId , $ "Connection '{ invocation . ConnectionId } ' is disconnected.") ;
9191
9292 invocation . Complete ( invocation . Tcs , message ) ;
9393 _pendingInvocations . TryRemove ( invocationId , out _ ) ;
@@ -128,7 +128,7 @@ public bool TryCompleteResult(string connectionId, ClientCompletionMessage messa
128128 if ( protocol == null )
129129 {
130130 var errorMessage = $ "Not supported protocol { message . Protocol } by server.";
131- return TryCompleteResult ( connectionId , new CompletionMessage ( message . InvocationId , errorMessage , null , false ) ) ;
131+ return TryCompleteResult ( connectionId , CompletionMessage . WithError ( message . InvocationId , errorMessage ) ) ;
132132 }
133133
134134 var payload = message . Payload ;
@@ -146,6 +146,12 @@ public bool TryCompleteResult(string connectionId, ClientCompletionMessage messa
146146 return false ;
147147 }
148148
149+ public bool TryCompleteResult ( string connectionId , ErrorCompletionMessage message )
150+ {
151+ var errorMessage = CompletionMessage . WithError ( message . InvocationId , message . Error ) ;
152+ return TryCompleteResult ( connectionId , errorMessage ) ;
153+ }
154+
149155 // Implemented for interface IInvocationBinder
150156 public Type GetReturnType ( string invocationId )
151157 {
0 commit comments