Better integration with OrleansDashboard

OrleansContrib · Aug 9, 2017 · 9d9242b · 9d9242b
1 parent ff7bdd7
commit 9d9242b
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 1 deletion.
diff --git a/Source/Orleankka.Runtime/Core/ActorEndpoint.Common.cs b/Source/Orleankka.Runtime/Core/ActorEndpoint.Common.cs
@@ -10,15 +10,18 @@ public Task Autorun()
             return Task.CompletedTask;
         }
 
+        [MessageArgumentAttribute]
         public Task<object> Receive(object message)
         {
             KeepAlive();
 
             return invoker.OnReceive(instance, message);
         }
 
+        [MessageArgumentAttribute]
         public Task ReceiveVoid(object message) => Receive(message);
 
+        [MessageArgumentAttribute]
         public Task Notify(object message) => Receive(message);
 
         async Task IRemindable.ReceiveReminder(string name, TickStatus status)

diff --git a/Source/Orleankka.Runtime/Core/ActorEndpoint.cs b/Source/Orleankka.Runtime/Core/ActorEndpoint.cs
@@ -28,15 +28,18 @@ public Task Autorun()
             return Task.CompletedTask;
         }
 
+        [MessageArgumentAttribute]
         public Task<object> Receive(object message)
         {
             KeepAlive();
 
             return invoker.OnReceive(instance, message);
         }
 
+        [MessageArgumentAttribute]
         public Task ReceiveVoid(object message) => Receive(message);
 
+        [MessageArgumentAttribute]
         public Task Notify(object message) => Receive(message);
 
         async Task IRemindable.ReceiveReminder(string name, TickStatus status)

diff --git a/Source/Orleankka.Runtime/Core/ActorTypeDeclaration.cs b/Source/Orleankka.Runtime/Core/ActorTypeDeclaration.cs
@@ -9,7 +9,6 @@
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp;
 
-using Orleans.CodeGeneration;
 using Orleans.Concurrency;
 using Orleans.MultiCluster;
 using Orleans.Placement;

diff --git a/Source/Orleankka.Runtime/Core/DashboardIntegration.cs b/Source/Orleankka.Runtime/Core/DashboardIntegration.cs
@@ -0,0 +1,11 @@
+using System;
+using System.Linq;
+
+namespace Orleankka.Core
+{
+    /// <summary>
+    /// FOR INTERNAL USE ONLY!
+    /// </summary>
+    [AttributeUsage(AttributeTargets.Method)]
+    public class MessageArgumentAttribute : Attribute {}
+}
diff --git a/Source/Orleankka.Runtime/Core/StatefulActorEndpoint.cs b/Source/Orleankka.Runtime/Core/StatefulActorEndpoint.cs
@@ -30,15 +30,18 @@ public Task Autorun()
             return Task.CompletedTask;
         }
 
+        [MessageArgumentAttribute]
         public Task<object> Receive(object message)
         {
             KeepAlive();
 
             return invoker.OnReceive(instance, message);
         }
 
+        [MessageArgumentAttribute]
         public Task ReceiveVoid(object message) => Receive(message);
 
+        [MessageArgumentAttribute]
         public Task Notify(object message) => Receive(message);
 
         async Task IRemindable.ReceiveReminder(string name, TickStatus status)

diff --git a/Source/Orleankka.Runtime/Orleankka.Runtime.csproj b/Source/Orleankka.Runtime/Orleankka.Runtime.csproj
@@ -74,6 +74,7 @@
     <Compile Include="Actor.cs" />
     <Compile Include="Cluster\StreamProviderConfiguration.cs" />
     <Compile Include="Core\IActorHost.cs" />
+    <Compile Include="Core\DashboardIntegration.cs" />
     <Compile Include="Core\StatefulActorEndpoint.cs">
       <AutoGen>True</AutoGen>
       <DesignTime>True</DesignTime>