public IDisposable BeginScope <TState>(TState state)
{
var scope = new TraceScope(_name, state);
scope.Context = TraceScope.Current?.Context ?? GetNewActivityContext();
return(TraceScope.Push(scope, _store));
}
public static IDisposable Push(TraceScope scope, TraceStore store)
{
if (scope == null)
{
throw new ArgumentNullException(nameof(scope));
}
if (store == null)
{
throw new ArgumentNullException(nameof(store));
}
//if (scope._name.Equals("Microsoft.AspNetCore.Hosting.Internal.WebHost", StringComparison.OrdinalIgnoreCase))
//{
// return new DisposableAction(() =>
// {
// //Current.Node.EndTime = DateTimeOffset.UtcNow;
// //Current = Current.Parent;
// });
//}
var temp = Current;
Current = scope;
Current.Parent = temp;
Current.Node = new ScopeNode()
{
StartTime = DateTimeOffset.UtcNow,
State = Current._state,
Name = Current._name
};
if (Current.Parent != null)
{
Current.Node.Parent = Current.Parent.Node;
Current.Parent.Node.Children.Add(Current.Node);
}
else
{
Current.Context.Root = Current.Node;
store.AddActivity(Current.Context);
}
return(new DisposableAction(() =>
{
Current.Node.EndTime = DateTimeOffset.UtcNow;
Current = Current.Parent;
}));
}