1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113
|
//------------------------------------------------------------
// Copyright (c) Microsoft Corporation. All rights reserved.
//------------------------------------------------------------
namespace System.ServiceModel.Channels
{
using System;
using System.Collections.Generic;
using System.ServiceModel;
abstract class ContextInputChannelBase<TChannel> : LayeredChannel<TChannel> where TChannel : class, IInputChannel
{
ContextExchangeMechanism contextExchangeMechanism;
ServiceContextProtocol contextProtocol;
protected ContextInputChannelBase(ChannelManagerBase channelManager, TChannel innerChannel, ContextExchangeMechanism contextExchangeMechanism)
: base(channelManager, innerChannel)
{
this.contextExchangeMechanism = contextExchangeMechanism;
this.contextProtocol = new ServiceContextProtocol(contextExchangeMechanism);
}
public EndpointAddress LocalAddress
{
get { return this.InnerChannel.LocalAddress; }
}
public IAsyncResult BeginReceive(TimeSpan timeout, AsyncCallback callback, object state)
{
return this.InnerChannel.BeginReceive(timeout, callback, state);
}
public IAsyncResult BeginReceive(AsyncCallback callback, object state)
{
return this.InnerChannel.BeginReceive(callback, state);
}
public IAsyncResult BeginTryReceive(TimeSpan timeout, AsyncCallback callback, object state)
{
return this.InnerChannel.BeginTryReceive(timeout, callback, state);
}
public IAsyncResult BeginWaitForMessage(TimeSpan timeout, AsyncCallback callback, object state)
{
return this.InnerChannel.BeginWaitForMessage(timeout, callback, state);
}
public Message EndReceive(IAsyncResult result)
{
Message message = this.InnerChannel.EndReceive(result);
ProcessContextHeader(message);
return message;
}
public bool EndTryReceive(IAsyncResult result, out Message message)
{
if (this.InnerChannel.EndTryReceive(result, out message))
{
ProcessContextHeader(message);
return true;
}
else
{
return false;
}
}
public bool EndWaitForMessage(IAsyncResult result)
{
return this.InnerChannel.EndWaitForMessage(result);
}
public Message Receive(TimeSpan timeout)
{
Message message = this.InnerChannel.Receive(timeout);
ProcessContextHeader(message);
return message;
}
public Message Receive()
{
Message message = this.InnerChannel.Receive();
ProcessContextHeader(message);
return message;
}
public bool TryReceive(TimeSpan timeout, out Message message)
{
if (this.InnerChannel.TryReceive(timeout, out message))
{
ProcessContextHeader(message);
return true;
}
else
{
return false;
}
}
public bool WaitForMessage(TimeSpan timeout)
{
return this.InnerChannel.WaitForMessage(timeout);
}
void ProcessContextHeader(Message message)
{
if (message != null)
{
contextProtocol.OnIncomingMessage(message);
}
}
}
}
|