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
|
//-----------------------------------------------------------------------------
// Copyright (c) Microsoft Corporation. All rights reserved.
//-----------------------------------------------------------------------------
namespace System.ServiceModel.Channels
{
using System.Runtime;
using System.ServiceModel.Dispatcher;
sealed class BufferedReceiveMessageProperty
{
const string PropertyName = "BufferedReceiveMessageProperty";
MessageBuffer messageBuffer;
static MessageBuffer dummyMessageBuffer = BufferedMessage.CreateMessage(MessageVersion.Default, string.Empty).CreateBufferedCopy(1);
internal BufferedReceiveMessageProperty(ref MessageRpc rpc)
{
this.RequestContext = new BufferedRequestContext(rpc.RequestContext);
rpc.RequestContext = this.RequestContext;
this.Notification = rpc.InvokeNotification;
}
public static string Name
{
get { return PropertyName; }
}
// Implementation specifc storage
public object UserState
{
get;
set;
}
// The original RequestContext that was created by the ChannelHandler
public BufferedRequestContext RequestContext
{
get;
private set;
}
// The 'Manual Concurrency' notification which allows higher layers to notify the dispatcher of an event
// e.g. The event associated with re-pumping the above buffered RequestContext again
internal IInvokeReceivedNotification Notification
{
get;
private set;
}
public void RegisterForReplay(OperationContext operationContext)
{
this.messageBuffer = (MessageBuffer)operationContext.IncomingMessageProperties[ChannelHandler.MessageBufferPropertyName];
// cannot remove the MessageBufferProperty from messageProperties because it causes the message buffer associated with the property
// to be disposed of. Assigning null to the property has the same effect, so we assign dummyMessageBuffer to the property.
operationContext.IncomingMessageProperties[ChannelHandler.MessageBufferPropertyName] = dummyMessageBuffer;
}
public void ReplayRequest()
{
Message requestMessage = this.messageBuffer.CreateMessage();
// re-injecting the MessageBufferProperty here so that it can be reused by the dispatch layer
// (ChannelHandler.DispatchAndReleasePump to be specific) instead of recreating the buffer for replay.
requestMessage.Properties[ChannelHandler.MessageBufferPropertyName] = this.messageBuffer;
this.RequestContext.ReInitialize(requestMessage);
}
public static bool TryGet(Message message, out BufferedReceiveMessageProperty property)
{
Fx.Assert(message != null, "The Message parameter is null");
return TryGet(message.Properties, out property);
}
public static bool TryGet(MessageProperties properties, out BufferedReceiveMessageProperty property)
{
Fx.Assert(properties != null, "The MessageProperties parameter is null");
object value = null;
if (properties.TryGetValue(PropertyName, out value))
{
property = value as BufferedReceiveMessageProperty;
}
else
{
property = null;
}
return property != null;
}
}
}
|
