// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
 
using System;
using System.ServiceModel.Channels;
 
/// <summary>
/// Base channel class that uses an ChannelMessageInterceptor
/// </summary>
class InterceptingChannelBase<TChannel> : ChannelBase
    where TChannel : class, IChannel
{
    ChannelMessageInterceptor interceptor;
    TChannel innerChannel;
 
    protected InterceptingChannelBase(
        ChannelManagerBase manager, ChannelMessageInterceptor interceptor, TChannel innerChannel)
        : base(manager)
    {
        if (innerChannel == null)
        {
            throw new ArgumentException("InterceptingChannelBase requires a non-null inner channel.", "innerChannel");
        }
 
        this.interceptor = interceptor;
        this.innerChannel = innerChannel;
    }
 
    protected TChannel InnerChannel
    {
        get
        {
            return this.innerChannel;
        }
    }
 
    public override T GetProperty<T>()
    {
        T baseProperty = base.GetProperty<T>();
        if (baseProperty != null)
        {
            return baseProperty;
        }
 
        return this.InnerChannel.GetProperty<T>();
    }
 
    protected override void OnAbort()
    {
        this.innerChannel.Abort();
    }
 
    protected override IAsyncResult OnBeginClose(TimeSpan timeout, AsyncCallback callback, object state)
    {
        return this.innerChannel.BeginClose(timeout, callback, state);
    }
 
    protected override IAsyncResult OnBeginOpen(TimeSpan timeout, AsyncCallback callback, object state)
    {
        return this.innerChannel.BeginOpen(timeout, callback, state);
    }
 
    protected override void OnClose(TimeSpan timeout)
    {
        this.innerChannel.Close(timeout);
    }
 
    protected override void OnEndClose(IAsyncResult result)
    {
        this.innerChannel.EndClose(result);
    }
 
    protected override void OnEndOpen(IAsyncResult result)
    {
        this.innerChannel.EndOpen(result);
    }
 
    protected override void OnOpen(TimeSpan timeout)
    {
        this.innerChannel.Open(timeout);
    }
 
    protected void OnReceive(ref Message message)
    {
        this.interceptor.OnReceive(ref message);
    }
 
    protected void OnSend(ref Message message)
    {
        this.interceptor.OnSend(ref message);
    }
}