Added probe support for diagnostics

MassTransit · Sep 15, 2016 · a4a0beb · a4a0beb
1 parent 93f54de
commit a4a0beb
Show file tree

Hide file tree

Showing 33 changed files with 293 additions and 20 deletions.
diff --git a/src/Automatonymous.Tests/AsyncActivity_Specs.cs b/src/Automatonymous.Tests/AsyncActivity_Specs.cs
@@ -14,6 +14,7 @@ namespace Automatonymous.Tests
 {
     using System;
     using System.Threading.Tasks;
+    using GreenPipes;
     using NUnit.Framework;
 
 
@@ -45,6 +46,10 @@ void Visitable.Accept(StateMachineVisitor visitor)
             {
                 visitor.Visit(this);
             }
+
+            public void Probe(ProbeContext context)
+            {
+            }
         }
 
 

diff --git a/src/Automatonymous.Tests/Dependency_Specs.cs b/src/Automatonymous.Tests/Dependency_Specs.cs
@@ -15,6 +15,7 @@ namespace Automatonymous.Tests
     using System;
     using System.Threading.Tasks;
     using Activities;
+    using GreenPipes;
     using NUnit.Framework;
 
 
@@ -76,6 +77,10 @@ public void Accept(StateMachineVisitor visitor)
             {
                 visitor.Visit(this);
             }
+
+            public void Probe(ProbeContext context)
+            {
+            }
         }
 
 

diff --git a/src/Automatonymous.Tests/Request_Specs.cs b/src/Automatonymous.Tests/Request_Specs.cs
@@ -19,6 +19,7 @@ namespace Request_Specs
         using System.Reflection;
         using System.Threading.Tasks;
         using Binders;
+        using GreenPipes;
         using NUnit.Framework;
 
 
@@ -363,6 +364,10 @@ public Task Faulted<TException>(BehaviorExceptionContext<TInstance, TData, TExce
             {
                 return next.Faulted(context);
             }
+
+            public void Probe(ProbeContext context)
+            {
+            }
         }
 
 

diff --git a/src/Automatonymous/Accessors/DefaultInstanceStateAccessor.cs b/src/Automatonymous/Accessors/DefaultInstanceStateAccessor.cs
@@ -18,6 +18,7 @@ namespace Automatonymous.Accessors
     using System.Linq.Expressions;
     using System.Reflection;
     using System.Threading.Tasks;
+    using GreenPipes;
 
 
     /// <summary>
@@ -82,5 +83,10 @@ StateAccessor<TInstance> CreateDefaultAccessor()
 
             return new InitialIfNullStateAccessor<TInstance>(_initialState, new RawStateAccessor<TInstance>(_machine, expression, _observer));
         }
+
+        public void Probe(ProbeContext context)
+        {
+            _accessor.Value.Probe(context);
+        }
     }
 }
diff --git a/src/Automatonymous/Accessors/InitialIfNullStateAccessor.cs b/src/Automatonymous/Accessors/InitialIfNullStateAccessor.cs
@@ -16,6 +16,7 @@ namespace Automatonymous.Accessors
     using Activities;
     using Behaviors;
     using Contexts;
+    using GreenPipes;
 
 
     public class InitialIfNullStateAccessor<TInstance> :
@@ -51,5 +52,10 @@ Task StateAccessor<TInstance>.Set(InstanceContext<TInstance> context, State<TIns
         {
             return _stateAccessor.Set(context, state);
         }
+
+        public void Probe(ProbeContext context)
+        {
+            _stateAccessor.Probe(context);
+        }
     }
 }
diff --git a/src/Automatonymous/Accessors/IntStateAccessor.cs b/src/Automatonymous/Accessors/IntStateAccessor.cs
@@ -16,6 +16,7 @@ namespace Automatonymous.Accessors
     using System.Linq.Expressions;
     using System.Reflection;
     using System.Threading.Tasks;
+    using GreenPipes;
     using GreenPipes.Internals.Extensions;
     using GreenPipes.Internals.Reflection;
     using GreenPipes.Util;
@@ -74,5 +75,10 @@ static ReadWriteProperty<TInstance, int> GetCurrentStateProperty(Expression<Func
 
             return new ReadWriteProperty<TInstance, int>(propertyInfo);
         }
+
+        public void Probe(ProbeContext context)
+        {
+            context.Add("currentStateProperty", _property.Property.Name);
+        }
     }
 }
diff --git a/src/Automatonymous/Accessors/RawStateAccessor.cs b/src/Automatonymous/Accessors/RawStateAccessor.cs
@@ -16,6 +16,7 @@ namespace Automatonymous.Accessors
     using System.Linq.Expressions;
     using System.Reflection;
     using System.Threading.Tasks;
+    using GreenPipes;
     using GreenPipes.Internals.Extensions;
     using GreenPipes.Internals.Reflection;
     using GreenPipes.Util;
@@ -65,6 +66,11 @@ Task StateAccessor<TInstance>.Set(InstanceContext<TInstance> context, State<TIns
             return _observer.StateChanged(context, state, previousState);
         }
 
+        public void Probe(ProbeContext context)
+        {
+            context.Add("currentStateProperty", _property.Property.Name);
+        }
+
         static ReadWriteProperty<TInstance, State> GetCurrentStateProperty(Expression<Func<TInstance, State>> currentStateExpression)
         {
             PropertyInfo propertyInfo = currentStateExpression.GetPropertyInfo();

diff --git a/src/Automatonymous/Accessors/StringStateAccessor.cs b/src/Automatonymous/Accessors/StringStateAccessor.cs
@@ -16,6 +16,7 @@ namespace Automatonymous.Accessors
     using System.Linq.Expressions;
     using System.Reflection;
     using System.Threading.Tasks;
+    using GreenPipes;
     using GreenPipes.Internals.Extensions;
     using GreenPipes.Internals.Reflection;
     using GreenPipes.Util;
@@ -75,5 +76,10 @@ static ReadWriteProperty<TInstance, string> GetCurrentStateProperty(Expression<F
 
             return new ReadWriteProperty<TInstance, string>(propertyInfo);
         }
+
+        public void Probe(ProbeContext context)
+        {
+            context.Add("currentStateProperty", _property.Property.Name);
+        }
     }
 }
diff --git a/src/Automatonymous/Activities/ActionActivity.cs b/src/Automatonymous/Activities/ActionActivity.cs
@@ -1,4 +1,4 @@
-// Copyright 2011-2014 Chris Patterson, Dru Sellers
+// Copyright 2011-2016 Chris Patterson, Dru Sellers
 // 
 // Licensed under the Apache License, Version 2.0 (the "License"); you may not use 
 // this file except in compliance with the License. You may obtain a copy of the 
@@ -14,6 +14,7 @@ namespace Automatonymous.Activities
 {
     using System;
     using System.Threading.Tasks;
+    using GreenPipes;
 
 
     public class ActionActivity<TInstance> :
@@ -31,6 +32,11 @@ void Visitable.Accept(StateMachineVisitor visitor)
             visitor.Visit(this);
         }
 
+        public void Probe(ProbeContext context)
+        {
+            context.CreateScope("action");
+        }
+
         async Task Activity<TInstance>.Execute(BehaviorContext<TInstance> context, Behavior<TInstance> next)
         {
             _action(context);
@@ -50,7 +56,8 @@ Task Activity<TInstance>.Faulted<TException>(BehaviorExceptionContext<TInstance,
             return next.Faulted(context);
         }
 
-        Task Activity<TInstance>.Faulted<T, TException>(BehaviorExceptionContext<TInstance, T, TException> context, Behavior<TInstance, T> next)
+        Task Activity<TInstance>.Faulted<T, TException>(BehaviorExceptionContext<TInstance, T, TException> context,
+            Behavior<TInstance, T> next)
         {
             return next.Faulted(context);
         }
@@ -72,14 +79,20 @@ void Visitable.Accept(StateMachineVisitor visitor)
             visitor.Visit(this);
         }
 
+        public void Probe(ProbeContext context)
+        {
+            context.CreateScope("action");
+        }
+
         async Task Activity<TInstance, TData>.Execute(BehaviorContext<TInstance, TData> context, Behavior<TInstance, TData> next)
         {
             _action(context);
 
             await next.Execute(context).ConfigureAwait(false);
         }
 
-        Task Activity<TInstance, TData>.Faulted<TException>(BehaviorExceptionContext<TInstance, TData, TException> context, Behavior<TInstance, TData> next)
+        Task Activity<TInstance, TData>.Faulted<TException>(BehaviorExceptionContext<TInstance, TData, TException> context,
+            Behavior<TInstance, TData> next)
         {
             return next.Faulted(context);
         }

diff --git a/src/Automatonymous/Activities/AsyncActivity.cs b/src/Automatonymous/Activities/AsyncActivity.cs
@@ -1,4 +1,4 @@
-// Copyright 2011-2015 Chris Patterson, Dru Sellers
+// Copyright 2011-2016 Chris Patterson, Dru Sellers
 // 
 // Licensed under the Apache License, Version 2.0 (the "License"); you may not use 
 // this file except in compliance with the License. You may obtain a copy of the 
@@ -14,6 +14,7 @@ namespace Automatonymous.Activities
 {
     using System;
     using System.Threading.Tasks;
+    using GreenPipes;
 
 
     public class AsyncActivity<TInstance> :
@@ -31,6 +32,11 @@ void Visitable.Accept(StateMachineVisitor visitor)
             visitor.Visit(this);
         }
 
+        public void Probe(ProbeContext context)
+        {
+            context.CreateScope("action");
+        }
+
         async Task Activity<TInstance>.Execute(BehaviorContext<TInstance> context, Behavior<TInstance> next)
         {
             await _asyncAction(context).ConfigureAwait(false);
@@ -73,6 +79,11 @@ void Visitable.Accept(StateMachineVisitor visitor)
             visitor.Visit(this);
         }
 
+        public void Probe(ProbeContext context)
+        {
+            context.CreateScope("action");
+        }
+
         async Task Activity<TInstance, TData>.Execute(BehaviorContext<TInstance, TData> context, Behavior<TInstance, TData> next)
         {
             await _asyncAction(context).ConfigureAwait(false);

diff --git a/src/Automatonymous/Activities/AsyncFactoryActivity.cs b/src/Automatonymous/Activities/AsyncFactoryActivity.cs
@@ -15,6 +15,7 @@ namespace Automatonymous.Activities
     using System;
     using System.Threading.Tasks;
     using Behaviors;
+    using GreenPipes;
 
 
     public class AsyncFactoryActivity<TInstance, TData> :
@@ -32,6 +33,11 @@ void Visitable.Accept(StateMachineVisitor visitor)
             visitor.Visit(this);
         }
 
+        public void Probe(ProbeContext context)
+        {
+            context.CreateScope("activityFactory");
+        }
+
         async Task Activity<TInstance, TData>.Execute(BehaviorContext<TInstance, TData> context, Behavior<TInstance, TData> next)
         {
             Activity<TInstance, TData> activity = await _activityFactory(context).ConfigureAwait(false);
@@ -62,6 +68,11 @@ void Visitable.Accept(StateMachineVisitor visitor)
             visitor.Visit(this);
         }
 
+        public void Probe(ProbeContext context)
+        {
+            context.CreateScope("activityFactory");
+        }
+
         async Task Activity<TInstance>.Execute(BehaviorContext<TInstance> context, Behavior<TInstance> next)
         {
             Activity<TInstance> activity = await _activityFactory(context).ConfigureAwait(false);

diff --git a/src/Automatonymous/Activities/CatchFaultActivity.cs b/src/Automatonymous/Activities/CatchFaultActivity.cs
@@ -1,4 +1,4 @@
-// Copyright 2011-2015 Chris Patterson, Dru Sellers
+// Copyright 2011-2016 Chris Patterson, Dru Sellers
 // 
 // Licensed under the Apache License, Version 2.0 (the "License"); you may not use 
 // this file except in compliance with the License. You may obtain a copy of the 
@@ -14,6 +14,7 @@ namespace Automatonymous.Activities
 {
     using System;
     using System.Threading.Tasks;
+    using GreenPipes;
 
 
     /// <summary>
@@ -38,6 +39,13 @@ void Visitable.Accept(StateMachineVisitor visitor)
             visitor.Visit(this, x => _behavior.Accept(visitor));
         }
 
+        public void Probe(ProbeContext context)
+        {
+            var scope = context.CreateScope("catchFault");
+
+            _behavior.Probe(scope);
+        }
+
         Task Activity<TInstance>.Execute(BehaviorContext<TInstance> context, Behavior<TInstance> next)
         {
             return next.Execute(context);

diff --git a/src/Automatonymous/Activities/CompositeEventActivity.cs b/src/Automatonymous/Activities/CompositeEventActivity.cs
@@ -14,6 +14,7 @@ namespace Automatonymous.Activities
 {
     using System.Threading.Tasks;
     using Accessors;
+    using GreenPipes;
     using GreenPipes.Util;
 
 
@@ -41,6 +42,13 @@ void Visitable.Accept(StateMachineVisitor visitor)
             visitor.Visit(this);
         }
 
+        public void Probe(ProbeContext context)
+        {
+            var scope = context.CreateScope("compositeEvent");
+            scope.Add("event", _event.Name);
+            scope.Add("flag", _flag.ToString("X8"));
+        }
+
         async Task Activity<TInstance>.Execute(BehaviorContext<TInstance> context, Behavior<TInstance> next)
         {
             await Execute(context).ConfigureAwait(false);

diff --git a/src/Automatonymous/Activities/ConditionalEventActivity.cs b/src/Automatonymous/Activities/ConditionalEventActivity.cs
@@ -13,6 +13,7 @@
 namespace Automatonymous.Activities
 {
     using System.Threading.Tasks;
+    using GreenPipes;
 
 
     /// <summary>
@@ -37,6 +38,13 @@ public void Accept(StateMachineVisitor visitor)
             visitor.Visit(this, x => _activity.Accept(visitor));
         }
 
+        public void Probe(ProbeContext context)
+        {
+            var scope = context.CreateScope("conditional");
+
+            _activity.Probe(scope);
+        }
+
         Task Activity<TInstance, TData>.Execute(BehaviorContext<TInstance, TData> context, Behavior<TInstance, TData> next)
         {
             if (_filter(context))

diff --git a/src/Automatonymous/Activities/DataConverterActivity.cs b/src/Automatonymous/Activities/DataConverterActivity.cs
@@ -1,4 +1,4 @@
-// Copyright 2011-2015 Chris Patterson, Dru Sellers
+// Copyright 2011-2016 Chris Patterson, Dru Sellers
 // 
 // Licensed under the Apache License, Version 2.0 (the "License"); you may not use 
 // this file except in compliance with the License. You may obtain a copy of the 
@@ -13,6 +13,7 @@
 namespace Automatonymous.Activities
 {
     using System.Threading.Tasks;
+    using GreenPipes;
 
 
     public class DataConverterActivity<TInstance, TData> :
@@ -30,6 +31,11 @@ public void Accept(StateMachineVisitor visitor)
             visitor.Visit(this, x => _activity.Accept(visitor));
         }
 
+        public void Probe(ProbeContext context)
+        {
+            _activity.Probe(context);
+        }
+
         Task Activity<TInstance>.Execute(BehaviorContext<TInstance> context, Behavior<TInstance> next)
         {
             throw new AutomatonymousException("This activity requires a body with the event, but no body was specified.");