Imported Upstream version 4.0.0~alpha1

Former-commit-id: 806294f5ded97629b74c85c09952f2a74fe182d9
This commit is contained in:
Jo Shields
2015-04-07 09:35:12 +01:00
parent 283343f570
commit 3c1f479b9d
22469 changed files with 2931443 additions and 869343 deletions

View File

@@ -0,0 +1,58 @@
//-----------------------------------------------------------------------------
// Copyright (c) Microsoft Corporation. All rights reserved.
//-----------------------------------------------------------------------------
namespace System.ServiceModel.Diagnostics
{
using System.Runtime;
using System.Diagnostics;
abstract class TraceAsyncResult : AsyncResult
{
static Action<AsyncCallback, IAsyncResult> waitResultCallback = new Action<AsyncCallback, IAsyncResult>(DoCallback);
protected TraceAsyncResult(AsyncCallback callback, object state) :
base(callback, state)
{
if (TraceUtility.MessageFlowTracingOnly)
{
this.CallbackActivity = ServiceModelActivity.CreateLightWeightAsyncActivity(Trace.CorrelationManager.ActivityId);
base.VirtualCallback = waitResultCallback;
}
else if (DiagnosticUtility.ShouldUseActivity)
{
this.CallbackActivity = ServiceModelActivity.Current;
if (this.CallbackActivity != null)
{
base.VirtualCallback = waitResultCallback;
}
}
}
public ServiceModelActivity CallbackActivity
{
get;
private set;
}
static void DoCallback(AsyncCallback callback, IAsyncResult result)
{
if (result is TraceAsyncResult)
{
TraceAsyncResult thisPtr = result as TraceAsyncResult;
Fx.Assert(thisPtr.CallbackActivity != null, "this shouldn't be hooked up if we don't have a CallbackActivity");
if (TraceUtility.MessageFlowTracingOnly)
{
Trace.CorrelationManager.ActivityId = thisPtr.CallbackActivity.Id;
thisPtr.CallbackActivity = null;
}
using (ServiceModelActivity.BoundOperation(thisPtr.CallbackActivity))
{
callback(result);
}
}
}
}
}