Ejemplo n.º 1
0
 protected override void DoClose()
 {
     if (this.state <= 1)
     {
         // Update all internal state before the closeFuture is notified.
         if (this.localAddress != null)
         {
             LocalChannelRegistry.Unregister(this.localAddress);
             this.localAddress = null;
         }
         this.state = 2;
     }
 }
Ejemplo n.º 2
0
 protected override void DoClose()
 {
     if (Volatile.Read(ref v_state) <= 1)
     {
         // Update all internal state before the closeFuture is notified.
         var thisLocalAddr = Volatile.Read(ref v_localAddress);
         if (thisLocalAddr is object)
         {
             LocalChannelRegistry.Unregister(thisLocalAddr);
             _ = Interlocked.Exchange(ref v_localAddress, null);
         }
         _ = Interlocked.Exchange(ref v_state, 2);
     }
 }
Ejemplo n.º 3
0
        protected override void DoClose()
        {
            var peer     = Volatile.Read(ref v_peer);
            var oldState = Volatile.Read(ref v_state);

            try
            {
                if (oldState != State.Closed)
                {
                    // Update all internal state before the closeFuture is notified.
                    var thisLocalAddr = Volatile.Read(ref v_localAddress);
                    if (thisLocalAddr is object)
                    {
                        if (Parent is null)
                        {
                            LocalChannelRegistry.Unregister(thisLocalAddr);
                        }
                        _ = Interlocked.Exchange(ref v_localAddress, null);
                    }

                    // State change must happen before finishPeerRead to ensure writes are released either in doWrite or
                    // channelRead.
                    _ = Interlocked.Exchange(ref v_state, State.Closed);

                    // Preserve order of event and force a read operation now before the close operation is processed.
                    if (SharedConstants.False < (uint)Volatile.Read(ref v_writeInProgress) && peer is object)
                    {
                        FinishPeerRead(peer);
                    }

                    TaskCompletionSource promise = Volatile.Read(ref v_connectPromise);
                    if (promise is object)
                    {
                        // Use tryFailure() instead of setFailure() to avoid the race against cancel().
                        _ = promise.TrySetException(DoCloseClosedChannelException);
                        _ = Interlocked.Exchange(ref v_connectPromise, null);
                    }
                }

                if (peer is object)
                {
                    _ = Interlocked.Exchange(ref v_peer, null);
                    // Always call peer.eventLoop().execute() even if peer.eventLoop().inEventLoop() is true.
                    // This ensures that if both channels are on the same event loop, the peer's channelInActive
                    // event is triggered *after* this peer's channelInActive event
                    IEventLoop peerEventLoop = peer.EventLoop;
                    bool       peerIsActive  = peer.IsActive;
                    try
                    {
                        peerEventLoop.Execute(() => peer.TryClose(peerIsActive));
                    }
                    catch (Exception cause)
                    {
                        Logger.Warn("Releasing Inbound Queues for channels {}-{} because exception occurred!", this, peer, cause);

                        if (peerEventLoop.InEventLoop)
                        {
                            peer.ReleaseInboundBuffers();
                        }
                        else
                        {
                            // inboundBuffers is a SPSC so we may leak if the event loop is shutdown prematurely or
                            // rejects the close Runnable but give a best effort.
                            _ = peer.CloseAsync();
                        }
                        throw;
                    }
                }
            }
            finally
            {
                // Release all buffers if the Channel was already registered in the past and if it was not closed before.
                if (oldState != State.Closed)
                {
                    // We need to release all the buffers that may be put into our inbound queue since we closed the Channel
                    // to ensure we not leak any memory. This is fine as it basically gives the same guarantees as TCP which
                    // means even if the promise was notified before its not really guaranteed that the "remote peer" will
                    // see the buffer at all.
                    ReleaseInboundBuffers();
                }
            }
        }
        protected override void DoClose()
        {
            var peer     = this.peer;
            var oldState = this.state;

            try
            {
                if (oldState != State.Closed)
                {
                    // Update all internal state before the closeFuture is notified.
                    if (this.localAddress != null)
                    {
                        if (this.Parent == null)
                        {
                            LocalChannelRegistry.Unregister(this.localAddress);
                        }
                        this.localAddress = null;
                    }

                    // State change must happen before finishPeerRead to ensure writes are released either in doWrite or
                    // channelRead.
                    this.state = State.Closed;

                    // Preserve order of event and force a read operation now before the close operation is processed.
                    this.FinishPeerRead(this);

                    TaskCompletionSource promise = this.connectPromise;
                    if (promise != null)
                    {
                        // Use tryFailure() instead of setFailure() to avoid the race against cancel().
                        promise.TrySetException(DoCloseClosedChannelException);
                        this.connectPromise = null;
                    }
                }

                if (peer != null)
                {
                    this.peer = null;
                    // Need to execute the close in the correct EventLoop (see https://github.com/netty/netty/issues/1777).
                    // Also check if the registration was not done yet. In this case we submit the close to the EventLoop
                    // to make sure its run after the registration completes
                    // (see https://github.com/netty/netty/issues/2144).
                    IEventLoop peerEventLoop = peer.EventLoop;
                    bool       peerIsActive  = peer.Active;
                    if (peerEventLoop.InEventLoop && !this.registerInProgress)
                    {
                        peer.TryClose(peerIsActive);
                    }
                    else
                    {
                        try
                        {
                            peerEventLoop.Execute(() => peer.TryClose(peerIsActive));
                        }
                        catch (Exception cause)
                        {
                            Logger.Warn("Releasing Inbound Queues for channels {}-{} because exception occurred!", this, peer, cause);

                            if (peerEventLoop.InEventLoop)
                            {
                                peer.ReleaseInboundBuffers();
                            }
                            else
                            {
                                // inboundBuffers is a SPSC so we may leak if the event loop is shutdown prematurely or
                                // rejects the close Runnable but give a best effort.
                                peer.CloseAsync();
                            }
                            throw;
                        }
                    }
                }
            }
            finally
            {
                // Release all buffers if the Channel was already registered in the past and if it was not closed before.
                if (oldState != State.Closed)
                {
                    // We need to release all the buffers that may be put into our inbound queue since we closed the Channel
                    // to ensure we not leak any memory. This is fine as it basically gives the same guarantees as TCP which
                    // means even if the promise was notified before its not really guaranteed that the "remote peer" will
                    // see the buffer at all.
                    this.ReleaseInboundBuffers();
                }
            }
        }