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
|
//------------------------------------------------------------
// Copyright (c) Microsoft Corporation. All rights reserved.
//------------------------------------------------------------
namespace System.ServiceModel.Dispatcher
{
using System.ServiceModel.Channels;
class DurableMessageDispatchInspector : IDispatchMessageInspector
{
public const string NewDurableInstanceIdPropertyName = "newDurableInstanceIdProperty";
const string suppressContextOnReply = "suppressContextOnReply";
SessionMode sessionMode;
public DurableMessageDispatchInspector(SessionMode sessionMode)
{
this.sessionMode = sessionMode;
}
public static void SuppressContextOnReply(OperationContext operationContext)
{
if (operationContext == null)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("operationContext");
}
operationContext.OutgoingMessageProperties[suppressContextOnReply] = true;
}
public object AfterReceiveRequest(ref Message request, IClientChannel channel, InstanceContext instanceContext)
{
if (instanceContext == null)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("instanceContext");
}
if (request == null)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("request");
}
if (sessionMode != SessionMode.NotAllowed)
{
object result = null;
if (request.Properties.TryGetValue(NewDurableInstanceIdPropertyName, out result))
{
return result.ToString();
}
}
return null;
}
public void BeforeSendReply(ref Message reply, object correlationState)
{
try
{
if (reply != null)
{
ContextMessageProperty context = null;
if (sessionMode == SessionMode.NotAllowed || reply.Properties.ContainsKey(suppressContextOnReply))
{
if (ContextMessageProperty.TryGet(reply, out context))
{
context.Context.Clear();
}
}
else
{
string newInstanceId = correlationState as string;
if (newInstanceId != null)
{
if (!ContextMessageProperty.TryGet(reply, out context))
{
context = new ContextMessageProperty();
context.Context[WellKnownContextProperties.InstanceId] = newInstanceId;
context.AddOrReplaceInMessage(reply);
}
else
{
context.Context[WellKnownContextProperties.InstanceId] = newInstanceId;
}
}
}
}
}
finally
{
DurableInstance durableInstance = OperationContext.Current.InstanceContext.Extensions.Find<DurableInstance>();
if (durableInstance == null)
{
throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(
new InvalidOperationException(
SR2.GetString(
SR2.RequiredInstanceContextExtensionNotFound,
typeof(DurableInstance).Name)));
}
//Decrement InstanceActivity Count
durableInstance.DecrementActivityCount();
}
}
}
}
|
