DotNetty/test/DotNetty.Transport.Tests.Performance/Utilities/ReadFinishedHandler.cs

// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.

namespace DotNetty.Transport.Tests.Performance.Utilities
{
    using System;
    using DotNetty.Common.Utilities;
    using DotNetty.Transport.Channels;

    public class ReadFinishedHandler : ChannelHandlerAdapter
    {
        readonly int expectedReads;
        readonly IReadFinishedSignal signal;
        int actualReads;

        public ReadFinishedHandler(IReadFinishedSignal signal, int expectedReads)
        {
            this.signal = signal;
            this.expectedReads = expectedReads;
        }

        public override void ChannelRead(IChannelHandlerContext context, object message)
        {
            ReferenceCountUtil.Release(message);
            if (++this.actualReads == this.expectedReads)
            {
                this.signal.Signal();
            }
        }

        public override void ExceptionCaught(IChannelHandlerContext context, Exception exception) => Console.WriteLine(exception.ToString());
    }
}