public void Disconnect(bool dispose, Exception exception)
{
Dictionary <uint, TaskCompletionSource <Message> > pendingMethods = null;
Dictionary <SignalMatchRule, SignalHandler> signalHandlers = null;
Dictionary <string, Action <ServiceOwnerChangedEventArgs, Exception> > nameOwnerWatchers = null;
lock (_gate)
{
if (_state == ConnectionState.Disconnected || _state == ConnectionState.Created)
{
return;
}
_state = ConnectionState.Disconnected;
_disposed = dispose;
_stream.Dispose();
_disconnectReason = exception;
pendingMethods = _pendingMethods;
_pendingMethods = null;
signalHandlers = _signalHandlers;
_signalHandlers = null;
nameOwnerWatchers = _nameOwnerWatchers;
_nameOwnerWatchers = null;
_serviceNameRegistrations.Clear();
_onDisconnect = null;
Func <Exception> createException = () =>
dispose?Connection.CreateDisposedException() : new DisconnectedException(exception);
foreach (var watcher in nameOwnerWatchers)
{
watcher.Value(default(ServiceOwnerChangedEventArgs), createException());
}
foreach (var handler in signalHandlers)
{
handler.Value(null, createException());
}
foreach (var tcs in pendingMethods.Values)
{
tcs.SetException(createException());
}
}
if (_onDisconnect != null)
{
_onDisconnect(dispose ? null : exception);
}
}
void Dispose(Exception disconnectReason)
{
Dictionary <uint, PendingMethodCont> pendingMethods = null;
lock (_gate)
{
pendingMethods = Interlocked.Exchange(ref _pendingMethods, null);
if (pendingMethods == null)
{
return;
}
_localName = null;
_stream.Dispose();
_signalHandlers.Clear();
foreach (var prov in proxyProviders)
{
prov.Dispose();
}
foreach (var h in GlobalInterfaceHandlers)
{
h.Value.Dispose();
}
GlobalInterfaceHandlers.Clear();
void di(IMethodHandlerPathPart mhpp)
{
foreach (var mh in mhpp.InterfaceHandlers)
{
mh.Value.Dispose();
}
foreach (var sp in mhpp.SubParts)
{
di(sp.Value);
}
}
di(RootPath);
RootPath.OurInterfaceHandlers.Clear();
}
var e = (disconnectReason != null) ?
(Exception) new DisconnectedException(disconnectReason) :
new ObjectDisposedException(typeof(Connection).FullName);
foreach (var mc in pendingMethods.Values)
{
mc.Reply.SetException(e);
}
_onDisconnect?.Invoke(disconnectReason);
}
private void RemoveStream(IMessageStream client, Exception e)
{
lock (_gate)
{
if (IsDisposed)
{
return;
}
var clientsUpdated = new IMessageStream[_clients.Length - 1];
for (int i = 0, j = 0; i < _clients.Length; i++)
{
if (_clients[i] != client)
{
clientsUpdated[j++] = _clients[i];
}
}
_clients = clientsUpdated;
}
client.Dispose();
}
